From 6952c4b1b31074db43b4b82a844be78a7169e4f7 Mon Sep 17 00:00:00 2001
From: icey-yu <1186114839@qq.com>
Date: Tue, 17 Dec 2024 10:16:54 +0800
Subject: [PATCH] fix: err

---
 internal/msggateway/client.go          | 3 +--
 internal/msggateway/message_handler.go | 4 ++--
 internal/msggateway/ws_server.go       | 2 +-
 3 files changed, 4 insertions(+), 5 deletions(-)

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
 }