diff --git a/internal/msggateway/init.go b/internal/msggateway/init.go index 24af9d433..808abf817 100644 --- a/internal/msggateway/init.go +++ b/internal/msggateway/init.go @@ -24,6 +24,9 @@ func RunWsAndServer(rpcPort, wsPort, prometheusPort int) error { } hubServer := NewServer(rpcPort) go hubServer.Start() + if hubServer.Notification() == nil { + panic("notification is nil") + } longServer.SetMessageHandler(hubServer.Notification()) hubServer.SetLongConnServer(longServer) go hubServer.LongConnServer.Run() diff --git a/internal/msggateway/n_ws_server.go b/internal/msggateway/n_ws_server.go index d0964b3fd..24bfa213b 100644 --- a/internal/msggateway/n_ws_server.go +++ b/internal/msggateway/n_ws_server.go @@ -3,6 +3,7 @@ package msggateway import ( "errors" "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/errs" "github.com/OpenIMSDK/Open-IM-Server/pkg/rpcclient/notification" @@ -51,6 +52,7 @@ type WsServer struct { } func (ws *WsServer) SetMessageHandler(rpcClient *notification.Check) { + log.ZDebug(nil, "msggateway SetMessageHandler", "rpcClient", rpcClient) ws.MessageHandler = NewGrpcHandler(ws.validate, rpcClient) }