fix ta client logic to avoid crash

1 files changed, 8 insertions, 2 deletions

diff --git a/ydb/library/yql/providers/common/token_accessor/client/token_accessor_client.cpp b/ydb/library/yql/providers/common/token_accessor/client/token_accessor_client.cpp
index f29acf6fa3..cd5150afa3 100644
--- a/ydb/library/yql/providers/common/token_accessor/client/token_accessor_client.cpp
+++ b/ydb/library/yql/providers/common/token_accessor/client/token_accessor_client.cpp
@@ -46,8 +46,13 @@ private:
             RequestInflight = true;
             auto resultPromise = NThreading::NewPromise();
 
+            auto context = Client->CreateContext();
+            if (!context) {
+                throw yexception() << "Client is being shutted down";
+            }
             std::weak_ptr<const TImpl> weakSelf = shared_from_this();
-            auto cb = [weakSelf, resultPromise, sync](NGrpc::TGrpcStatus&& status, GetTokenResponse&& result) mutable {
+            // hold context until reply
+            auto cb = [weakSelf, resultPromise, sync, context](NGrpc::TGrpcStatus&& status, GetTokenResponse&& result) mutable {
                 if (auto self = weakSelf.lock()) {
                     self->ProcessResponse(std::move(status), std::move(result), sync);
                 }
@@ -66,7 +71,8 @@ private:
                     &TokenAccessorService::Stub::AsyncGetToken,
                     {
                         {}, {}, RequestTimeout
-                    }
+                    },
+                    context.get()
                 );
             }
             if (sync) {