diff --git a/internal/msggateway/client.go b/internal/msggateway/client.go index 7e1ba6405..161300d6d 100644 --- a/internal/msggateway/client.go +++ b/internal/msggateway/client.go @@ -18,7 +18,6 @@ import ( "context" "encoding/json" "fmt" - "runtime/debug" "sync" "sync/atomic" "time" @@ -132,7 +131,7 @@ func (c *Client) readMessage() { defer func() { if r := recover(); r != nil { c.closedErr = ErrPanic - fmt.Println("socket have panic err:", r, string(debug.Stack())) + log.ZPanic(c.ctx, "socket have panic err:", r) } c.close() }() diff --git a/internal/msggateway/message_handler.go b/internal/msggateway/message_handler.go index dafa4cabe..d88d2fbfd 100644 --- a/internal/msggateway/message_handler.go +++ b/internal/msggateway/message_handler.go @@ -115,8 +115,8 @@ type GrpcHandler struct { validate *validator.Validate } -func NewGrpcHandler() *GrpcHandler { - return &GrpcHandler{} +func NewGrpcHandler(validate *validator.Validate) *GrpcHandler { + return &GrpcHandler{validate: validate} } func (g GrpcHandler) GetSeq(ctx context.Context, data *Req) ([]byte, error) { diff --git a/internal/msggateway/ws_server.go b/internal/msggateway/ws_server.go index 86a9aba71..7271c3727 100644 --- a/internal/msggateway/ws_server.go +++ b/internal/msggateway/ws_server.go @@ -70,7 +70,7 @@ type kickHandler struct { } func (ws *WsServer) SetDiscoveryRegistry(disCov discovery.SvcDiscoveryRegistry, config *Config) { - ws.MessageHandler = NewGrpcHandler() + ws.MessageHandler = NewGrpcHandler(ws.validate) ws.disCov = disCov }