fix: connection.close()

Author: hantmac

databend_sqlalchemy/connector.py

@@ -102,7 +102,7 @@ def __init__(self, db_url='http://root:@localhost:8081'):
         self.client = Client.from_url(db_url)
 
     def close(self):
-        pass
+        self.client.disconnect()
 
     def commit(self):
         pass
@@ -166,7 +166,7 @@ def description(self):
         return self._columns
 
     def close(self):
-        pass
+        self._reset_state()
 
     def execute(self, operation, parameters=None, is_response=True):
         """Prepare and execute a database operation (query or command). """

databend_sqlalchemy/test.py

@@ -2,14 +2,14 @@
 
 
 def test():
-    cursor = connector.connect('http://root:@localhost:8081').cursor()
-    cursor.execute("select null as db, name as name, database as schema, if(engine = 'VIEW', 'view', 'table') as type from system.tables where database = 'default';")
+    conn = connector.connect("http://root:[email protected]:8000/default")
+    cursor = conn.cursor()
+    cursor.execute(
+        "select null as db, name as name, database as schema, if(engine = 'VIEW', 'view', 'table') as type from system.tables where database = 'default';")
     # print(cursor.fetchone())
     print(cursor.fetchall())
     print(cursor.description)
-
-    # for i in cursor.next():
-    #     print(i)
+    conn.close()
 
 
 if __name__ == '__main__':