Reset g_session and g_session_path, cleanup on closing :memory:

Author: auxten

chdb/session/state.py

@@ -51,7 +51,7 @@ def __init__(self, path=None):
             )
             g_session.close()
             g_session_path = None
-        if path is None:
+        if path is None or ":memory:" in path:
             self._cleanup = True
             self._path = tempfile.mkdtemp()
         else:
@@ -89,13 +89,19 @@ def close(self):
         if self._conn is not None:
             self._conn.close()
             self._conn = None
+        global g_session, g_session_path
+        g_session = None
+        g_session_path = None
 
     def cleanup(self):
         try:
             if self._conn is not None:
                 self._conn.close()
                 self._conn = None
             shutil.rmtree(self._path)
+            global g_session, g_session_path
+            g_session = None
+            g_session_path = None
         except:  # noqa
             pass