blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-04-06 04:15:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

chain unary interceptor

Browse Source
This commit is contained in:
withchao 2023-07-06 11:23:57 +08:00
parent 28fad04d5c
commit ff22ea7108
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cmd/api/main.go
View File

@ -57,7 +57,7 @@ func run(port int) error {
if err != nil { if err != nil {
return err return err
} }
if client.CreateRpcRootNodes(config.GetServiceNames()); err != nil { if err := client.CreateRpcRootNodes(config.GetServiceNames()); err != nil {
return err return err
} }
fmt.Println("api init discov client success") fmt.Println("api init discov client success")

2
pkg/common/mw/rpc_client_interceptor.go
View File

@ -16,7 +16,7 @@ import (
) )
func GrpcClient() grpc.DialOption { func GrpcClient() grpc.DialOption {
return grpc.WithUnaryInterceptor(RpcClientInterceptor) return grpc.WithChainUnaryInterceptor(RpcClientInterceptor)
} }
func RpcClientInterceptor(ctx context.Context, method string, req, resp interface{}, cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts ...grpc.CallOption) (err error) { func RpcClientInterceptor(ctx context.Context, method string, req, resp interface{}, cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts ...grpc.CallOption) (err error) {

2
pkg/common/mw/rpc_server_interceptor.go
View File

@ -153,5 +153,5 @@ func RpcServerInterceptor(ctx context.Context, req interface{}, info *grpc.Unary
} }
func GrpcServer() grpc.ServerOption { func GrpcServer() grpc.ServerOption {
return grpc.UnaryInterceptor(RpcServerInterceptor) return grpc.ChainUnaryInterceptor(RpcServerInterceptor)
} }