@agentforge/core 0.18.0 → 0.18.1

package/dist/index.cjs

@@ -4244,6 +4244,23 @@ function createInitialPoolStats() {
     healthChecksFailed: 0
   };
 }
+function checkoutConnection(runtime, pooled) {
+  const previousLastUsedAt = pooled.lastUsedAt;
+  pooled.inUse = true;
+  pooled.lastUsedAt = Date.now();
+  runtime.stats.available--;
+  runtime.stats.acquired++;
+  try {
+    runtime.options.onAcquire?.(pooled.connection);
+    return pooled.connection;
+  } catch (error) {
+    pooled.inUse = false;
+    pooled.lastUsedAt = previousLastUsedAt;
+    runtime.stats.available++;
+    runtime.stats.acquired--;
+    throw error;
+  }
+}
 async function initializeConnections(runtime, count) {
   runtime.creating += count;
   const promises = Array.from({ length: count }, async () => {
@@ -4265,23 +4282,22 @@ async function initializeConnections(runtime, count) {
   await Promise.all(promises);
 }
 function assignConnectionToPendingAcquire(runtime, connection) {
-  const pending = runtime.pending.shift();
-  if (!pending) {
-    return;
-  }
   const pooled = runtime.connections.find((item) => item.connection === connection);
   if (!pooled) {
-    runtime.pending.unshift(pending);
     return;
   }
-  clearTimeout(pending.timeout);
-  runtime.stats.pending--;
-  pooled.inUse = true;
-  pooled.lastUsedAt = Date.now();
-  runtime.stats.available--;
-  runtime.stats.acquired++;
-  runtime.options.onAcquire?.(connection);
-  pending.resolve(connection);
+  let pending = runtime.pending.shift();
+  while (pending) {
+    clearTimeout(pending.timeout);
+    runtime.stats.pending--;
+    try {
+      pending.resolve(checkoutConnection(runtime, pooled));
+      return;
+    } catch (error) {
+      pending.reject(error instanceof Error ? error : new Error(String(error)));
+      pending = runtime.pending.shift();
+    }
+  }
 }
 async function createConnection(runtime) {
   const connection = await runtime.options.factory();
@@ -4297,22 +4313,30 @@ async function createConnection(runtime) {
   runtime.stats.available++;
   return connection;
 }
+function finalizeDestroyedConnection(runtime, pooled) {
+  runtime.stats.destroyed++;
+  runtime.stats.size--;
+  if (pooled.inUse) {
+    runtime.stats.acquired--;
+  } else {
+    runtime.stats.available--;
+  }
+}
 async function destroyConnection(runtime, pooled) {
   const index = runtime.connections.indexOf(pooled);
   if (index !== -1) {
     runtime.connections.splice(index, 1);
   }
   if (runtime.options.destroyer) {
-    await runtime.options.destroyer(pooled.connection);
+    try {
+      await runtime.options.destroyer(pooled.connection);
+    } catch (error) {
+      finalizeDestroyedConnection(runtime, pooled);
+      throw error;
+    }
   }
   runtime.options.onDestroy?.(pooled.connection);
-  runtime.stats.destroyed++;
-  runtime.stats.size--;
-  if (pooled.inUse) {
-    runtime.stats.acquired--;
-  } else {
-    runtime.stats.available--;
-  }
+  finalizeDestroyedConnection(runtime, pooled);
 }
 function getPoolConfig(runtime) {
   return runtime.options.pool || {};
@@ -4331,12 +4355,7 @@ async function acquireConnection(runtime) {
   }
   const available = runtime.connections.find((connection) => !connection.inUse);
   if (available) {
-    available.inUse = true;
-    available.lastUsedAt = Date.now();
-    runtime.stats.available--;
-    runtime.stats.acquired++;
-    runtime.options.onAcquire?.(available.connection);
-    return available.connection;
+    return checkoutConnection(runtime, available);
   }
   const poolConfig = getPoolConfig(runtime);
   const max = poolConfig.max || 10;
@@ -4352,12 +4371,7 @@ async function acquireConnection(runtime) {
         await destroyConnection(runtime, pooled);
         throw new Error("Pool is draining");
       }
-      pooled.inUse = true;
-      pooled.lastUsedAt = Date.now();
-      runtime.stats.available--;
-      runtime.stats.acquired++;
-      runtime.options.onAcquire?.(connection);
-      return connection;
+      return checkoutConnection(runtime, pooled);
     } finally {
       runtime.creating--;
     }
@@ -4389,18 +4403,7 @@ async function releaseConnection(runtime, connection) {
   runtime.stats.available++;
   runtime.stats.acquired--;
   runtime.options.onRelease?.(connection);
-  const pending = runtime.pending.shift();
-  if (!pending) {
-    return;
-  }
-  clearTimeout(pending.timeout);
-  runtime.stats.pending--;
-  pooled.inUse = true;
-  pooled.lastUsedAt = Date.now();
-  runtime.stats.available--;
-  runtime.stats.acquired++;
-  runtime.options.onAcquire?.(connection);
-  pending.resolve(connection);
+  assignConnectionToPendingAcquire(runtime, connection);
 }
 
 // src/resources/pool-eviction.ts

package/dist/index.js

@@ -4068,6 +4068,23 @@ function createInitialPoolStats() {
     healthChecksFailed: 0
   };
 }
+function checkoutConnection(runtime, pooled) {
+  const previousLastUsedAt = pooled.lastUsedAt;
+  pooled.inUse = true;
+  pooled.lastUsedAt = Date.now();
+  runtime.stats.available--;
+  runtime.stats.acquired++;
+  try {
+    runtime.options.onAcquire?.(pooled.connection);
+    return pooled.connection;
+  } catch (error) {
+    pooled.inUse = false;
+    pooled.lastUsedAt = previousLastUsedAt;
+    runtime.stats.available++;
+    runtime.stats.acquired--;
+    throw error;
+  }
+}
 async function initializeConnections(runtime, count) {
   runtime.creating += count;
   const promises = Array.from({ length: count }, async () => {
@@ -4089,23 +4106,22 @@ async function initializeConnections(runtime, count) {
   await Promise.all(promises);
 }
 function assignConnectionToPendingAcquire(runtime, connection) {
-  const pending = runtime.pending.shift();
-  if (!pending) {
-    return;
-  }
   const pooled = runtime.connections.find((item) => item.connection === connection);
   if (!pooled) {
-    runtime.pending.unshift(pending);
     return;
   }
-  clearTimeout(pending.timeout);
-  runtime.stats.pending--;
-  pooled.inUse = true;
-  pooled.lastUsedAt = Date.now();
-  runtime.stats.available--;
-  runtime.stats.acquired++;
-  runtime.options.onAcquire?.(connection);
-  pending.resolve(connection);
+  let pending = runtime.pending.shift();
+  while (pending) {
+    clearTimeout(pending.timeout);
+    runtime.stats.pending--;
+    try {
+      pending.resolve(checkoutConnection(runtime, pooled));
+      return;
+    } catch (error) {
+      pending.reject(error instanceof Error ? error : new Error(String(error)));
+      pending = runtime.pending.shift();
+    }
+  }
 }
 async function createConnection(runtime) {
   const connection = await runtime.options.factory();
@@ -4121,22 +4137,30 @@ async function createConnection(runtime) {
   runtime.stats.available++;
   return connection;
 }
+function finalizeDestroyedConnection(runtime, pooled) {
+  runtime.stats.destroyed++;
+  runtime.stats.size--;
+  if (pooled.inUse) {
+    runtime.stats.acquired--;
+  } else {
+    runtime.stats.available--;
+  }
+}
 async function destroyConnection(runtime, pooled) {
   const index = runtime.connections.indexOf(pooled);
   if (index !== -1) {
     runtime.connections.splice(index, 1);
   }
   if (runtime.options.destroyer) {
-    await runtime.options.destroyer(pooled.connection);
+    try {
+      await runtime.options.destroyer(pooled.connection);
+    } catch (error) {
+      finalizeDestroyedConnection(runtime, pooled);
+      throw error;
+    }
   }
   runtime.options.onDestroy?.(pooled.connection);
-  runtime.stats.destroyed++;
-  runtime.stats.size--;
-  if (pooled.inUse) {
-    runtime.stats.acquired--;
-  } else {
-    runtime.stats.available--;
-  }
+  finalizeDestroyedConnection(runtime, pooled);
 }
 function getPoolConfig(runtime) {
   return runtime.options.pool || {};
@@ -4155,12 +4179,7 @@ async function acquireConnection(runtime) {
   }
   const available = runtime.connections.find((connection) => !connection.inUse);
   if (available) {
-    available.inUse = true;
-    available.lastUsedAt = Date.now();
-    runtime.stats.available--;
-    runtime.stats.acquired++;
-    runtime.options.onAcquire?.(available.connection);
-    return available.connection;
+    return checkoutConnection(runtime, available);
   }
   const poolConfig = getPoolConfig(runtime);
   const max = poolConfig.max || 10;
@@ -4176,12 +4195,7 @@ async function acquireConnection(runtime) {
         await destroyConnection(runtime, pooled);
         throw new Error("Pool is draining");
       }
-      pooled.inUse = true;
-      pooled.lastUsedAt = Date.now();
-      runtime.stats.available--;
-      runtime.stats.acquired++;
-      runtime.options.onAcquire?.(connection);
-      return connection;
+      return checkoutConnection(runtime, pooled);
     } finally {
       runtime.creating--;
     }
@@ -4213,18 +4227,7 @@ async function releaseConnection(runtime, connection) {
   runtime.stats.available++;
   runtime.stats.acquired--;
   runtime.options.onRelease?.(connection);
-  const pending = runtime.pending.shift();
-  if (!pending) {
-    return;
-  }
-  clearTimeout(pending.timeout);
-  runtime.stats.pending--;
-  pooled.inUse = true;
-  pooled.lastUsedAt = Date.now();
-  runtime.stats.available--;
-  runtime.stats.acquired++;
-  runtime.options.onAcquire?.(connection);
-  pending.resolve(connection);
+  assignConnectionToPendingAcquire(runtime, connection);
 }
 
 // src/resources/pool-eviction.ts

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@agentforge/core",
-  "version": "0.18.0",
+  "version": "0.18.1",
   "description": "Production-ready TypeScript agent framework built on LangGraph with orchestration, middleware, and typed abstractions.",
   "type": "module",
   "main": "./dist/index.cjs",