diff --git a/network/websocketclient.go b/network/websocketclient.go index fa687a0..6d82eb9 100644 --- a/network/websocketclient.go +++ b/network/websocketclient.go @@ -38,7 +38,7 @@ type WebsocketClient struct { timeoutsec time.Duration bRun bool - ping string + ping []byte } const ( @@ -71,12 +71,12 @@ func (ws *WebsocketClient) Init(slf IWebsocketClient, strurl, strProxyPath strin } ws.url = strurl - ws.ping = `ping` + ws.ping = []byte(`ping`) return nil } func (ws *WebsocketClient) SetPing(ping string) { - ws.ping = ping + ws.ping = []byte(ping) } //OnRun ... @@ -182,7 +182,7 @@ func (ws *WebsocketClient) writeMsg() error { case <-timerC: if ws.state == 2 { - err := ws.WriteMessage([]byte(ws.ping)) + err := ws.WriteMessage(ws.ping) if err != nil { service.GetLogger().Printf(service.LEVER_WARN, "websocket client is disconnect [%s],information is %v", ws.url, err) ws.state = 0 diff --git a/network/websocketserver.go b/network/websocketserver.go index 145f2ed..1eb16c6 100644 --- a/network/websocketserver.go +++ b/network/websocketserver.go @@ -23,6 +23,8 @@ type IWebsocketServer interface { CreateClient(conn *websocket.Conn) *WSClient Disconnect(clientid uint64) ReleaseClient(pclient *WSClient) + Clients() []uint64 + BroadcastMsg(messageType int, msg []byte) int } type IMessageReceiver interface {