@gemus/mcp-proxy 0.1.7 → 0.1.9

package/src/upstreamHttp.mjs

@@ -0,0 +1,69 @@
+import { Agent, Pool, fetch as undiciFetch } from 'undici'
+
+const DEFAULT_ATTEMPT_TIMEOUT_MS = 1000
+const MIN_ATTEMPT_TIMEOUT_MS = 10
+const SHUTDOWN_TIMEOUT_MS = 2000
+
+function attemptTimeoutFromEnvironment(value) {
+  const parsed = Number(value)
+  if (!value || !Number.isFinite(parsed) || parsed <= 0) return DEFAULT_ATTEMPT_TIMEOUT_MS
+  return Math.max(MIN_ATTEMPT_TIMEOUT_MS, Math.trunc(parsed))
+}
+
+function settleClose(closePromise) {
+  return new Promise((resolve) => {
+    const timer = setTimeout(() => resolve('timed_out'), SHUTDOWN_TIMEOUT_MS)
+    Promise.resolve(closePromise).then(
+      () => {
+        clearTimeout(timer)
+        resolve('settled')
+      },
+      () => {
+        clearTimeout(timer)
+        resolve('rejected')
+      },
+    )
+  })
+}
+
+/** Owns the MCP upstream dispatcher and its bounded shutdown. */
+export function createUpstreamHttp() {
+  const attemptTimeoutMs = attemptTimeoutFromEnvironment(process.env.PROXY_CONNECT_ATTEMPT_TIMEOUT_MS)
+  const originDispatchers = new Set()
+  const dispatcher = new Agent({
+    factory(origin, options) {
+      const originDispatcher = new Pool(origin, options)
+      originDispatchers.add(originDispatcher)
+      return originDispatcher
+    },
+    connect: {
+      autoSelectFamily: true,
+      autoSelectFamilyAttemptTimeout: attemptTimeoutMs,
+    },
+  })
+  let shutdownPromise
+
+  function upstreamFetch(input, init) {
+    return undiciFetch(input, { ...init, dispatcher })
+  }
+
+  function shutdown() {
+    if (!shutdownPromise) {
+      shutdownPromise = (async () => {
+        const closePromise = dispatcher.close()
+        const outcome = await settleClose(closePromise)
+        if (outcome !== 'settled') {
+          const error = new Error('MCP upstream dispatcher shutdown timed out')
+          await Promise.allSettled([
+            dispatcher.destroy(error),
+            ...Array.from(originDispatchers, (originDispatcher) => originDispatcher.destroy(error)),
+          ])
+        }
+        originDispatchers.clear()
+      })()
+    }
+    return shutdownPromise
+  }
+
+  return { fetch: upstreamFetch, shutdown, attemptTimeoutMs }
+}