diff --git a/pkg/rgrpc/todo/pool/pool.go b/pkg/rgrpc/todo/pool/pool.go index c6bf42c2c9..702eceff32 100644 --- a/pkg/rgrpc/todo/pool/pool.go +++ b/pkg/rgrpc/todo/pool/pool.go @@ -97,6 +97,13 @@ func NewConn(options Options) (*grpc.ClientConn, error) { return conn, nil } +func RegisterGatewayServiceClient(client gateway.GatewayAPIClient, endpoint string) { + gatewayProviders.m.Lock() + defer gatewayProviders.m.Unlock() + + gatewayProviders.conn[endpoint] = client +} + // GetGatewayServiceClient returns a GatewayServiceClient. func GetGatewayServiceClient(opts ...Option) (gateway.GatewayAPIClient, error) { gatewayProviders.m.Lock()