From 22ba0e5b97fb49e5ce0efa1a91953f05d8239fa1 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=BA=90=E6=96=87=E9=9B=A8?=
 <41315874+fumiama@users.noreply.github.com>
Date: Fri, 28 Jul 2023 13:41:29 +0800
Subject: [PATCH] fix: WSServer Unmarshaller

---
 driver/wsserver.go | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/driver/wsserver.go b/driver/wsserver.go
index 2c2efb5..0f37b76 100644
--- a/driver/wsserver.go
+++ b/driver/wsserver.go
@@ -1,6 +1,7 @@
 package driver
 
 import (
+	"encoding/json"
 	"io"
 	"net"
 	"net/http"
@@ -25,6 +26,18 @@ type WSServer struct {
 	AccessToken string
 	lstn        net.Listener
 	caller      chan *WSSCaller
+
+	json.Unmarshaler
+}
+
+// UnmarshalJSON init WSServer with waitn=16
+func (wss *WSServer) UnmarshalJSON(data []byte) error {
+	err := json.Unmarshal(data, wss)
+	if err != nil {
+		return err
+	}
+	wss.caller = make(chan *WSSCaller, 16)
+	return nil
 }
 
 // NewWebSocketServer 使用反向WS通信