Open access to QueryClient pool stats

ydb-platform · Sep 25, 2024 · a981125 · a981125
1 parent fef5bde
commit a981125
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/query/src/main/java/tech/ydb/query/impl/QueryClientImpl.java b/query/src/main/java/tech/ydb/query/impl/QueryClientImpl.java
@@ -11,6 +11,7 @@
 import tech.ydb.core.grpc.GrpcTransport;
 import tech.ydb.query.QueryClient;
 import tech.ydb.query.QuerySession;
+import tech.ydb.table.SessionPoolStats;
 
 /**
  *
@@ -46,8 +47,8 @@ public void updatePoolMaxSize(int maxSize) {
         pool.updateMaxSize(maxSize);
     }
 
-    SessionPool getSessionPool() {
-        return pool;
+    public SessionPoolStats getSessionPoolStats() {
+        return pool.getStats();
     }
 
     @Override

diff --git a/query/src/main/java/tech/ydb/query/impl/SessionPool.java b/query/src/main/java/tech/ydb/query/impl/SessionPool.java
@@ -250,6 +250,8 @@ public void destroy() {
 
         @Override
         public void close() {
+            logger.trace("QuerySession[{}] closed with broke status {}", getId(), isBroken);
+
             stats.released.increment();
             if (isBroken || isStopped) {
                 queue.delete(this);

diff --git a/query/src/main/java/tech/ydb/query/impl/TableClientImpl.java b/query/src/main/java/tech/ydb/query/impl/TableClientImpl.java
@@ -49,7 +49,7 @@ public ScheduledExecutorService getScheduler() {
 
     @Override
     public SessionPoolStats sessionPoolStats() {
-        return proxy.getSessionPool().getStats();
+        return proxy.getSessionPoolStats();
     }
 
     @Override