pull/2969/head
icey-yu 9 months ago
parent fc6619497c
commit 6952c4b1b3

@ -18,7 +18,6 @@ import (
"context" "context"
"encoding/json" "encoding/json"
"fmt" "fmt"
"runtime/debug"
"sync" "sync"
"sync/atomic" "sync/atomic"
"time" "time"
@ -132,7 +131,7 @@ func (c *Client) readMessage() {
defer func() { defer func() {
if r := recover(); r != nil { if r := recover(); r != nil {
c.closedErr = ErrPanic c.closedErr = ErrPanic
fmt.Println("socket have panic err:", r, string(debug.Stack())) log.ZPanic(c.ctx, "socket have panic err:", r)
} }
c.close() c.close()
}() }()

@ -115,8 +115,8 @@ type GrpcHandler struct {
validate *validator.Validate validate *validator.Validate
} }
func NewGrpcHandler() *GrpcHandler { func NewGrpcHandler(validate *validator.Validate) *GrpcHandler {
return &GrpcHandler{} return &GrpcHandler{validate: validate}
} }
func (g GrpcHandler) GetSeq(ctx context.Context, data *Req) ([]byte, error) { func (g GrpcHandler) GetSeq(ctx context.Context, data *Req) ([]byte, error) {

@ -70,7 +70,7 @@ type kickHandler struct {
} }
func (ws *WsServer) SetDiscoveryRegistry(disCov discovery.SvcDiscoveryRegistry, config *Config) { func (ws *WsServer) SetDiscoveryRegistry(disCov discovery.SvcDiscoveryRegistry, config *Config) {
ws.MessageHandler = NewGrpcHandler() ws.MessageHandler = NewGrpcHandler(ws.validate)
ws.disCov = disCov ws.disCov = disCov
} }

Loading…
Cancel
Save