mmorpg/service-common
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat 拦截器取usn

Browse Source
This commit is contained in:
DESKTOP-V763RJ7\Administrator
2026-01-14 16:00:26 +08:00
parent 53d4e4513e
commit a2025e8901
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
net/grpc/service/service.go
View File

@@ -5,6 +5,7 @@ import (
"fmt" "fmt"
"git.hlsq.asia/mmorpg/service-common/discover" "git.hlsq.asia/mmorpg/service-common/discover"
"git.hlsq.asia/mmorpg/service-common/log" "git.hlsq.asia/mmorpg/service-common/log"
"git.hlsq.asia/mmorpg/service-common/net/http/http_resp"
"git.hlsq.asia/mmorpg/service-common/utils" "git.hlsq.asia/mmorpg/service-common/utils"
"google.golang.org/grpc" "google.golang.org/grpc"
"google.golang.org/grpc/codes" "google.golang.org/grpc/codes"
@@ -56,8 +57,13 @@ func (s *Base) Init(addr string, port int32) {
err = status.Error(codes.Internal, fmt.Sprintf("%v", r)) err = status.Error(codes.Internal, fmt.Sprintf("%v", r))
} }
}() }()
resp, err = handler(ctx, req) if usnReq, ok := req.(interface{ GetUSN() string }); ok {
return usn := usnReq.GetUSN()
if !utils.ShouldBindUsn(ctx, &usn) {
return nil, http_resp.ParamError
}
}
return handler(ctx, req)
}, },
), ),
grpc.KeepaliveEnforcementPolicy(keepalive.EnforcementPolicy{ grpc.KeepaliveEnforcementPolicy(keepalive.EnforcementPolicy{