From 53c83367f35ac88d243d1e9fe12d98b054460fb8 Mon Sep 17 00:00:00 2001
From: SeongUk Cho <issacs98@gmail.com>
Date: Mon, 6 Feb 2023 22:26:30 +0900
Subject: [PATCH] generic codec

---
 go.mod                 |  2 +-
 websocket/websocket.go | 12 ++++++------
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/go.mod b/go.mod
index 1b6c3b80d3..dd851e214b 100644
--- a/go.mod
+++ b/go.mod
@@ -1,6 +1,6 @@
 module golang.org/x/net
 
-go 1.17
+go 1.18
 
 require (
 	golang.org/x/sys v0.4.0
diff --git a/websocket/websocket.go b/websocket/websocket.go
index 90a2257cd5..5dfb107e0c 100644
--- a/websocket/websocket.go
+++ b/websocket/websocket.go
@@ -298,13 +298,13 @@ func (ws *Conn) Config() *Config { return ws.config }
 func (ws *Conn) Request() *http.Request { return ws.request }
 
 // Codec represents a symmetric pair of functions that implement a codec.
-type Codec struct {
-	Marshal   func(v interface{}) (data []byte, payloadType byte, err error)
+type Codec[T any] struct {
+	Marshal   func(v T) (data []byte, payloadType byte, err error)
 	Unmarshal func(data []byte, payloadType byte, v interface{}) (err error)
 }
 
 // Send sends v marshaled by cd.Marshal as single frame to ws.
-func (cd Codec) Send(ws *Conn, v interface{}) (err error) {
+func (cd Codec[T]) Send(ws *Conn, v T) (err error) {
 	data, payloadType, err := cd.Marshal(v)
 	if err != nil {
 		return err
@@ -326,7 +326,7 @@ func (cd Codec) Send(ws *Conn, v interface{}) (err error) {
 // limit, ErrFrameTooLarge is returned; in this case frame is not read off wire
 // completely. The next call to Receive would read and discard leftover data of
 // previous oversized frame before processing next frame.
-func (cd Codec) Receive(ws *Conn, v interface{}) (err error) {
+func (cd Codec[T]) Receive(ws *Conn, v interface{}) (err error) {
 	ws.rio.Lock()
 	defer ws.rio.Unlock()
 	if ws.frameReader != nil {
@@ -416,7 +416,7 @@ Trivial usage:
 	data = []byte{0, 1, 2}
 	websocket.Message.Send(ws, data)
 */
-var Message = Codec{marshal, unmarshal}
+var Message = Codec[interface{}]{Marshal: marshal, Unmarshal: unmarshal}
 
 func jsonMarshal(v interface{}) (msg []byte, payloadType byte, err error) {
 	msg, err = json.Marshal(v)
@@ -446,4 +446,4 @@ Trivial usage:
 	// send JSON type T
 	websocket.JSON.Send(ws, data)
 */
-var JSON = Codec{jsonMarshal, jsonUnmarshal}
+var JSON = Codec[interface{}]{Marshal: jsonMarshal, Unmarshal: jsonUnmarshal}