Fix querysession pool autoresize

alex268 · alex268 · commit eed4122bf1b0 · 2024-09-25T19:29:50.000+01:00
diff --git a/jdbc/src/main/java/tech/ydb/jdbc/context/YdbContext.java b/jdbc/src/main/java/tech/ydb/jdbc/context/YdbContext.java
@@ -219,6 +219,7 @@ public void register() {
             int newSize = maxSize + SESSION_POOL_RESIZE_STEP;
             if (maxSize == tableClient.sessionPoolStats().getMaxSize()) {
                 tableClient.updatePoolMaxSize(newSize);
+                queryClient.updatePoolMaxSize(newSize);
             }
         }
     }
@@ -231,6 +232,7 @@ public void deregister() {
                 int newSize = maxSize - SESSION_POOL_RESIZE_STEP;
                 if (maxSize == tableClient.sessionPoolStats().getMaxSize()) {
                     tableClient.updatePoolMaxSize(newSize);
+                    queryClient.updatePoolMaxSize(newSize);
                 }
             }
         }

Original file line number	Diff line number	Diff line change
`@@ -219,6 +219,7 @@ public void register() {`
`219`	`219`	`int newSize = maxSize + SESSION_POOL_RESIZE_STEP;`
`220`	`220`	`if (maxSize == tableClient.sessionPoolStats().getMaxSize()) {`
`221`	`221`	`tableClient.updatePoolMaxSize(newSize);`
	`222`	`+ queryClient.updatePoolMaxSize(newSize);`
`222`	`223`	`}`
`223`	`224`	`}`
`224`	`225`	`}`
`@@ -231,6 +232,7 @@ public void deregister() {`
`231`	`232`	`int newSize = maxSize - SESSION_POOL_RESIZE_STEP;`
`232`	`233`	`if (maxSize == tableClient.sessionPoolStats().getMaxSize()) {`
`233`	`234`	`tableClient.updatePoolMaxSize(newSize);`
	`235`	`+ queryClient.updatePoolMaxSize(newSize);`
`234`	`236`	`}`
`235`	`237`	`}`
`236`	`238`	`}`