Merge pull request #353 from gridnevvvit/gridnevvvit-patch-1 Fixed ac…

…cess to connection if connection cannot be found by node id
ydb-platform · Sep 21, 2023 · 8938db2 · 8938db2
2 parents f90cbcc + 6709a63
commit 8938db2
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,3 +1,6 @@
+## 3.5.1 ##
+* Fixed access to connection if connection cannot be found by node id
+
 ## 3.5.0 ##
 * Added functions for check column table type table of any type of scheme entry
 

diff --git a/ydb/aio/pool.py b/ydb/aio/pool.py
@@ -34,7 +34,7 @@ async def get(self, preferred_endpoint=None, fast_fail=False, wait_timeout=10):
             return self.connections_by_node_id[preferred_endpoint.node_id]
 
         if preferred_endpoint is not None and preferred_endpoint.endpoint in self.connections:
-            return self.connections[preferred_endpoint]
+            return self.connections[preferred_endpoint.endpoint]
 
         for conn_lst in self.conn_lst_order:
             try:

diff --git a/ydb/pool.py b/ydb/pool.py
@@ -129,7 +129,7 @@ def get(self, preferred_endpoint=None) -> Connection:
                 return self.connections_by_node_id[preferred_endpoint.node_id]
 
             if preferred_endpoint is not None and preferred_endpoint.endpoint in self.connections:
-                return self.connections[preferred_endpoint]
+                return self.connections[preferred_endpoint.endpoint]
 
             for conn_lst in self.conn_lst_order:
                 try: