IM/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2026-05-04 09:05:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'errcode' of github.com:OpenIMSDK/Open-IM-Server into errcode

Browse Source
This commit is contained in:
wangchuxiao
2023-03-09 14:43:28 +08:00
parent 10cf431ccf 7776db83ea
commit 3bcc696826
7 changed files with 98 additions and 141 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/common/mw/rpc.go
+5 -2
View File
@@ -15,6 +15,9 @@ import (
"runtime/debug"
)
const OperationID = "operationID"
const OpUserID = "opUserID"
func rpcServerInterceptor(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (resp interface{}, err error) {
var operationID string
defer func() {
@@ -27,13 +30,13 @@ func rpcServerInterceptor(ctx context.Context, req interface{}, info *grpc.Unary
if !ok {
return nil, status.New(codes.InvalidArgument, "missing metadata").Err()
}
if opts := md.Get(constant.OperationID); len(opts) != 1 || opts[0] == "" {
if opts := md.Get(OperationID); len(opts) != 1 || opts[0] == "" {
return nil, status.New(codes.InvalidArgument, "operationID error").Err()
} else {
operationID = opts[0]
}
var opUserID string
if opts := md.Get(constant.OpUserID); len(opts) == 1 {
if opts := md.Get(OpUserID); len(opts) == 1 {
opUserID = opts[0]
}
ctx = tracelog.NewRpcCtx(ctx, funcName, operationID)