Add close function

Author: cevian

nbs/00_vector.ipynb

@@ -383,6 +383,10 @@
     "\n",
     "            self.pool = await asyncpg.create_pool(dsn=self.service_url, init=init)\n",
     "        return self.pool.acquire()\n",
+    "    \n",
+    "    async def close(self):\n",
+    "        if self.pool != None:\n",
+    "            await self.pool.close()\n",
     "\n",
     "    async def table_is_empty(self):\n",
     "        \"\"\"\n",
@@ -783,7 +787,8 @@
     "await vec.delete_all()\n",
     "assert await vec.table_is_empty()\n",
     "\n",
-    "await vec.drop_table()"
+    "await vec.drop_table()\n",
+    "await vec.close()"
    ]
   },
   {
@@ -848,6 +853,10 @@
     "        finally:            \n",
     "            self.pool.putconn(connection)\n",
     "\n",
+    "    def close(self):\n",
+    "        if self.pool != None:\n",
+    "            self.pool.closeall()\n",
+    "\n",
     "    def _translate_to_pyformat(self, query_string, params):\n",
     "        \"\"\"\n",
     "        Translates dollar sign number parameters and list parameters to pyformat strings.\n",
@@ -1327,7 +1336,9 @@
     "vec.delete_all()\n",
     "assert vec.table_is_empty()\n",
     "\n",
-    "vec.drop_table()"
+    "vec.drop_table()\n",
+    "\n",
+    "vec.close()"
    ]
   },
   {

timescale_vector/_modidx.py

@@ -12,6 +12,7 @@
                                                                                                          'timescale_vector/client.py'),
                                          'timescale_vector.client.Async._get_approx_count': ( 'vector.html#async._get_approx_count',
                                                                                               'timescale_vector/client.py'),
+                                         'timescale_vector.client.Async.close': ('vector.html#async.close', 'timescale_vector/client.py'),
                                          'timescale_vector.client.Async.connect': ( 'vector.html#async.connect',
                                                                                     'timescale_vector/client.py'),
                                          'timescale_vector.client.Async.create_ivfflat_index': ( 'vector.html#async.create_ivfflat_index',
@@ -72,6 +73,7 @@
                                                                                              'timescale_vector/client.py'),
                                          'timescale_vector.client.Sync._translate_to_pyformat': ( 'vector.html#sync._translate_to_pyformat',
                                                                                                   'timescale_vector/client.py'),
+                                         'timescale_vector.client.Sync.close': ('vector.html#sync.close', 'timescale_vector/client.py'),
                                          'timescale_vector.client.Sync.connect': ('vector.html#sync.connect', 'timescale_vector/client.py'),
                                          'timescale_vector.client.Sync.create_ivfflat_index': ( 'vector.html#sync.create_ivfflat_index',
                                                                                                 'timescale_vector/client.py'),

timescale_vector/client.py

@@ -248,6 +248,10 @@ async def init(conn):
 
             self.pool = await asyncpg.create_pool(dsn=self.service_url, init=init)
         return self.pool.acquire()
+    
+    async def close(self):
+        if self.pool != None:
+            await self.pool.close()
 
     async def table_is_empty(self):
         """
@@ -429,6 +433,10 @@ def connect(self):
         finally:            
             self.pool.putconn(connection)
 
+    def close(self):
+        if self.pool != None:
+            self.pool.closeall()
+
     def _translate_to_pyformat(self, query_string, params):
         """
         Translates dollar sign number parameters and list parameters to pyformat strings.