@neetru/sdk 3.2.1 → 3.2.3

package/dist/db.mjs

@@ -274,7 +274,99 @@ var init_http = __esm({
   }
 });
 
+// src/db/sql/resilient-pool.ts
+function isResilientPool(x) {
+  return typeof x === "object" && x !== null && x[RESILIENT_POOL_BRAND] === true;
+}
+function isConnectionEstablishmentError(err) {
+  const code = err && typeof err === "object" && "code" in err ? err.code : void 0;
+  if (code === "ECONNREFUSED" || code === "ETIMEDOUT" || code === "ENOTFOUND" || code === "EHOSTUNREACH" || code === "EAI_AGAIN") {
+    return true;
+  }
+  if (typeof code === "string") {
+    const upper = code.toUpperCase();
+    if (upper === "CERT_HAS_EXPIRED" || upper.startsWith("ERR_SSL") || upper.startsWith("ERR_TLS") || upper.includes("CERTIFICATE_EXPIRED")) {
+      return true;
+    }
+  }
+  const rawMessage = err instanceof Error ? err.message : typeof err === "string" ? err : "";
+  const message = rawMessage.toLowerCase();
+  if (message.length === 0) return false;
+  if (message.includes("certificate expired") || message.includes("certificate has expired") || message.includes("cert_has_expired") || message.includes("alert certificate expired") || message.includes("err_ssl") || message.includes("sslv3 alert") || message.includes("tlsv1 alert")) {
+    return true;
+  }
+  if (message.includes("econnrefused") || message.includes("etimedout") || message.includes("enotfound") || message.includes("ehostunreach") || message.includes("getaddrinfo")) {
+    return true;
+  }
+  return false;
+}
+function createResilientPool(initialInner) {
+  const state = { current: initialInner };
+  let reLease = null;
+  const controls = {
+    __neetruSwapInner(newInner) {
+      const old = state.current;
+      state.current = newInner;
+      return old;
+    },
+    __neetruGetInner() {
+      return state.current;
+    },
+    __neetruSetReLease(fn) {
+      reLease = fn;
+    }
+  };
+  const invokeQuery = (pool, args) => pool.query(...args);
+  const wrappedQuery = (...args) => {
+    if (args.length > 0 && typeof args[args.length - 1] === "function") {
+      return invokeQuery(state.current, args);
+    }
+    return (async () => {
+      try {
+        return await invokeQuery(state.current, args);
+      } catch (err) {
+        const fn = reLease;
+        if (fn === null || !isConnectionEstablishmentError(err)) {
+          throw err;
+        }
+        try {
+          await fn();
+        } catch {
+          throw err;
+        }
+        return await invokeQuery(state.current, args);
+      }
+    })();
+  };
+  const target = Object.create(Object.getPrototypeOf(initialInner));
+  const proxy = new Proxy(target, {
+    get(_t, prop) {
+      if (prop === RESILIENT_POOL_BRAND) return true;
+      if (prop === "__neetruSwapInner") return controls.__neetruSwapInner;
+      if (prop === "__neetruGetInner") return controls.__neetruGetInner;
+      if (prop === "__neetruSetReLease") return controls.__neetruSetReLease;
+      if (prop === "query") return wrappedQuery;
+      if (prop === "constructor") {
+        return state.current.constructor;
+      }
+      const current = state.current;
+      const value = Reflect.get(current, prop, current);
+      return typeof value === "function" ? value.bind(current) : value;
+    }
+  });
+  return proxy;
+}
+var RESILIENT_POOL_BRAND;
+var init_resilient_pool = __esm({
+  "src/db/sql/resilient-pool.ts"() {
+    RESILIENT_POOL_BRAND = /* @__PURE__ */ Symbol.for("neetru.sdk.resilientPool");
+  }
+});
+
 // src/db/sql/lease.ts
+function hasConnectionChanged(oldLease, newLease) {
+  return oldLease.leaseId !== newLease.leaseId || oldLease.credentialVersion !== newLease.credentialVersion || oldLease.host !== newLease.host || oldLease.port !== newLease.port || oldLease.dbName !== newLease.dbName || oldLease.user !== newLease.user || oldLease.password !== newLease.password || oldLease.sslca !== newLease.sslca || oldLease.clientCert !== newLease.clientCert || oldLease.clientKey !== newLease.clientKey;
+}
 function mapPoolError(err) {
   if (err instanceof NeetruDbError) return err;
   const message = err instanceof Error ? err.message : String(err);
@@ -300,6 +392,7 @@ var RENEWAL_THRESHOLD, MIN_RENEWAL_DELAY_MS, SqlLeaseManager;
 var init_lease = __esm({
   "src/db/sql/lease.ts"() {
     init_db_errors();
+    init_resilient_pool();
     RENEWAL_THRESHOLD = 0.8;
     MIN_RENEWAL_DELAY_MS = 3e4;
     SqlLeaseManager = class {
@@ -308,6 +401,12 @@ var init_lease = __esm({
       _orm;
       _renewalTimer = null;
       _closed = false;
+      /**
+       * Promessa de re-lease REATIVO em voo — coalesce chamadas concorrentes
+       * (N queries falhando ao mesmo tempo disparam UM único re-lease). Ver
+       * `_reactiveReLease`.
+       */
+      _reLeaseInFlight = null;
       _deps;
       constructor(lease, pool, orm, deps) {
         this._lease = lease;
@@ -379,37 +478,90 @@ var init_lease = __esm({
       async _doRenew() {
         if (this._closed) return;
         try {
-          const renewOpts = {
-            leaseId: this._lease.leaseId
+          const withDb = (o) => {
+            if (this._deps.database !== void 0) o.database = this._deps.database;
+            return o;
           };
-          if (this._deps.database !== void 0) {
-            renewOpts.database = this._deps.database;
+          let newLease;
+          try {
+            newLease = await this._deps.fetchLease(withDb({ leaseId: this._lease.leaseId }));
+          } catch (renewErr) {
+            newLease = await this._deps.fetchLease(withDb({}));
           }
-          const newLease = await this._deps.fetchLease(renewOpts);
           await this._swapPool(newLease);
         } catch (err) {
           this._renewalTimer = setTimeout(() => {
             void this._doRenew();
           }, MIN_RENEWAL_DELAY_MS);
-          if (process.env["NODE_ENV"] !== "production") {
-            console.warn("[SqlLeaseManager] Lease renewal failed, will retry:", err);
-          }
+          console.warn(
+            `[SqlLeaseManager] renova\xE7\xE3o E emiss\xE3o de lease novo falharam; re-tentando em ${MIN_RENEWAL_DELAY_MS}ms (cert mTLS expira ~15min ap\xF3s o \xFAltimo swap):`,
+            err instanceof Error ? err.message : err
+          );
           return;
         }
         this._scheduleRenewal();
       }
+      // ─── Re-lease REATIVO (P1.5) ───────────────────────────────────────────────
+      /**
+       * Recuperação REATIVA: chamado pelo wrapper resiliente (`resilient-pool.ts`)
+       * quando uma query plana falha num erro de ESTABELECIMENTO de conexão (cert
+       * mTLS expirado no meio de uma request, ou timer de renovação que não
+       * disparou). Emite um lease NOVO e faz o swap do pool interno do wrapper com
+       * o certificado fresco; o wrapper então re-tenta a query 1x no pool novo.
+       *
+       * Vai DIRETO ao lease novo (sem `leaseId`) — a conexão falhou porque o
+       * certificado do lease atual está ruim, então `renew` (que depende do lease
+       * antigo estar vivo) não é o caminho mais robusto. Emitir do zero é o que já
+       * cura o death-spiral no caminho proativo (ver `_doRenew`).
+       *
+       * Coalesce concorrência: N queries falhando ao mesmo tempo compartilham UM
+       * único re-lease (não N leases). Após o swap, cada query re-tenta no pool
+       * vigente.
+       *
+       * NÃO engole erro: se o re-lease falhar, propaga — o wrapper então relança o
+       * erro ORIGINAL da query pro consumer.
+       */
+      async _reactiveReLease() {
+        if (this._closed) {
+          throw new NeetruDbError(
+            "db_unavailable",
+            "SqlLeaseManager foi fechado \u2014 re-lease reativo abortado."
+          );
+        }
+        if (this._reLeaseInFlight !== null) return this._reLeaseInFlight;
+        const run = (async () => {
+          const withDb = (o) => {
+            if (this._deps.database !== void 0) o.database = this._deps.database;
+            return o;
+          };
+          const fresh = await this._deps.fetchLease(withDb({}));
+          await this._swapPool(fresh);
+        })();
+        this._reLeaseInFlight = run;
+        try {
+          await run;
+        } finally {
+          this._reLeaseInFlight = null;
+        }
+      }
       /**
        * Hot-swap do pool: cria pool novo, atualiza ponteiros, drena pool antigo.
        *
        * Exposto como método protegido para testes.
        */
       async _swapPool(newLease) {
-        const credRotated = newLease.credentialVersion !== this._lease.credentialVersion;
-        if (!credRotated && newLease.leaseId === this._lease.leaseId) {
+        if (!hasConnectionChanged(this._lease, newLease)) {
           this._lease = newLease;
           return;
         }
         const newPool = this._deps.createPool(newLease);
+        if (isResilientPool(this._pool)) {
+          const oldInner = this._pool.__neetruSwapInner(newPool);
+          this._lease = newLease;
+          void oldInner.end().catch(() => {
+          });
+          return;
+        }
         const newOrm = this._deps.createDrizzle(newPool, this._deps.schema);
         const oldPool = this._pool;
         this._lease = newLease;
@@ -564,23 +716,26 @@ async function createSqlClientWithDeps(deps, options) {
     const msg = err instanceof Error ? err.message : String(err);
     throw new NeetruDbError("db_unavailable", `Falha ao obter lease inicial: ${msg}`);
   }
-  let pool;
+  let innerPool;
   try {
-    pool = depsWithDb.createPool(lease);
+    innerPool = depsWithDb.createPool(lease);
   } catch (err) {
     const msg = err instanceof Error ? err.message : String(err);
     throw new NeetruDbError("db_unavailable", `Falha ao abrir pool: ${msg}`);
   }
+  const pool = createResilientPool(innerPool);
   let orm;
   try {
     orm = depsWithDb.createDrizzle(pool, depsWithDb.schema);
   } catch (err) {
-    await pool.end().catch(() => {
+    await innerPool.end().catch(() => {
     });
     const msg = err instanceof Error ? err.message : String(err);
     throw new NeetruDbError("db_unavailable", `Falha ao criar handle Drizzle: ${msg}`);
   }
-  return new SqlLeaseManager(lease, pool, orm, depsWithDb);
+  const manager = new SqlLeaseManager(lease, pool, orm, depsWithDb);
+  pool.__neetruSetReLease(() => manager._reactiveReLease());
+  return manager;
 }
 async function createSqlClientFromConfig(config, schema, options) {
   const fetchLease = createHttpLeaseFetcher(config);
@@ -639,6 +794,7 @@ var init_sql_client = __esm({
   "src/db/sql/sql-client.ts"() {
     init_db_errors();
     init_lease();
+    init_resilient_pool();
     LEASE_ENDPOINT = "/api/sdk/v1/db/lease";
     LEASE_RENEW_ENDPOINT = "/api/sdk/v1/db/lease/renew";
   }