tradeblocks-mcp 3.10.1 → 3.10.2

package/dist/{chunk-JTNPDIFG.js → chunk-TZRDK5VL.js}

@@ -3435,6 +3435,10 @@ async function migrateMetadataToJson(conn, dataDir, blocksDir) {
 }
 
 // src/db/connection.ts
+var leaseCount = 0;
+var idleTimer = null;
+var openInFlight = null;
+var closeInFlight = null;
 var instance = null;
 var connection = null;
 var connectionMode = null;
@@ -3631,7 +3635,7 @@ async function openReadWriteConnection(dbPath, threads, memoryLimit) {
   await createMarketParquetViews(connection, dataRoot);
   await ensureMarketDataTables(connection);
   try {
-    const blocksDir = (await import("./sync-7O6U2M2G.js")).getBlocksDir(dataRoot);
+    const blocksDir = (await import("./sync-DVSXIIUZ.js")).getBlocksDir(dataRoot);
     await migrateMetadataToJson(connection, dataRoot, blocksDir);
   } catch (err) {
     console.warn(
@@ -3657,6 +3661,7 @@ async function openReadOnlyConnection(dbPath, threads, memoryLimit) {
   return connection;
 }
 function resetConnectionState() {
+  cancelIdleRelease();
   if (connection) {
     try {
       connection.closeSync();
@@ -3674,9 +3679,24 @@ function resetConnectionState() {
   connectionMode = null;
 }
 async function getConnection(dataDir) {
+  if (closeInFlight) {
+    await closeInFlight;
+  }
   if (connection) {
     return connection;
   }
+  if (openInFlight) {
+    return openInFlight;
+  }
+  openInFlight = openConnectionUnshared(dataDir).then((conn) => {
+    if (leaseCount === 0) scheduleIdleRelease();
+    return conn;
+  }).finally(() => {
+    openInFlight = null;
+  });
+  return openInFlight;
+}
+async function openConnectionUnshared(dataDir) {
   const dbPath = path11.join(dataDir, "analytics.duckdb");
   const threads = process.env.DUCKDB_THREADS || defaultThreads();
   const memoryLimit = process.env.DUCKDB_MEMORY_LIMIT || defaultMemoryLimit();
@@ -3728,14 +3748,14 @@ async function openWithLockContention(dbPath, dataDir, threads, memoryLimit, for
       if (!isLockError(roMessage)) throw roError;
       resetConnectionState();
     }
-    const terminated = await tryRecoverLockByTerminatingStaleProcess(
-      lastLockError,
-      dbPath,
-      forceRecovery
-    );
-    if (terminated) {
-      terminations += 1;
-      if (terminations <= MAX_TERMINATIONS) {
+    if (terminations < MAX_TERMINATIONS) {
+      const terminated = await tryRecoverLockByTerminatingStaleProcess(
+        lastLockError,
+        dbPath,
+        forceRecovery
+      );
+      if (terminated) {
+        terminations += 1;
         attempt = 0;
         continue;
       }
@@ -3754,12 +3774,74 @@ async function openWithLockContention(dbPath, dataDir, threads, memoryLimit, for
     await new Promise((r) => setTimeout(r, Math.min(delayMs, Math.max(0, deadline - Date.now()))));
   }
 }
-async function closeConnection() {
+var DEFAULT_IDLE_RELEASE_MS = 3e3;
+function idleReleaseMs() {
+  const raw = Number.parseInt(process.env.DUCKDB_IDLE_RELEASE_MS || "", 10);
+  return Number.isFinite(raw) && raw >= 0 ? raw : DEFAULT_IDLE_RELEASE_MS;
+}
+function cancelIdleRelease() {
+  if (idleTimer) {
+    clearTimeout(idleTimer);
+    idleTimer = null;
+  }
+}
+function scheduleIdleRelease() {
+  cancelIdleRelease();
+  if (leaseCount > 0) return;
+  const delay = idleReleaseMs();
+  idleTimer = setTimeout(() => {
+    idleTimer = null;
+    if (leaseCount > 0) return;
+    closeConnection({ abortIfLeased: true }).catch((error) => {
+      console.error(
+        `Failed to release the idle DuckDB handle: ${error instanceof Error ? error.message : String(error)}`
+      );
+    });
+  }, delay);
+  idleTimer.unref?.();
+}
+function acquireConnectionLease() {
+  leaseCount += 1;
+  cancelIdleRelease();
+}
+function releaseConnectionLease() {
+  if (leaseCount === 0) return;
+  leaseCount -= 1;
+  if (leaseCount === 0) {
+    scheduleIdleRelease();
+  }
+}
+function getConnectionLeaseCount() {
+  return leaseCount;
+}
+async function withConnectionLease(work) {
+  acquireConnectionLease();
+  try {
+    return await work();
+  } finally {
+    releaseConnectionLease();
+  }
+}
+async function closeConnection(options) {
+  if (closeInFlight) {
+    return closeInFlight;
+  }
+  closeInFlight = closeConnectionUnshared(options).finally(() => {
+    closeInFlight = null;
+  });
+  return closeInFlight;
+}
+async function closeConnectionUnshared(options) {
+  const abortIfLeased = options?.abortIfLeased === true;
+  cancelIdleRelease();
   if (connection) {
     try {
       await connection.run("CHECKPOINT");
     } catch {
     }
+    if (abortIfLeased && leaseCount > 0) {
+      return;
+    }
     try {
       await detachMarketDb(connection);
     } catch {
@@ -3790,7 +3872,8 @@ async function upgradeToReadWrite(dataDir, options) {
   if (!storedMarketDbPath) {
     storedMarketDbPath = resolveMarketDbPath(dataDir);
   }
-  const maxRetries = 4;
+  const configuredRetries = Number.parseInt(process.env.DUCKDB_WRITE_LOCK_RETRIES || "", 10);
+  const maxRetries = Number.isFinite(configuredRetries) && configuredRetries >= 0 ? configuredRetries : 10;
   const retryDelayMs = 1e3;
   let lastError = null;
   for (let attempt = 0; attempt <= maxRetries; attempt++) {
@@ -3949,7 +4032,7 @@ function isConnected() {
 function getCurrentConnection() {
   if (!connection) {
     throw new Error(
-      "No active DuckDB connection. Call getConnection(dataDir) during server init before accessing store contexts."
+      "No active DuckDB connection. The handle is released when idle (#445), so a store context must only be used inside a leased unit of work \u2014 the tool wrapper in index.ts takes that lease for every registered tool. Code reaching here runs outside any lease: wrap it in withConnectionLease()."
     );
   }
   return connection;
@@ -5521,6 +5604,10 @@ export {
   createMarketParquetViews,
   migrateMetadataToJson,
   getConnection,
+  acquireConnectionLease,
+  releaseConnectionLease,
+  getConnectionLeaseCount,
+  withConnectionLease,
   closeConnection,
   upgradeToReadWrite,
   downgradeToReadOnly,
@@ -5536,4 +5623,4 @@ export {
   loadReportingLog,
   importCsv
 };
-//# sourceMappingURL=chunk-JTNPDIFG.js.map
+//# sourceMappingURL=chunk-TZRDK5VL.js.map