From 2c99f849541c465cabd91986efc3342d77dd79a7 Mon Sep 17 00:00:00 2001
From: Vladislav Polyakov <polRk@ydb.tech>
Date: Wed, 30 Oct 2024 22:00:20 +0300
Subject: [PATCH] feat: add support for poolId parameter in query service
 requests

Signed-off-by: Vladislav Polyakov <polRk@ydb.tech>
---
 src/query/query-session-execute.ts | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/query/query-session-execute.ts b/src/query/query-session-execute.ts
index db0ccb0b..7df06cd8 100644
--- a/src/query/query-session-execute.ts
+++ b/src/query/query-session-execute.ts
@@ -47,6 +47,8 @@ export type IExecuteArgs = {
      */
     rowMode?: RowType,
     idempotent?: boolean,
+
+    poolId?: string,
 };
 
 export type IExecuteResult = {
@@ -107,6 +109,7 @@ export function execute(this: QuerySession, args: IExecuteArgs): Promise<IExecut
             syntax: args.syntax ?? Ydb.Query.Syntax.SYNTAX_YQL_V1,
         },
         execMode: args.execMode ?? Ydb.Query.ExecMode.EXEC_MODE_EXECUTE,
+        poolId: args.poolId,
     };
     if (args.statsMode) executeQueryRequest.statsMode = args.statsMode;
     if (args.parameters) executeQueryRequest.parameters = args.parameters;