optimize: use sonic instead of encoding/json

json.go

@@ -8,7 +8,7 @@
 package websocket
 
 import (
-	"encoding/json"
+	"github.com/bytedance/sonic"
 	"io"
 )
 
@@ -21,7 +21,7 @@ func (c *Conn) WriteJSON(v interface{}) error {
 	if err != nil {
 		return err
 	}
-	err1 := json.NewEncoder(w).Encode(v)
+	err1 := sonic.ConfigDefault.NewEncoder(w).Encode(v)
 	err2 := w.Close()
 	if err1 != nil {
 		return err1
@@ -39,7 +39,7 @@ func (c *Conn) ReadJSON(v interface{}) error {
 	if err != nil {
 		return err
 	}
-	err = json.NewDecoder(r).Decode(v)
+	err = sonic.ConfigDefault.NewDecoder(r).Decode(v)
 	if err == io.EOF {
 		// One value is expected in the message.
 		err = io.ErrUnexpectedEOF

json_test.go

@@ -9,7 +9,9 @@ package websocket
 
 import (
 	"bytes"
-	"encoding/json"
+	"errors"
+	"github.com/bytedance/sonic"
+	"github.com/bytedance/sonic/decoder"
 	"io"
 	"reflect"
 	"testing"
@@ -56,7 +58,7 @@ func TestPartialJSONRead(t *testing.T) {
 
 	// Partial JSON values.
 
-	data, err := json.Marshal(v)
+	data, err := sonic.Marshal(v)
 	if err != nil {
 		t.Fatal(err)
 	}
@@ -82,7 +84,7 @@ func TestPartialJSONRead(t *testing.T) {
 
 	for i := 0; i < messageCount; i++ {
 		err := rc.ReadJSON(&v)
-		if err != io.ErrUnexpectedEOF {
+		if err != io.ErrUnexpectedEOF && !errors.As(err, &decoder.SyntaxError{}) {
 			t.Error("read", i, err)
 		}
 	}