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

fix: err

Browse Source
This commit is contained in:
icey-yu 2024-12-17 10:16:54 +08:00
parent fc6619497c
commit 6952c4b1b3
3 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
internal/msggateway/client.go
View File

@ -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()
}() }()

4
internal/msggateway/message_handler.go
View File

@ -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) {

2
internal/msggateway/ws_server.go
View File

@ -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
} }