add zrpc client interceptor

This commit is contained in:
kevin
2020-09-29 16:09:11 +08:00
parent b3ead4d76c
commit dbca20e3df
5 changed files with 42 additions and 27 deletions

View File

@@ -17,6 +17,7 @@ var (
type (
Client interface {
AddInterceptor(interceptor grpc.UnaryClientInterceptor)
Conn() *grpc.ClientConn
}
@@ -78,6 +79,10 @@ func NewClientWithTarget(target string, opts ...internal.ClientOption) (Client,
return internal.NewClient(target, opts...)
}
func (rc *RpcClient) AddInterceptor(interceptor grpc.UnaryClientInterceptor) {
rc.client.AddInterceptor(interceptor)
}
func (rc *RpcClient) Conn() *grpc.ClientConn {
return rc.client.Conn()
}