From 12e118e371003bb9082227688a9d6b428e2d650a Mon Sep 17 00:00:00 2001
From: Ivan Zuev <i-zuev@yandex-team.ru>
Date: Wed, 12 Jan 2022 14:33:06 +0300
Subject: [PATCH] feat: allow to override endpoint in service factory

---
 src/session.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/session.ts b/src/session.ts
index 6102c020..8e94fdfb 100644
--- a/src/session.ts
+++ b/src/session.ts
@@ -94,8 +94,8 @@ export class Session {
         return this.config.pollInterval;
     }
 
-    client<S extends ServiceDefinition>(clientClass: GeneratedServiceClientCtor<S>): WrappedServiceClientType<S> {
-        const endpoint = getServiceClientEndpoint(clientClass);
+    client<S extends ServiceDefinition>(clientClass: GeneratedServiceClientCtor<S>, customEndpoint?: string): WrappedServiceClientType<S> {
+        const endpoint = customEndpoint || getServiceClientEndpoint(clientClass);
         const channel = createChannel(endpoint, this.channelCredentials);
 
         return clientFactory.create(clientClass.service, channel);