blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-05-17 09:44:47 +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

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

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