snowflake-kysely-dialect 0.1.0 → 0.1.1

package/dist/SnowflakeDriver.d.ts

@@ -10,5 +10,6 @@ export declare class SnowflakeDriver implements Driver {
     commitTransaction(conn: SnowflakeConnection): Promise<void>;
     rollbackTransaction(conn: SnowflakeConnection): Promise<void>;
     releaseConnection(conn: SnowflakeConnection): Promise<void>;
+    destroyConnection(conn: SnowflakeConnection): Promise<void>;
     destroy(): Promise<void>;
 }

package/dist/SnowflakeDriver.js

@@ -16,7 +16,8 @@ const DEFAULT_POOL_OPTIONS = {
 class SnowflakeDriver {
     #config;
     #pool = null;
-    #pendingReleases = new Map();
+    // Tracks the underlying SDK connection for each wrapper so we can release or destroy it.
+    #resources = new Map();
     constructor(config) {
         this.#config = config;
     }
@@ -24,15 +25,10 @@ class SnowflakeDriver {
         this.#pool = snowflake_sdk_1.default.createPool(this.#config.connection, { ...DEFAULT_POOL_OPTIONS, ...this.#config.poolOptions });
     }
     async acquireConnection() {
-        return new Promise((resolveAcquire, rejectAcquire) => {
-            this.#pool.use(async (sdkConn) => {
-                const conn = new SnowflakeConnection_js_1.SnowflakeConnection(sdkConn);
-                return new Promise((resolveRelease) => {
-                    this.#pendingReleases.set(conn, resolveRelease);
-                    resolveAcquire(conn);
-                });
-            }).catch(rejectAcquire);
-        });
+        const sdkConn = await this.#pool.acquire();
+        const conn = new SnowflakeConnection_js_1.SnowflakeConnection(sdkConn);
+        this.#resources.set(conn, sdkConn);
+        return conn;
     }
     async beginTransaction(conn) {
         return conn.beginTransaction();
@@ -44,10 +40,17 @@ class SnowflakeDriver {
         return conn.rollbackTransaction();
     }
     async releaseConnection(conn) {
-        const release = this.#pendingReleases.get(conn);
-        if (release) {
-            this.#pendingReleases.delete(conn);
-            release();
+        const sdkConn = this.#resources.get(conn);
+        if (sdkConn) {
+            this.#resources.delete(conn);
+            await this.#pool.release(sdkConn);
+        }
+    }
+    async destroyConnection(conn) {
+        const sdkConn = this.#resources.get(conn);
+        if (sdkConn) {
+            this.#resources.delete(conn);
+            await this.#pool.destroy(sdkConn);
         }
     }
     async destroy() {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "snowflake-kysely-dialect",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "description": "Snowflake dialect for Kysely",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",