fix

vesoft-inc · May 24, 2024 · 8969e7b · 8969e7b
1 parent 222da73
commit 8969e7b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/nebula3/gclient/net/Session.py b/nebula3/gclient/net/Session.py
@@ -297,7 +297,6 @@ def release(self):
             return
         self._connection.signout(self._session_id)
         self._connection.is_used = False
-        self._connection.close()
         self._connection = None
 
     def ping(self):

diff --git a/nebula3/gclient/net/SessionPool.py b/nebula3/gclient/net/SessionPool.py
@@ -425,9 +425,11 @@ def _new_session(self):
                     resp = session.execute("USE {}".format(self._space_name))
                 except Exception:
                     session.release()
+                    connection.close()
                     raise
                 if not resp.is_succeeded():
                     session.release()
+                    connection.close()
                     raise RuntimeError(
                         "Failed to get session, cannot set the session space to {} error: {} {}".format(
                             self._space_name, resp.error_code(), resp.error_msg()