fix: close

zetxx · Dec 21, 2023 · 6b31bcd · 6b31bcd
1 parent 7a20818
commit 6b31bcd
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/lib/mssql/index.js b/lib/mssql/index.js
@@ -56,7 +56,8 @@ const Link = async(config) => {
                 throw e;
             }
         },
-        request: () => cPool.request()
+        request: () => cPool.request(),
+        close: () => cPool.close()
     };
 };
 
@@ -296,5 +297,10 @@ module.exports = async(config) => {
     const tt = await tableTypes();
     const methods = await build(tt);
 
-    return methods;
+    return {
+        methods,
+        async stop() {
+            return await link.close();
+        }
+    };
 };
diff --git a/lib/postgres/index.js b/lib/postgres/index.js
@@ -126,6 +126,9 @@ module.exports = async(config) => {
 
     return {
         methods,
+        async stop() {
+            return await link.end();
+        },
         async txBegin() {
             const id = ++txId;
             const client = await link.connect();