Merge pull request #353 from gridnevvvit/gridnevvvit-patch-1 Fixed access to connection if connection cannot be found by node id

Author: rekby

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
 

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:

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: