From bc831b75ddef1286ab82ff83eb9e32fe06c4ebc9 Mon Sep 17 00:00:00 2001 From: kevin Date: Sat, 26 Sep 2020 22:05:57 +0800 Subject: [PATCH] export AddOptions, AddStreamInterceptors, AddUnaryInterceptors --- zrpc/server.go | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/zrpc/server.go b/zrpc/server.go index 4de020daf..90c8150ea 100644 --- a/zrpc/server.go +++ b/zrpc/server.go @@ -13,6 +13,7 @@ import ( "github.com/tal-tech/go-zero/zrpc/internal" "github.com/tal-tech/go-zero/zrpc/internal/auth" "github.com/tal-tech/go-zero/zrpc/internal/serverinterceptors" + "google.golang.org/grpc" ) const envPodIp = "POD_IP" @@ -65,6 +66,18 @@ func NewServer(c RpcServerConf, register internal.RegisterFn) (*RpcServer, error return rpcServer, nil } +func (rs *RpcServer) AddOptions(options ...grpc.ServerOption) { + rs.AddOptions(options...) +} + +func (rs *RpcServer) AddStreamInterceptors(interceptors ...grpc.StreamServerInterceptor) { + rs.AddStreamInterceptors(interceptors...) +} + +func (rs *RpcServer) AddUnaryInterceptors(interceptors ...grpc.UnaryServerInterceptor) { + rs.AddUnaryInterceptors(interceptors...) +} + func (rs *RpcServer) Start() { if err := rs.server.Start(rs.register); err != nil { logx.Error(err)