fix: WSServer Unmarshaller

wdvxdr1123 · Jul 28, 2023 · 8d5946e · 8d5946e
1 parent 22ba0e5
commit 8d5946e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/driver/wsserver.go b/driver/wsserver.go
@@ -11,6 +11,7 @@ import (
 	"sync"
 	"sync/atomic"
 	"time"
+	"unsafe"
 
 	"github.com/RomiChan/websocket"
 	log "github.com/sirupsen/logrus"
@@ -32,7 +33,11 @@ type WSServer struct {
 
 // UnmarshalJSON init WSServer with waitn=16
 func (wss *WSServer) UnmarshalJSON(data []byte) error {
-	err := json.Unmarshal(data, wss)
+	type jsoncfg struct {
+		Url         string // ws连接地址
+		AccessToken string
+	}
+	err := json.Unmarshal(data, (*jsoncfg)(unsafe.Pointer(wss)))
 	if err != nil {
 		return err
 	}