blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-04-24 10:22:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

ws update

Browse Source
This commit is contained in:
Gordon 2023-03-23 13:17:06 +08:00
parent c96d6b48d1
commit ac8a53b1b3
2 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
internal/msggateway/init.go
View File

@ -24,6 +24,9 @@ func RunWsAndServer(rpcPort, wsPort, prometheusPort int) error {
} }
hubServer := NewServer(rpcPort) hubServer := NewServer(rpcPort)
go hubServer.Start() go hubServer.Start()
if hubServer.Notification() == nil {
panic("notification is nil")
}
longServer.SetMessageHandler(hubServer.Notification()) longServer.SetMessageHandler(hubServer.Notification())
hubServer.SetLongConnServer(longServer) hubServer.SetLongConnServer(longServer)
go hubServer.LongConnServer.Run() go hubServer.LongConnServer.Run()

2
internal/msggateway/n_ws_server.go
View File

@ -3,6 +3,7 @@ package msggateway
import ( import (
"errors" "errors"
"fmt" "fmt"
"github.com/OpenIMSDK/Open-IM-Server/pkg/common/log"
"github.com/OpenIMSDK/Open-IM-Server/pkg/common/tokenverify" "github.com/OpenIMSDK/Open-IM-Server/pkg/common/tokenverify"
"github.com/OpenIMSDK/Open-IM-Server/pkg/errs" "github.com/OpenIMSDK/Open-IM-Server/pkg/errs"
"github.com/OpenIMSDK/Open-IM-Server/pkg/rpcclient/notification" "github.com/OpenIMSDK/Open-IM-Server/pkg/rpcclient/notification"
@ -51,6 +52,7 @@ type WsServer struct {
} }
func (ws *WsServer) SetMessageHandler(rpcClient *notification.Check) { func (ws *WsServer) SetMessageHandler(rpcClient *notification.Check) {
log.ZDebug(nil, "msggateway SetMessageHandler", "rpcClient", rpcClient)
ws.MessageHandler = NewGrpcHandler(ws.validate, rpcClient) ws.MessageHandler = NewGrpcHandler(ws.validate, rpcClient)
} }