blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-05-17 01:19:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

1

Browse Source
This commit is contained in:
withchao 2023-01-12 16:17:52 +08:00
parent 16864dab7d
commit d06dc3e4b5
1 changed files with 16 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
pkg/common/middleware/rpc.go
View File

@ -4,6 +4,7 @@ import (
"Open_IM/pkg/common/constant"
"Open_IM/pkg/common/log"
"Open_IM/pkg/common/trace_log"
"Open_IM/pkg/utils"
"context"
"fmt"
"google.golang.org/grpc"
@ -66,19 +67,18 @@ func rpcString(v interface{}) string {
return fmt.Sprintf("%+v", v)
}
//func RpcClientInterceptor(ctx context.Context, method string, req, reply interface{}, cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts ...grpc.CallOption) (err error) {
// //if cc == nil {
// // return utils.Wrap(constant.ErrRpcConn, "")
// //}
// operationID, ok := ctx.Value("operationID").(string)
// if !ok {
// return utils.Wrap(constant.ErrArgs, "ctx missing operationID")
// }
// opUserID, ok := ctx.Value("opUserID").(string)
// if !ok {
// return utils.Wrap(constant.ErrArgs, "ctx missing opUserID")
// }
// md := metadata.Pairs("operationID", operationID, "opUserID", opUserID)
// return invoker(metadata.NewOutgoingContext(ctx, md), method, req, reply, cc, opts...)
//}
//
func RpcClientInterceptor(ctx context.Context, method string, req, reply interface{}, cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts ...grpc.CallOption) (err error) {
//if cc == nil {
// return utils.Wrap(constant.ErrRpcConn, "")
//}
operationID, ok := ctx.Value("operationID").(string)
if !ok {
return utils.Wrap(constant.ErrArgs, "ctx missing operationID")
}
opUserID, ok := ctx.Value("opUserID").(string)
if !ok {
return utils.Wrap(constant.ErrArgs, "ctx missing opUserID")
}
md := metadata.Pairs("operationID", operationID, "opUserID", opUserID)
return invoker(metadata.NewOutgoingContext(ctx, md), method, req, reply, cc, opts...)
}