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)