@take-out/postgres 0.0.28

package/dist/esm/helpers/getDBClient.js

@@ -0,0 +1,155 @@
+import pg from "pg";
+const cache = /* @__PURE__ */ new Map(), createPoolKey = (connectionString) => connectionString, getOrCreatePoolCache = (connectionString, config) => {
+  const key = createPoolKey(connectionString);
+  return cache.has(key) || cache.set(key, {
+    pool: new pg.Pool(config),
+    maxConnections: null,
+    reservedConnections: null,
+    openedConnections: null,
+    openedConnectionsLastUpdate: null
+  }), cache.get(key);
+};
+async function getDBClient(options = {}) {
+  const { pool, connectionString, retries = 8 } = options;
+  if (!pool && !connectionString)
+    throw new Error("Either pool or connectionString must be provided");
+  let client = null;
+  try {
+    return client = await tryToGetNewClientFromPool(pool, connectionString, retries), client;
+  } catch (error) {
+    throw console.error("Failed to get DB client:", error), error;
+  }
+}
+async function tryToGetNewClientFromPool(providedPool, connectionString, retries) {
+  const { default: retry } = await import("async-retry");
+  return await retry(
+    async () => {
+      if (providedPool) {
+        console.info("Connecting to provided pool...");
+        const client2 = await providedPool.connect();
+        return console.info("Connected to pool"), client2;
+      }
+      if (!connectionString)
+        throw new Error("No connection string provided");
+      const poolCache = getOrCreatePoolCache(connectionString, {
+        connectionString,
+        connectionTimeoutMillis: 5e3,
+        // idle_session_timeout set to 35s on server, client timeout at 30s
+        // fix via https://github.com/brianc/node-postgres/issues/2718#issuecomment-2094885323
+        idleTimeoutMillis: 3e4,
+        allowExitOnIdle: !0
+      });
+      console.info(`Connecting to pool ${connectionString}...`);
+      const client = await poolCache.pool.connect();
+      return console.info("Connected to pool"), client;
+    },
+    {
+      retries,
+      minTimeout: 300,
+      factor: 2,
+      maxTimeout: 8e3
+    }
+  );
+}
+async function queryDb(queryText, params, options = {}) {
+  let client = null;
+  try {
+    return client = await tryToGetNewClientFromPool(
+      options.pool,
+      options.connectionString,
+      options.retries || 8
+    ), await client.query(queryText, params);
+  } catch (error) {
+    throw console.error("Database query failed:", {
+      query: queryText,
+      error: error instanceof Error ? error.message : String(error)
+    }), error;
+  } finally {
+    if (client && options.connectionString)
+      if (await checkForTooManyConnections(
+        client,
+        options.connectionString
+      )) {
+        const poolCache = cache.get(createPoolKey(options.connectionString));
+        client.release(), await poolCache?.pool.end(), poolCache && cache.delete(createPoolKey(options.connectionString));
+      } else
+        client.release();
+    else client && client.release();
+  }
+}
+async function checkForTooManyConnections(client, connectionString) {
+  const poolCache = cache.get(createPoolKey(connectionString));
+  if (!poolCache) return !1;
+  const currentTime = Date.now(), openedConnectionsMaxAge = 1e4, maxConnectionsTolerance = 0.9;
+  if (poolCache.maxConnections === null || poolCache.reservedConnections === null) {
+    const [maxConnections, reservedConnections] = await getConnectionLimits(client);
+    console.info(
+      `Max connections: ${maxConnections}, Reserved connections: ${reservedConnections}`
+    ), poolCache.maxConnections = maxConnections, poolCache.reservedConnections = reservedConnections;
+  }
+  if (poolCache.openedConnections === null || poolCache.openedConnectionsLastUpdate === null || currentTime - poolCache.openedConnectionsLastUpdate > openedConnectionsMaxAge) {
+    const openedConnections = await getOpenedConnections(client, connectionString);
+    poolCache.openedConnections = openedConnections, poolCache.openedConnectionsLastUpdate = currentTime;
+  }
+  return poolCache.openedConnections > (poolCache.maxConnections - poolCache.reservedConnections) * maxConnectionsTolerance ? (console.warn(
+    `Too many connections detected: ${poolCache.openedConnections}/${poolCache.maxConnections - poolCache.reservedConnections}`
+  ), !0) : !1;
+}
+async function getConnectionLimits(client) {
+  console.info("Getting connection limits...");
+  const maxConnectionsResult = await client.query("SHOW max_connections"), reservedConnectionResult = await client.query(
+    "SHOW superuser_reserved_connections"
+  );
+  return [
+    Number.parseInt(maxConnectionsResult.rows[0].max_connections, 10),
+    Number.parseInt(reservedConnectionResult.rows[0].superuser_reserved_connections, 10)
+  ];
+}
+async function getOpenedConnections(client, connectionString) {
+  const dbName = new URL(connectionString).pathname.slice(1);
+  console.info("Getting opened connections...");
+  const openConnectionsResult = await client.query(
+    "SELECT numbackends as opened_connections FROM pg_stat_database WHERE datname = $1",
+    [dbName]
+  ), result = Number.parseInt(
+    openConnectionsResult.rows[0]?.opened_connections || 0,
+    10
+  );
+  return console.info(`Opened connections: ${result}`), result;
+}
+async function getNewClient(options = {}) {
+  const { connectionString } = options;
+  if (!connectionString)
+    throw new Error("connectionString is required for getNewClient");
+  try {
+    return await tryToGetNewClient(connectionString);
+  } catch (error) {
+    throw console.error("Failed to get new client:", error), error;
+  }
+}
+async function tryToGetNewClient(connectionString) {
+  const configurations = {
+    connectionString,
+    connectionTimeoutMillis: 5e3,
+    idleTimeoutMillis: 3e4,
+    allowExitOnIdle: !0
+  }, { default: retry } = await import("async-retry");
+  return await retry(
+    async () => {
+      const newClient = new pg.Client(configurations);
+      return await newClient.connect(), newClient;
+    },
+    {
+      retries: 10,
+      minTimeout: 100,
+      factor: 2,
+      maxTimeout: 5e3
+    }
+  );
+}
+export {
+  getDBClient,
+  getNewClient,
+  queryDb
+};
+//# sourceMappingURL=getDBClient.js.map

package/dist/esm/helpers/getDBClient.js.map

@@ -0,0 +1,6 @@
+{
+  "version": 3,
+  "sources": ["../../../src/helpers/getDBClient.ts"],
+  "mappings": "AAAA,OAAO,QAAwC;AAY/C,MAAM,QAAQ,oBAAI,IAShB,GAEI,gBAAgB,CAAC,qBAA6B,kBAE9C,uBAAuB,CAAC,kBAA0B,WAA0B;AAChF,QAAM,MAAM,cAAc,gBAAgB;AAE1C,SAAK,MAAM,IAAI,GAAG,KAChB,MAAM,IAAI,KAAK;AAAA,IACb,MAAM,IAAI,GAAG,KAAK,MAAM;AAAA,IACxB,gBAAgB;AAAA,IAChB,qBAAqB;AAAA,IACrB,mBAAmB;AAAA,IACnB,6BAA6B;AAAA,EAC/B,CAAC,GAGI,MAAM,IAAI,GAAG;AACtB;AAEA,eAAsB,YAAY,UAA8B,CAAC,GAAwB;AACvF,QAAM,EAAE,MAAM,kBAAkB,UAAU,EAAE,IAAI;AAEhD,MAAI,CAAC,QAAQ,CAAC;AACZ,UAAM,IAAI,MAAM,kDAAkD;AAGpE,MAAI,SAA4B;AAEhC,MAAI;AACF,oBAAS,MAAM,0BAA0B,MAAM,kBAAkB,OAAO,GACjE;AAAA,EACT,SAAS,OAAO;AACd,kBAAQ,MAAM,4BAA4B,KAAK,GACzC;AAAA,EACR;AACF;AAEA,eAAe,0BACb,cACA,kBACA,SACqB;AACrB,QAAM,EAAE,SAAS,MAAM,IAAI,MAAM,OAAO,aAAa;AAsCrD,SArCuB,MAAM;AAAA,IAC3B,YAAY;AACV,UAAI,cAAc;AAChB,gBAAQ,KAAK,gCAAgC;AAC7C,cAAMA,UAAS,MAAM,aAAa,QAAQ;AAC1C,uBAAQ,KAAK,mBAAmB,GACzBA;AAAA,MACT;AAEA,UAAI,CAAC;AACH,cAAM,IAAI,MAAM,+BAA+B;AAYjD,YAAM,YAAY,qBAAqB,kBATD;AAAA,QACpC;AAAA,QACA,yBAAyB;AAAA;AAAA;AAAA,QAGzB,mBAAmB;AAAA,QACnB,iBAAiB;AAAA,MACnB,CAEuE;AAEvE,cAAQ,KAAK,sBAAsB,gBAAgB,KAAK;AACxD,YAAM,SAAS,MAAM,UAAU,KAAK,QAAQ;AAC5C,qBAAQ,KAAK,mBAAmB,GACzB;AAAA,IACT;AAAA,IACA;AAAA,MACE;AAAA,MACA,YAAY;AAAA,MACZ,QAAQ;AAAA,MACR,YAAY;AAAA,IACd;AAAA,EACF;AAGF;AAEA,eAAsB,QACpB,WACA,QACA,UAA8B,CAAC,GACD;AAC9B,MAAI,SAA4B;AAEhC,MAAI;AACF,oBAAS,MAAM;AAAA,MACb,QAAQ;AAAA,MACR,QAAQ;AAAA,MACR,QAAQ,WAAW;AAAA,IACrB,GACO,MAAM,OAAO,MAAM,WAAW,MAAM;AAAA,EAC7C,SAAS,OAAO;AACd,kBAAQ,MAAM,0BAA0B;AAAA,MACtC,OAAO;AAAA,MACP,OAAO,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK;AAAA,IAC9D,CAAC,GACK;AAAA,EACR,UAAE;AACA,QAAI,UAAU,QAAQ;AAMpB,UAL2B,MAAM;AAAA,QAC/B;AAAA,QACA,QAAQ;AAAA,MACV,GAEwB;AACtB,cAAM,YAAY,MAAM,IAAI,cAAc,QAAQ,gBAAgB,CAAC;AACnE,eAAO,QAAQ,GACf,MAAM,WAAW,KAAK,IAAI,GACtB,aACF,MAAM,OAAO,cAAc,QAAQ,gBAAgB,CAAC;AAAA,MAExD;AACE,eAAO,QAAQ;AAAA,QAEZ,CAAI,UACT,OAAO,QAAQ;AAAA,EAEnB;AACF;AAEA,eAAe,2BACb,QACA,kBACkB;AAClB,QAAM,YAAY,MAAM,IAAI,cAAc,gBAAgB,CAAC;AAC3D,MAAI,CAAC,UAAW,QAAO;AAEvB,QAAM,cAAc,KAAK,IAAI,GACvB,0BAA0B,KAC1B,0BAA0B;AAEhC,MAAI,UAAU,mBAAmB,QAAQ,UAAU,wBAAwB,MAAM;AAC/E,UAAM,CAAC,gBAAgB,mBAAmB,IAAI,MAAM,oBAAoB,MAAM;AAC9E,YAAQ;AAAA,MACN,oBAAoB,cAAc,2BAA2B,mBAAmB;AAAA,IAClF,GACA,UAAU,iBAAiB,gBAC3B,UAAU,sBAAsB;AAAA,EAClC;AAEA,MACE,UAAU,sBAAsB,QAChC,UAAU,gCAAgC,QAC1C,cAAc,UAAU,8BAA8B,yBACtD;AACA,UAAM,oBAAoB,MAAM,qBAAqB,QAAQ,gBAAgB;AAC7E,cAAU,oBAAoB,mBAC9B,UAAU,8BAA8B;AAAA,EAC1C;AAEA,SACE,UAAU,qBACT,UAAU,iBAAiB,UAAU,uBAAuB,2BAE7D,QAAQ;AAAA,IACN,kCAAkC,UAAU,iBAAiB,IAAI,UAAU,iBAAiB,UAAU,mBAAmB;AAAA,EAC3H,GACO,MAGF;AACT;AAEA,eAAe,oBAAoB,QAA+C;AAChF,UAAQ,KAAK,8BAA8B;AAC3C,QAAM,uBAAuB,MAAM,OAAO,MAAM,sBAAsB,GAChE,2BAA2B,MAAM,OAAO;AAAA,IAC5C;AAAA,EACF;AAEA,SAAO;AAAA,IACL,OAAO,SAAS,qBAAqB,KAAK,CAAC,EAAE,iBAAiB,EAAE;AAAA,IAChE,OAAO,SAAS,yBAAyB,KAAK,CAAC,EAAE,gCAAgC,EAAE;AAAA,EACrF;AACF;AAEA,eAAe,qBACb,QACA,kBACiB;AAEjB,QAAM,SAAS,IAAI,IAAI,gBAAgB,EAAE,SAAS,MAAM,CAAC;AACzD,UAAQ,KAAK,+BAA+B;AAC5C,QAAM,wBAAwB,MAAM,OAAO;AAAA,IACzC;AAAA,IACA,CAAC,MAAM;AAAA,EACT,GACM,SAAS,OAAO;AAAA,IACpB,sBAAsB,KAAK,CAAC,GAAG,sBAAsB;AAAA,IACrD;AAAA,EACF;AACA,iBAAQ,KAAK,uBAAuB,MAAM,EAAE,GACrC;AACT;AAEA,eAAsB,aAAa,UAA8B,CAAC,GAAuB;AACvF,QAAM,EAAE,iBAAiB,IAAI;AAE7B,MAAI,CAAC;AACH,UAAM,IAAI,MAAM,+CAA+C;AAGjE,MAAI;AAEF,WADe,MAAM,kBAAkB,gBAAgB;AAAA,EAEzD,SAAS,OAAO;AACd,kBAAQ,MAAM,6BAA6B,KAAK,GAC1C;AAAA,EACR;AACF;AAEA,eAAe,kBAAkB,kBAA8C;AAC7E,QAAM,iBAAgC;AAAA,IACpC;AAAA,IACA,yBAAyB;AAAA,IACzB,mBAAmB;AAAA,IACnB,iBAAiB;AAAA,EACnB,GAEM,EAAE,SAAS,MAAM,IAAI,MAAM,OAAO,aAAa;AAerD,SAde,MAAM;AAAA,IACnB,YAAY;AACV,YAAM,YAAY,IAAI,GAAG,OAAO,cAAc;AAC9C,mBAAM,UAAU,QAAQ,GACjB;AAAA,IACT;AAAA,IACA;AAAA,MACE,SAAS;AAAA,MACT,YAAY;AAAA,MACZ,QAAQ;AAAA,MACR,YAAY;AAAA,IACd;AAAA,EACF;AAGF;",
+  "names": ["client"]
+}

package/dist/esm/helpers/getDBClient.mjs

@@ -0,0 +1,136 @@
+import pg from "pg";
+const cache = /* @__PURE__ */new Map(),
+  createPoolKey = connectionString => connectionString,
+  getOrCreatePoolCache = (connectionString, config) => {
+    const key = createPoolKey(connectionString);
+    return cache.has(key) || cache.set(key, {
+      pool: new pg.Pool(config),
+      maxConnections: null,
+      reservedConnections: null,
+      openedConnections: null,
+      openedConnectionsLastUpdate: null
+    }), cache.get(key);
+  };
+async function getDBClient(options = {}) {
+  const {
+    pool,
+    connectionString,
+    retries = 8
+  } = options;
+  if (!pool && !connectionString) throw new Error("Either pool or connectionString must be provided");
+  let client = null;
+  try {
+    return client = await tryToGetNewClientFromPool(pool, connectionString, retries), client;
+  } catch (error) {
+    throw console.error("Failed to get DB client:", error), error;
+  }
+}
+async function tryToGetNewClientFromPool(providedPool, connectionString, retries) {
+  const {
+    default: retry
+  } = await import("async-retry");
+  return await retry(async () => {
+    if (providedPool) {
+      console.info("Connecting to provided pool...");
+      const client2 = await providedPool.connect();
+      return console.info("Connected to pool"), client2;
+    }
+    if (!connectionString) throw new Error("No connection string provided");
+    const poolCache = getOrCreatePoolCache(connectionString, {
+      connectionString,
+      connectionTimeoutMillis: 5e3,
+      // idle_session_timeout set to 35s on server, client timeout at 30s
+      // fix via https://github.com/brianc/node-postgres/issues/2718#issuecomment-2094885323
+      idleTimeoutMillis: 3e4,
+      allowExitOnIdle: !0
+    });
+    console.info(`Connecting to pool ${connectionString}...`);
+    const client = await poolCache.pool.connect();
+    return console.info("Connected to pool"), client;
+  }, {
+    retries,
+    minTimeout: 300,
+    factor: 2,
+    maxTimeout: 8e3
+  });
+}
+async function queryDb(queryText, params, options = {}) {
+  let client = null;
+  try {
+    return client = await tryToGetNewClientFromPool(options.pool, options.connectionString, options.retries || 8), await client.query(queryText, params);
+  } catch (error) {
+    throw console.error("Database query failed:", {
+      query: queryText,
+      error: error instanceof Error ? error.message : String(error)
+    }), error;
+  } finally {
+    if (client && options.connectionString) {
+      if (await checkForTooManyConnections(client, options.connectionString)) {
+        const poolCache = cache.get(createPoolKey(options.connectionString));
+        client.release(), await poolCache?.pool.end(), poolCache && cache.delete(createPoolKey(options.connectionString));
+      } else client.release();
+    } else client && client.release();
+  }
+}
+async function checkForTooManyConnections(client, connectionString) {
+  const poolCache = cache.get(createPoolKey(connectionString));
+  if (!poolCache) return !1;
+  const currentTime = Date.now(),
+    openedConnectionsMaxAge = 1e4,
+    maxConnectionsTolerance = 0.9;
+  if (poolCache.maxConnections === null || poolCache.reservedConnections === null) {
+    const [maxConnections, reservedConnections] = await getConnectionLimits(client);
+    console.info(`Max connections: ${maxConnections}, Reserved connections: ${reservedConnections}`), poolCache.maxConnections = maxConnections, poolCache.reservedConnections = reservedConnections;
+  }
+  if (poolCache.openedConnections === null || poolCache.openedConnectionsLastUpdate === null || currentTime - poolCache.openedConnectionsLastUpdate > openedConnectionsMaxAge) {
+    const openedConnections = await getOpenedConnections(client, connectionString);
+    poolCache.openedConnections = openedConnections, poolCache.openedConnectionsLastUpdate = currentTime;
+  }
+  return poolCache.openedConnections > (poolCache.maxConnections - poolCache.reservedConnections) * maxConnectionsTolerance ? (console.warn(`Too many connections detected: ${poolCache.openedConnections}/${poolCache.maxConnections - poolCache.reservedConnections}`), !0) : !1;
+}
+async function getConnectionLimits(client) {
+  console.info("Getting connection limits...");
+  const maxConnectionsResult = await client.query("SHOW max_connections"),
+    reservedConnectionResult = await client.query("SHOW superuser_reserved_connections");
+  return [Number.parseInt(maxConnectionsResult.rows[0].max_connections, 10), Number.parseInt(reservedConnectionResult.rows[0].superuser_reserved_connections, 10)];
+}
+async function getOpenedConnections(client, connectionString) {
+  const dbName = new URL(connectionString).pathname.slice(1);
+  console.info("Getting opened connections...");
+  const openConnectionsResult = await client.query("SELECT numbackends as opened_connections FROM pg_stat_database WHERE datname = $1", [dbName]),
+    result = Number.parseInt(openConnectionsResult.rows[0]?.opened_connections || 0, 10);
+  return console.info(`Opened connections: ${result}`), result;
+}
+async function getNewClient(options = {}) {
+  const {
+    connectionString
+  } = options;
+  if (!connectionString) throw new Error("connectionString is required for getNewClient");
+  try {
+    return await tryToGetNewClient(connectionString);
+  } catch (error) {
+    throw console.error("Failed to get new client:", error), error;
+  }
+}
+async function tryToGetNewClient(connectionString) {
+  const configurations = {
+      connectionString,
+      connectionTimeoutMillis: 5e3,
+      idleTimeoutMillis: 3e4,
+      allowExitOnIdle: !0
+    },
+    {
+      default: retry
+    } = await import("async-retry");
+  return await retry(async () => {
+    const newClient = new pg.Client(configurations);
+    return await newClient.connect(), newClient;
+  }, {
+    retries: 10,
+    minTimeout: 100,
+    factor: 2,
+    maxTimeout: 5e3
+  });
+}
+export { getDBClient, getNewClient, queryDb };
+//# sourceMappingURL=getDBClient.mjs.map

package/dist/esm/helpers/getDBClient.mjs.map

@@ -0,0 +1 @@
+{"version":3,"names":["pg","cache","Map","createPoolKey","connectionString","getOrCreatePoolCache","config","key","has","set","pool","Pool","maxConnections","reservedConnections","openedConnections","openedConnectionsLastUpdate","get","getDBClient","options","retries","Error","client","tryToGetNewClientFromPool","error","console","providedPool","default","retry","info","client2","connect","poolCache","connectionTimeoutMillis","idleTimeoutMillis","allowExitOnIdle","minTimeout","factor","maxTimeout","queryDb","queryText","params","query","message","String","checkForTooManyConnections","release","end","delete","currentTime","Date","now","openedConnectionsMaxAge","maxConnectionsTolerance","getConnectionLimits","getOpenedConnections","warn","maxConnectionsResult","reservedConnectionResult","Number","parseInt","rows","max_connections","superuser_reserved_connections","dbName","URL","pathname","slice","openConnectionsResult","result","opened_connections","getNewClient","tryToGetNewClient","configurations","newClient","Client"],"sources":["../../../src/helpers/getDBClient.ts"],"sourcesContent":[null],"mappings":"AAAA,OAAOA,EAAA,MAAwC;AAY/C,MAAMC,KAAA,GAAQ,mBAAIC,GAAA,CAShB;EAEIC,aAAA,GAAiBC,gBAAA,IAA6BA,gBAAA;EAE9CC,oBAAA,GAAuBA,CAACD,gBAAA,EAA0BE,MAAA,KAA0B;IAChF,MAAMC,GAAA,GAAMJ,aAAA,CAAcC,gBAAgB;IAE1C,OAAKH,KAAA,CAAMO,GAAA,CAAID,GAAG,KAChBN,KAAA,CAAMQ,GAAA,CAAIF,GAAA,EAAK;MACbG,IAAA,EAAM,IAAIV,EAAA,CAAGW,IAAA,CAAKL,MAAM;MACxBM,cAAA,EAAgB;MAChBC,mBAAA,EAAqB;MACrBC,iBAAA,EAAmB;MACnBC,2BAAA,EAA6B;IAC/B,CAAC,GAGId,KAAA,CAAMe,GAAA,CAAIT,GAAG;EACtB;AAEA,eAAsBU,YAAYC,OAAA,GAA8B,CAAC,GAAwB;EACvF,MAAM;IAAER,IAAA;IAAMN,gBAAA;IAAkBe,OAAA,GAAU;EAAE,IAAID,OAAA;EAEhD,IAAI,CAACR,IAAA,IAAQ,CAACN,gBAAA,EACZ,MAAM,IAAIgB,KAAA,CAAM,kDAAkD;EAGpE,IAAIC,MAAA,GAA4B;EAEhC,IAAI;IACF,OAAAA,MAAA,GAAS,MAAMC,yBAAA,CAA0BZ,IAAA,EAAMN,gBAAA,EAAkBe,OAAO,GACjEE,MAAA;EACT,SAASE,KAAA,EAAO;IACd,MAAAC,OAAA,CAAQD,KAAA,CAAM,4BAA4BA,KAAK,GACzCA,KAAA;EACR;AACF;AAEA,eAAeD,0BACbG,YAAA,EACArB,gBAAA,EACAe,OAAA,EACqB;EACrB,MAAM;IAAEO,OAAA,EAASC;EAAM,IAAI,MAAM,OAAO,aAAa;EAsCrD,OArCuB,MAAMA,KAAA,CAC3B,YAAY;IACV,IAAIF,YAAA,EAAc;MAChBD,OAAA,CAAQI,IAAA,CAAK,gCAAgC;MAC7C,MAAMC,OAAA,GAAS,MAAMJ,YAAA,CAAaK,OAAA,CAAQ;MAC1C,OAAAN,OAAA,CAAQI,IAAA,CAAK,mBAAmB,GACzBC,OAAA;IACT;IAEA,IAAI,CAACzB,gBAAA,EACH,MAAM,IAAIgB,KAAA,CAAM,+BAA+B;IAYjD,MAAMW,SAAA,GAAY1B,oBAAA,CAAqBD,gBAAA,EATD;MACpCA,gBAAA;MACA4B,uBAAA,EAAyB;MAAA;MAAA;MAGzBC,iBAAA,EAAmB;MACnBC,eAAA,EAAiB;IACnB,CAEuE;IAEvEV,OAAA,CAAQI,IAAA,CAAK,sBAAsBxB,gBAAgB,KAAK;IACxD,MAAMiB,MAAA,GAAS,MAAMU,SAAA,CAAUrB,IAAA,CAAKoB,OAAA,CAAQ;IAC5C,OAAAN,OAAA,CAAQI,IAAA,CAAK,mBAAmB,GACzBP,MAAA;EACT,GACA;IACEF,OAAA;IACAgB,UAAA,EAAY;IACZC,MAAA,EAAQ;IACRC,UAAA,EAAY;EACd,CACF;AAGF;AAEA,eAAsBC,QACpBC,SAAA,EACAC,MAAA,EACAtB,OAAA,GAA8B,CAAC,GACD;EAC9B,IAAIG,MAAA,GAA4B;EAEhC,IAAI;IACF,OAAAA,MAAA,GAAS,MAAMC,yBAAA,CACbJ,OAAA,CAAQR,IAAA,EACRQ,OAAA,CAAQd,gBAAA,EACRc,OAAA,CAAQC,OAAA,IAAW,CACrB,GACO,MAAME,MAAA,CAAOoB,KAAA,CAAMF,SAAA,EAAWC,MAAM;EAC7C,SAASjB,KAAA,EAAO;IACd,MAAAC,OAAA,CAAQD,KAAA,CAAM,0BAA0B;MACtCkB,KAAA,EAAOF,SAAA;MACPhB,KAAA,EAAOA,KAAA,YAAiBH,KAAA,GAAQG,KAAA,CAAMmB,OAAA,GAAUC,MAAA,CAAOpB,KAAK;IAC9D,CAAC,GACKA,KAAA;EACR,UAAE;IACA,IAAIF,MAAA,IAAUH,OAAA,CAAQd,gBAAA;MAMpB,IAL2B,MAAMwC,0BAAA,CAC/BvB,MAAA,EACAH,OAAA,CAAQd,gBACV,GAEwB;QACtB,MAAM2B,SAAA,GAAY9B,KAAA,CAAMe,GAAA,CAAIb,aAAA,CAAce,OAAA,CAAQd,gBAAgB,CAAC;QACnEiB,MAAA,CAAOwB,OAAA,CAAQ,GACf,MAAMd,SAAA,EAAWrB,IAAA,CAAKoC,GAAA,CAAI,GACtBf,SAAA,IACF9B,KAAA,CAAM8C,MAAA,CAAO5C,aAAA,CAAce,OAAA,CAAQd,gBAAgB,CAAC;MAExD,OACEiB,MAAA,CAAOwB,OAAA,CAAQ;IAAA,OAERxB,MAAA,IACTA,MAAA,CAAOwB,OAAA,CAAQ;EAEnB;AACF;AAEA,eAAeD,2BACbvB,MAAA,EACAjB,gBAAA,EACkB;EAClB,MAAM2B,SAAA,GAAY9B,KAAA,CAAMe,GAAA,CAAIb,aAAA,CAAcC,gBAAgB,CAAC;EAC3D,IAAI,CAAC2B,SAAA,EAAW,OAAO;EAEvB,MAAMiB,WAAA,GAAcC,IAAA,CAAKC,GAAA,CAAI;IACvBC,uBAAA,GAA0B;IAC1BC,uBAAA,GAA0B;EAEhC,IAAIrB,SAAA,CAAUnB,cAAA,KAAmB,QAAQmB,SAAA,CAAUlB,mBAAA,KAAwB,MAAM;IAC/E,MAAM,CAACD,cAAA,EAAgBC,mBAAmB,IAAI,MAAMwC,mBAAA,CAAoBhC,MAAM;IAC9EG,OAAA,CAAQI,IAAA,CACN,oBAAoBhB,cAAc,2BAA2BC,mBAAmB,EAClF,GACAkB,SAAA,CAAUnB,cAAA,GAAiBA,cAAA,EAC3BmB,SAAA,CAAUlB,mBAAA,GAAsBA,mBAAA;EAClC;EAEA,IACEkB,SAAA,CAAUjB,iBAAA,KAAsB,QAChCiB,SAAA,CAAUhB,2BAAA,KAAgC,QAC1CiC,WAAA,GAAcjB,SAAA,CAAUhB,2BAAA,GAA8BoC,uBAAA,EACtD;IACA,MAAMrC,iBAAA,GAAoB,MAAMwC,oBAAA,CAAqBjC,MAAA,EAAQjB,gBAAgB;IAC7E2B,SAAA,CAAUjB,iBAAA,GAAoBA,iBAAA,EAC9BiB,SAAA,CAAUhB,2BAAA,GAA8BiC,WAAA;EAC1C;EAEA,OACEjB,SAAA,CAAUjB,iBAAA,IACTiB,SAAA,CAAUnB,cAAA,GAAiBmB,SAAA,CAAUlB,mBAAA,IAAuBuC,uBAAA,IAE7D5B,OAAA,CAAQ+B,IAAA,CACN,kCAAkCxB,SAAA,CAAUjB,iBAAiB,IAAIiB,SAAA,CAAUnB,cAAA,GAAiBmB,SAAA,CAAUlB,mBAAmB,EAC3H,GACO,MAGF;AACT;AAEA,eAAewC,oBAAoBhC,MAAA,EAA+C;EAChFG,OAAA,CAAQI,IAAA,CAAK,8BAA8B;EAC3C,MAAM4B,oBAAA,GAAuB,MAAMnC,MAAA,CAAOoB,KAAA,CAAM,sBAAsB;IAChEgB,wBAAA,GAA2B,MAAMpC,MAAA,CAAOoB,KAAA,CAC5C,qCACF;EAEA,OAAO,CACLiB,MAAA,CAAOC,QAAA,CAASH,oBAAA,CAAqBI,IAAA,CAAK,CAAC,EAAEC,eAAA,EAAiB,EAAE,GAChEH,MAAA,CAAOC,QAAA,CAASF,wBAAA,CAAyBG,IAAA,CAAK,CAAC,EAAEE,8BAAA,EAAgC,EAAE,EACrF;AACF;AAEA,eAAeR,qBACbjC,MAAA,EACAjB,gBAAA,EACiB;EAEjB,MAAM2D,MAAA,GAAS,IAAIC,GAAA,CAAI5D,gBAAgB,EAAE6D,QAAA,CAASC,KAAA,CAAM,CAAC;EACzD1C,OAAA,CAAQI,IAAA,CAAK,+BAA+B;EAC5C,MAAMuC,qBAAA,GAAwB,MAAM9C,MAAA,CAAOoB,KAAA,CACzC,qFACA,CAACsB,MAAM,CACT;IACMK,MAAA,GAASV,MAAA,CAAOC,QAAA,CACpBQ,qBAAA,CAAsBP,IAAA,CAAK,CAAC,GAAGS,kBAAA,IAAsB,GACrD,EACF;EACA,OAAA7C,OAAA,CAAQI,IAAA,CAAK,uBAAuBwC,MAAM,EAAE,GACrCA,MAAA;AACT;AAEA,eAAsBE,aAAapD,OAAA,GAA8B,CAAC,GAAuB;EACvF,MAAM;IAAEd;EAAiB,IAAIc,OAAA;EAE7B,IAAI,CAACd,gBAAA,EACH,MAAM,IAAIgB,KAAA,CAAM,+CAA+C;EAGjE,IAAI;IAEF,OADe,MAAMmD,iBAAA,CAAkBnE,gBAAgB;EAEzD,SAASmB,KAAA,EAAO;IACd,MAAAC,OAAA,CAAQD,KAAA,CAAM,6BAA6BA,KAAK,GAC1CA,KAAA;EACR;AACF;AAEA,eAAegD,kBAAkBnE,gBAAA,EAA8C;EAC7E,MAAMoE,cAAA,GAAgC;MACpCpE,gBAAA;MACA4B,uBAAA,EAAyB;MACzBC,iBAAA,EAAmB;MACnBC,eAAA,EAAiB;IACnB;IAEM;MAAER,OAAA,EAASC;IAAM,IAAI,MAAM,OAAO,aAAa;EAerD,OAde,MAAMA,KAAA,CACnB,YAAY;IACV,MAAM8C,SAAA,GAAY,IAAIzE,EAAA,CAAG0E,MAAA,CAAOF,cAAc;IAC9C,aAAMC,SAAA,CAAU3C,OAAA,CAAQ,GACjB2C,SAAA;EACT,GACA;IACEtD,OAAA,EAAS;IACTgB,UAAA,EAAY;IACZC,MAAA,EAAQ;IACRC,UAAA,EAAY;EACd,CACF;AAGF","ignoreList":[]}

package/dist/esm/helpers/getDBClient.native.js

@@ -0,0 +1,150 @@
+import pg from "pg";
+function _instanceof(left, right) {
+  return right != null && typeof Symbol < "u" && right[Symbol.hasInstance] ? !!right[Symbol.hasInstance](left) : left instanceof right;
+}
+var cache = /* @__PURE__ */new Map(),
+  createPoolKey = function (connectionString) {
+    return connectionString;
+  },
+  getOrCreatePoolCache = function (connectionString, config) {
+    var key = createPoolKey(connectionString);
+    return cache.has(key) || cache.set(key, {
+      pool: new pg.Pool(config),
+      maxConnections: null,
+      reservedConnections: null,
+      openedConnections: null,
+      openedConnectionsLastUpdate: null
+    }), cache.get(key);
+  };
+async function getDBClient() {
+  var options = arguments.length > 0 && arguments[0] !== void 0 ? arguments[0] : {},
+    {
+      pool,
+      connectionString,
+      retries = 8
+    } = options;
+  if (!pool && !connectionString) throw new Error("Either pool or connectionString must be provided");
+  var client = null;
+  try {
+    return client = await tryToGetNewClientFromPool(pool, connectionString, retries), client;
+  } catch (error) {
+    throw console.error("Failed to get DB client:", error), error;
+  }
+}
+async function tryToGetNewClientFromPool(providedPool, connectionString, retries) {
+  var {
+      default: retry
+    } = await import("async-retry"),
+    clientFromPool = await retry(async function () {
+      if (providedPool) {
+        console.info("Connecting to provided pool...");
+        var client = await providedPool.connect();
+        return console.info("Connected to pool"), client;
+      }
+      if (!connectionString) throw new Error("No connection string provided");
+      var configurations = {
+          connectionString,
+          connectionTimeoutMillis: 5e3,
+          // idle_session_timeout set to 35s on server, client timeout at 30s
+          // fix via https://github.com/brianc/node-postgres/issues/2718#issuecomment-2094885323
+          idleTimeoutMillis: 3e4,
+          allowExitOnIdle: !0
+        },
+        poolCache = getOrCreatePoolCache(connectionString, configurations);
+      console.info(`Connecting to pool ${connectionString}...`);
+      var client1 = await poolCache.pool.connect();
+      return console.info("Connected to pool"), client1;
+    }, {
+      retries,
+      minTimeout: 300,
+      factor: 2,
+      maxTimeout: 8e3
+    });
+  return clientFromPool;
+}
+async function queryDb(queryText, params) {
+  var options = arguments.length > 2 && arguments[2] !== void 0 ? arguments[2] : {},
+    client = null;
+  try {
+    return client = await tryToGetNewClientFromPool(options.pool, options.connectionString, options.retries || 8), await client.query(queryText, params);
+  } catch (error) {
+    throw console.error("Database query failed:", {
+      query: queryText,
+      error: _instanceof(error, Error) ? error.message : String(error)
+    }), error;
+  } finally {
+    if (client && options.connectionString) {
+      var tooManyConnections = await checkForTooManyConnections(client, options.connectionString);
+      if (tooManyConnections) {
+        var poolCache = cache.get(createPoolKey(options.connectionString));
+        client.release(), await poolCache?.pool.end(), poolCache && cache.delete(createPoolKey(options.connectionString));
+      } else client.release();
+    } else client && client.release();
+  }
+}
+async function checkForTooManyConnections(client, connectionString) {
+  var poolCache = cache.get(createPoolKey(connectionString));
+  if (!poolCache) return !1;
+  var currentTime = Date.now(),
+    openedConnectionsMaxAge = 1e4,
+    maxConnectionsTolerance = 0.9;
+  if (poolCache.maxConnections === null || poolCache.reservedConnections === null) {
+    var [maxConnections, reservedConnections] = await getConnectionLimits(client);
+    console.info(`Max connections: ${maxConnections}, Reserved connections: ${reservedConnections}`), poolCache.maxConnections = maxConnections, poolCache.reservedConnections = reservedConnections;
+  }
+  if (poolCache.openedConnections === null || poolCache.openedConnectionsLastUpdate === null || currentTime - poolCache.openedConnectionsLastUpdate > openedConnectionsMaxAge) {
+    var openedConnections = await getOpenedConnections(client, connectionString);
+    poolCache.openedConnections = openedConnections, poolCache.openedConnectionsLastUpdate = currentTime;
+  }
+  return poolCache.openedConnections > (poolCache.maxConnections - poolCache.reservedConnections) * maxConnectionsTolerance ? (console.warn(`Too many connections detected: ${poolCache.openedConnections}/${poolCache.maxConnections - poolCache.reservedConnections}`), !0) : !1;
+}
+async function getConnectionLimits(client) {
+  console.info("Getting connection limits...");
+  var maxConnectionsResult = await client.query("SHOW max_connections"),
+    reservedConnectionResult = await client.query("SHOW superuser_reserved_connections");
+  return [Number.parseInt(maxConnectionsResult.rows[0].max_connections, 10), Number.parseInt(reservedConnectionResult.rows[0].superuser_reserved_connections, 10)];
+}
+async function getOpenedConnections(client, connectionString) {
+  var _openConnectionsResult_rows_,
+    dbName = new URL(connectionString).pathname.slice(1);
+  console.info("Getting opened connections...");
+  var openConnectionsResult = await client.query("SELECT numbackends as opened_connections FROM pg_stat_database WHERE datname = $1", [dbName]),
+    result = Number.parseInt(((_openConnectionsResult_rows_ = openConnectionsResult.rows[0]) === null || _openConnectionsResult_rows_ === void 0 ? void 0 : _openConnectionsResult_rows_.opened_connections) || 0, 10);
+  return console.info(`Opened connections: ${result}`), result;
+}
+async function getNewClient() {
+  var options = arguments.length > 0 && arguments[0] !== void 0 ? arguments[0] : {},
+    {
+      connectionString
+    } = options;
+  if (!connectionString) throw new Error("connectionString is required for getNewClient");
+  try {
+    var client = await tryToGetNewClient(connectionString);
+    return client;
+  } catch (error) {
+    throw console.error("Failed to get new client:", error), error;
+  }
+}
+async function tryToGetNewClient(connectionString) {
+  var configurations = {
+      connectionString,
+      connectionTimeoutMillis: 5e3,
+      idleTimeoutMillis: 3e4,
+      allowExitOnIdle: !0
+    },
+    {
+      default: retry
+    } = await import("async-retry"),
+    client = await retry(async function () {
+      var newClient = new pg.Client(configurations);
+      return await newClient.connect(), newClient;
+    }, {
+      retries: 10,
+      minTimeout: 100,
+      factor: 2,
+      maxTimeout: 5e3
+    });
+  return client;
+}
+export { getDBClient, getNewClient, queryDb };
+//# sourceMappingURL=getDBClient.native.js.map

package/dist/esm/helpers/getDBClient.native.js.map

@@ -0,0 +1 @@
+{"version":3,"names":["pg","_instanceof","left","right","Symbol","hasInstance","cache","Map","createPoolKey","connectionString","getOrCreatePoolCache","config","key","has","set","pool","Pool","maxConnections","reservedConnections","openedConnections","openedConnectionsLastUpdate","get","getDBClient","options","arguments","length","retries","Error","client","tryToGetNewClientFromPool","error","console","providedPool","default","retry","clientFromPool","info","connect","configurations","connectionTimeoutMillis","idleTimeoutMillis","allowExitOnIdle","poolCache","client1","minTimeout","factor","maxTimeout","queryDb","queryText","params","query","message","String","tooManyConnections","checkForTooManyConnections","release","end","delete","currentTime","Date","now","openedConnectionsMaxAge","maxConnectionsTolerance","getConnectionLimits","getOpenedConnections","warn","maxConnectionsResult","reservedConnectionResult","Number","parseInt","rows","max_connections","superuser_reserved_connections","_openConnectionsResult_rows_","dbName","URL","pathname","slice","openConnectionsResult","result","opened_connections","getNewClient","tryToGetNewClient","newClient","Client"],"sources":["../../../src/helpers/getDBClient.ts"],"sourcesContent":[null],"mappings":"AAAA,OAAOA,EAAA,MAAwC;AAY/C,SAAMC,WAAQA,CAAAC,IAAA,EAAAC,KAAA,EAAI;EAchB,OAAMA,KAAM,mBAAcC,MAAA,MAAgB,IAAAD,KAAA,CAAAC,MAAA,CAAAC,WAAA,MAAAF,KAAA,CAAAC,MAAA,CAAAC,WAAA,EAAAH,IAAA,IAAAA,IAAA,YAAAC,KAAA;AAE1C;AACiB,IACbG,KAAA,GAAM,eAAkB,IAAAC,GAAA;EAAAC,aAAA,YAAAA,CAAAC,gBAAA;IAAA,OACxBA,gBAAgB;EAAA;EAAAC,oBAChB,GAAqB,SAAAA,CAAAD,gBAAA,EAAAE,MAAA;IAAA,IACrBC,GAAA,GAAAJ,aAAmB,CAAAC,gBAAA;IAAA,OACnBH,KAAA,CAAAO,GAAA,CAAAD,GAAA,KAAAN,KAAA,CAA6BQ,GAAA,CAAAF,GAAA;MAC9BG,IAGI,MAAMf,EAAI,CAAAgB,IAAG,CAAAL,MAAA;MACtBM,cAAA;MAEAC,mBAAsB,MAAY;MAChCC,iBAAc;MAEdC,2BAAc;IACZ,IAAAd,KAAM,CAAAe,GAAI,CAAAT,GAAA,CAAM;EAGlB;AAEA,eAAIU,YAAA;EACF,IAAAC,OAAA,GAAAC,SAAS,CAAMC,MAAA,QAAAD,SAAA,GAA0B,UAAM,IAAAA,SAAA,CAAkB,OAAO;IAAA;MACjET,IAAA;MAAAN,gBAAA;MAAAiB,OAAA;IAAA,IAAAH,OAAA;EACT,KAAAR,IAAS,KAAAN,gBAAO,EACd,UAAAkB,KAAQ,mDACF;EACR,IAAAC,MAAA;EACF;IAEA,OAAAA,MAAe,SAAAC,yBACb,CAAAd,IACA,EAAAN,gBACA,EAAAiB,OACqB,GAAAE,MAAA;EACrB,SAAQE,KAAA;IAsCR,MArCuBC,OAAM,CAAAD,KAAA,6BAAAA,KAAA,GAAAA,KAAA;EAAA;AAEzB;AACE,eAAAD,yBAAaA,CAAAG,YAAgC,EAAAvB,gBAAA,EAAAiB,OAAA;EAC7C;MAAAO,OAAM,EAAAC;IAAA,CAAS,SAAM,oBAAqB;IAAAC,cAAA,SAAAD,KAAA;MAC1C,IAAAF,YAAA;QAEFD,OAAA,CAAAK,IAAA;QAEA,IAAIR,MAAC,SAAAI,YAAA,CAAAK,OAAA;QACH,OAAAN,OAAU,CAAAK,IAAM,uBAAAR,MAAA;MAYlB;MATsC,IACpC,CAAAnB,gBAAA,QACA,IAAAkB,KAAA,gCAAyB;MAAA,IAAAW,cAAA;UAAA7B,gBAAA;UAAA8B,uBAGN;UAAA;UAErB;UAIAC,iBAAa;UACbC,eAAe;QACf;QAAAC,SAAA,GAAAhC,oBAAa,CAAAD,gBACN,EAAA6B,cAAA;MACTP,OAAA,CAAAK,IAAA,uBAAA3B,gBAAA;MACA,IAAAkC,OAAA,SAAAD,SAAA,CAAA3B,IAAA,CAAAsB,OAAA;MAAA,OACEN,OAAA,CAAAK,IAAA,uBAAAO,OAAA;IAAA;MACYjB,OACZ;MAAQkB,UACR,KAAY;MACdC,MAAA;MACFC,UAAA;IAGF;EAEA,OAAAX,cACE;AAIA;AAEA,eAAIY,QAAAC,SAAA,EAAAC,MAAA;EACF,IAAA1B,OAAA,GAAAC,SAAS,CAAMC,MAAA,QAAAD,SAAA,iBAAAA,SAAA;IAAAI,MAAA;EAAA,IACb;IAAQ,OACRA,MAAQ,SAAAC,yBAAA,CAAAN,OAAA,CAAAR,IAAA,EAAAQ,OAAA,CAAAd,gBAAA,EAAAc,OAAA,CAAAG,OAAA,cAAAE,MAAA,CAAAsB,KAAA,CAAAF,SAAA,EAAAC,MAAA;EAAA,SACRnB,KAAQ;IACV,MACOC,OAAM,CAAAD,KAAO,yBAAuB;MAC7CoB,KAAS,EAAAF,SAAO;MACdlB,KAAA,EAAA7B,WAAc,CAAA6B,KAAA,EAAAH,KAAA,IAAAG,KAAA,CAAAqB,OAA0B,GAAAC,MAAA,CAAAtB,KAAA;IAAA,EACtC,EAAAA,KAAO;EAAA,UACP;IACF,IACMF,MAAA,IAAAL,OAAA,CAAAd,gBAAA;MACR,IAAA4C,kBAAE,SAAAC,0BAAA,CAAA1B,MAAA,EAAAL,OAAA,CAAAd,gBAAA;MACA,IAAI4C,kBAAkB;QAMpB,IAL2BX,SAAM,GAAApC,KAAA,CAAAe,GAAA,CAAAb,aAAA,CAAAe,OAAA,CAAAd,gBAAA;QAC/BmB,MAAA,CAAA2B,OAAA,UAAAb,SAAA,EAAA3B,IAAA,CAAAyC,GAAA,IAAAd,SAAA,IAAApC,KAAA,CAAAmD,MAAA,CAAAjD,aAAA,CAAAe,OAAA,CAAAd,gBAAA;MAAA,OAEFmB,MAEwB,CAAA2B,OAAA;IACtB,OAAA3B,MAAM,IAAAA,MAAY,CAAA2B,OAAM,CAAI;EAC5B;AAGsD;AAGtD,eAAOD,0BAAQA,CAAA1B,MAAA,EAAAnB,gBAAA;EAAA,IAAAiC,SAER,GAAApC,KACT,CAAAe,GAAO,CAAAb,aAAQ,CAAAC,gBAAA;EAEnB,KAAAiC,SAAA;EACF,IAAAgB,WAAA,GAAAC,IAAA,CAAAC,GAAA;IAAAC,uBAAA;IAAAC,uBAAA;EAEA,IAAApB,SAAe,CAAAzB,cAAA,aACbyB,SACA,CAAAxB,mBACkB;IAClB,IAAM,CAAAD,cAAY,EAAAC,mBAAwB,UAAA6C,mBAAiB,CAAAnC,MAAA;IAC3DG,OAAK,CAAAK,IAAA,CAAW,oBAAOnB,cAAA,2BAAAC,mBAAA,KAAAwB,SAAA,CAAAzB,cAAA,GAAAA,cAAA,EAAAyB,SAAA,CAAAxB,mBAAA,GAAAA,mBAAA;EAEvB;EAIA,IAAIwB,SAAA,CAAUvB,iBAAA,KAAmB,QAAQuB,SAAU,CAAAtB,2BAA8B,aAAAsC,WAAA,GAAAhB,SAAA,CAAAtB,2BAAA,GAAAyC,uBAAA;IAC/E,IAAA1C,iBAAO,GAAgB,MAAA6C,oBAAuB,CAAApC,MAAM,EAAAnB,gBAAoB;IACxEiC,SAAQ,CAAAvB,iBAAA,GAAAA,iBAAA,EAAAuB,SAAA,CAAAtB,2BAAA,GAAAsC,WAAA;EAAA;EAC0E,OAElFhB,SAAU,CAAAvB,iBAAiB,IAAAuB,SAAA,CAC3BzB,cAAU,GAAAyB,SAAA,CAAAxB,mBAAsB,IAAA4C,uBAAA,IAAA/B,OAAA,CAAAkC,IAAA,mCAAAvB,SAAA,CAAAvB,iBAAA,IAAAuB,SAAA,CAAAzB,cAAA,GAAAyB,SAAA,CAAAxB,mBAAA;AAAA;AAGlC,eACE6C,mBAAUA,CAAAnC,MAAsB;EAIhCG,OAAA,CAAMK,IAAA,+BAA0B;EAChC,IAAA8B,oBAAU,SAAoBtC,MAAA,CAAAsB,KAAA,uBACpB;IAAAiB,wBAA8B,SAAAvC,MAAA,CAAAsB,KAAA;EAC1C,QAEAkB,MACE,CAAAC,QAAU,CAAAH,oBACT,CAAAI,IAAA,IAAAC,eAAU,IAAiB,GAG1BH,MAAA,CAAAC,QAAA,CAAAF,wBAAkC,CAAAG,IAAU,IAAAE,8BAA+B,MAC7E;AAKJ;AAEA,eAAeR,oBAAoBA,CAAApC,MAAA,EAA+CnB,gBAAA;EAChF,IAAAgE,4BAAa;IAAAC,MAAA,GAA8B,IAAAC,GAAA,CAAAlE,gBAAA,EAAAmE,QAAA,CAAAC,KAAA;EAC3C9C,OAAM,CAAAK,IAAA,gCAAoC;EACI,IAC5C0C,qBAAA,SAAAlD,MAAA,CAAAsB,KAAA,uFACFwB,MAAA,CAEA;IAAAK,MAAO,GAAAX,MAAA,CAAAC,QAAA,GAAAI,4BAAA,GAAAK,qBAAA,CAAAR,IAAA,iBAAAG,4BAAA,uBAAAA,4BAAA,CAAAO,kBAAA;EAAA,OACLjD,OAAO,CAAAK,IAAS,wBAAqB2C,MAAQ,KAAAA,MAAA;AAAmB;AACmB,eACrFE,aAAA;EACF,IAAA1D,OAAA,GAAAC,SAAA,CAAAC,MAAA,QAAAD,SAAA,iBAAAA,SAAA;IAAA;MAAAf;IAAA,IAAAc,OAAA;EAEA,KAAAd,gBAAe,EAKb,MAAM,IAAAkB,KAAS,gDAA0C;EACzD;IACA,IAAMC,MAAA,SAAAsD,iBAA8B,CAAAzE,gBAAO;IACzC,OAAAmB,MAAA;EAAA,EACA,OAAOE,KAAA;IACT,MACMC,OAAS,CAAAD,KAAO,8BAAAA,KAAA,GAAAA,KAAA;EAAA;AACiC;AACrD,eACFoD,kBAAAzE,gBAAA;EACA,IAAA6B,cAAQ,GAAK;MAEf7B,gBAAA;MAEA8B,uBAAsB,EAAa;MACjCC,iBAAQ;MAERC,eAAK;IACH;IAAA;MAAAR,OAAU,EAAAC;IAAM;IAAAN,MAAA,SAA+CM,KAAA;MAGjE,IAAIiD,SAAA,OAAAnF,EAAA,CAAAoF,MAAA,CAAA9C,cAAA;MAEF,OADe,MAAM6C,SAAA,CAAA9C,OAAA,CAAkB,GAAA8C,SAAA;IAEzC;MACEzD,OAAA;MAEFkB,UAAA;MACFC,MAAA;MAEAC,UAAA,EAAe;IACb;EAAsC,OACpClB,MAAA;AAAA;AACyB,SACNN,WACnB,EACF2D,YAEQ,EAeRlC,OAde","ignoreList":[]}

package/dist/esm/index.js

@@ -0,0 +1,29 @@
+import { drizzle } from "drizzle-orm/node-postgres";
+import pg from "pg";
+import { createSql } from "./sql";
+import { createServerHelpers } from "./createServerHelpers";
+import {
+  getDBClient,
+  queryDb
+} from "./helpers/getDBClient";
+import { processInChunks, updateInChunks } from "./helpers/chunkedQuery";
+const createPool = (connectionString) => new pg.Pool({
+  connectionString
+}), createDb = (connectionString, schema) => {
+  const pool = createPool(connectionString);
+  return drizzle(pool, {
+    schema,
+    logger: !1
+  });
+};
+export {
+  createDb,
+  createPool,
+  createServerHelpers,
+  createSql,
+  getDBClient,
+  processInChunks,
+  queryDb,
+  updateInChunks
+};
+//# sourceMappingURL=index.js.map

package/dist/esm/index.js.map

@@ -0,0 +1,6 @@
+{
+  "version": 3,
+  "sources": ["../../src/index.ts"],
+  "mappings": "AACA,SAAS,eAAe;AAExB,OAAO,QAAQ;AAsBf,SAAS,iBAAgC;AAEzC,SAAS,2BAA+C;AAExD;AAAA,EACE;AAAA,EACA;AAAA,OAEK;AAEP,SAAS,iBAAiB,sBAAsB;AA9BzC,MAAM,aAAa,CAAC,qBAClB,IAAI,GAAG,KAAK;AAAA,EACjB;AACF,CAAC,GAGU,WAAW,CACtB,kBACA,WAC4B;AAC5B,QAAM,OAAO,WAAW,gBAAgB;AACxC,SAAO,QAAQ,MAAM;AAAA,IACnB;AAAA,IACA,QAAQ;AAAA,EACV,CAAC;AACH;",
+  "names": []
+}

package/dist/esm/index.mjs

@@ -0,0 +1,18 @@
+import { drizzle } from "drizzle-orm/node-postgres";
+import pg from "pg";
+import { createSql } from "./sql.mjs";
+import { createServerHelpers } from "./createServerHelpers.mjs";
+import { getDBClient, queryDb } from "./helpers/getDBClient.mjs";
+import { processInChunks, updateInChunks } from "./helpers/chunkedQuery.mjs";
+const createPool = connectionString => new pg.Pool({
+    connectionString
+  }),
+  createDb = (connectionString, schema) => {
+    const pool = createPool(connectionString);
+    return drizzle(pool, {
+      schema,
+      logger: !1
+    });
+  };
+export { createDb, createPool, createServerHelpers, createSql, getDBClient, processInChunks, queryDb, updateInChunks };
+//# sourceMappingURL=index.mjs.map

package/dist/esm/index.mjs.map

@@ -0,0 +1 @@
+{"version":3,"names":["drizzle","pg","createSql","createServerHelpers","getDBClient","queryDb","processInChunks","updateInChunks","createPool","connectionString","Pool","createDb","schema","pool","logger"],"sources":["../../src/index.ts"],"sourcesContent":[null],"mappings":"AACA,SAASA,OAAA,QAAe;AAExB,OAAOC,EAAA,MAAQ;AAsBf,SAASC,SAAA,QAAgC;AAEzC,SAASC,mBAAA,QAA+C;AAExD,SACEC,WAAA,EACAC,OAAA,QAEK;AAEP,SAASC,eAAA,EAAiBC,cAAA,QAAsB;AA9BzC,MAAMC,UAAA,GAAcC,gBAAA,IAClB,IAAIR,EAAA,CAAGS,IAAA,CAAK;IACjBD;EACF,CAAC;EAGUE,QAAA,GAAWA,CACtBF,gBAAA,EACAG,MAAA,KAC4B;IAC5B,MAAMC,IAAA,GAAOL,UAAA,CAAWC,gBAAgB;IACxC,OAAOT,OAAA,CAAQa,IAAA,EAAM;MACnBD,MAAA;MACAE,MAAA,EAAQ;IACV,CAAC;EACH","ignoreList":[]}

package/dist/esm/index.native.js

@@ -0,0 +1,20 @@
+import { drizzle } from "drizzle-orm/node-postgres";
+import pg from "pg";
+import { createSql } from "./sql.native.js";
+import { createServerHelpers } from "./createServerHelpers.native.js";
+import { getDBClient, queryDb } from "./helpers/getDBClient.native.js";
+import { processInChunks, updateInChunks } from "./helpers/chunkedQuery.native.js";
+var createPool = function (connectionString) {
+    return new pg.Pool({
+      connectionString
+    });
+  },
+  createDb = function (connectionString, schema) {
+    var pool = createPool(connectionString);
+    return drizzle(pool, {
+      schema,
+      logger: !1
+    });
+  };
+export { createDb, createPool, createServerHelpers, createSql, getDBClient, processInChunks, queryDb, updateInChunks };
+//# sourceMappingURL=index.native.js.map

package/dist/esm/index.native.js.map

@@ -0,0 +1 @@
+{"version":3,"names":["drizzle","pg","createSql","createServerHelpers","getDBClient","queryDb","processInChunks","updateInChunks","createPool","connectionString","Pool","createDb","schema","pool","logger"],"sources":["../../src/index.ts"],"sourcesContent":[null],"mappings":"AACA,SAASA,OAAA,QAAe;AAExB,OAAOC,EAAA,MAAQ;AAsBf,SAASC,SAAA,QAAgC;AAEzC,SAASC,mBAAA,QAA+C;AAExD,SAAAC,WAAA,EAAAC,OAAA;AAAA,SACEC,eAAA,EAAAC,cAAA;AAAA,IACAC,UAAA,YAAAA,CAAAC,gBAAA;IAAA,OAEK,IAAAR,EAAA,CAAAS,IAAA;MAEPD;IA9BO;EACc;EACjBE,QAAA,YAAAA,CAAAF,gBAAA,EAAAG,MAAA;IACD,IAGUC,IAAA,GAAAL,UACX,CAAAC,gBACA;IAEA,OAAMT,OAAO,CAAAa,IAAA;MACbD,MAAO;MACLE,MAAA;IAAA,EACA;EAAQ;AAEZ,S","ignoreList":[]}

package/dist/esm/migrate.js

@@ -0,0 +1,91 @@
+import { basename } from "node:path";
+import { getDBClient } from "./helpers/getDBClient";
+const isServerless = !!(process.env.AWS_LAMBDA_FUNCTION_NAME || process.env.AWS_LAMBDA_RUNTIME_API || process.env.LAMBDA_RUNTIME_DIR || process.env.IS_SERVERLESS);
+async function migrate(options) {
+  const {
+    connectionString,
+    migrationsGlob,
+    createDatabases = [],
+    onMigrationComplete,
+    gitSha,
+    cvrDb,
+    changeDb
+  } = options;
+  console.info(`Running migrations${gitSha ? ` for git version: ${gitSha}` : ""}`);
+  const client = await getDBClient({ connectionString }), hasDB = async (name) => {
+    const result = !!(await client.query(`
+        SELECT 1 FROM pg_database WHERE datname = '${name}'
+      `)).rows.length;
+    return console.info(result ? `${name} db exists` : `creating ${name} db`), result;
+  };
+  if (cvrDb || changeDb) {
+    if (!cvrDb)
+      throw new Error("Missing cvrDb");
+    const zeroDBNames = [basename(cvrDb || ""), basename(changeDb || "")].filter(Boolean);
+    for (const name of zeroDBNames)
+      await hasDB(name) || await client.query(`CREATE DATABASE ${name};`);
+  }
+  for (const dbUrl of createDatabases) {
+    const name = basename(dbUrl);
+    await hasDB(name) || await client.query(`CREATE DATABASE ${name};`);
+  }
+  try {
+    await client.query("BEGIN"), await client.query(`
+      CREATE TABLE IF NOT EXISTS migrations (
+        id SERIAL PRIMARY KEY,
+        name VARCHAR(255) NOT NULL,
+        run_on TIMESTAMP NOT NULL DEFAULT NOW()
+      )
+    `);
+    const appliedMigrations = await client.query("SELECT name FROM migrations"), appliedMigrationNames = new Set(appliedMigrations.rows.map((row) => row.name)), tsMigrationsSorted = Object.entries(migrationsGlob).sort(([a], [b]) => a.localeCompare(b)).map(([file, run]) => ({
+      name: basename(file).replace(".ts", ""),
+      run
+    })).filter(({ name }) => /^[\d]+/.test(name));
+    console.info(`Found ${tsMigrationsSorted.length} TypeScript migrations`);
+    const migrations = [...await Promise.all(
+      tsMigrationsSorted.map(async ({ name, run }) => {
+        if (appliedMigrationNames.has(name))
+          return console.info(`TypeScript migration applied already: ${name}`), null;
+        try {
+          return { ...await run(), name };
+        } catch (error) {
+          throw console.error(`Failed to load TypeScript migration ${name}:`, error), error;
+        }
+      })
+    ).then((migrations2) => migrations2.filter(Boolean))].sort((a, b) => a.name.localeCompare(b.name));
+    if (!migrations.length)
+      console.info("No migrations to apply!"), await client.query("COMMIT");
+    else {
+      for (const migration of migrations)
+        console.info(`Migrating: ${migration.name}`), migration.up && (console.info(`Applying migration: ${migration.name}`), await migration.up(client)), await client.query("INSERT INTO migrations (name) VALUES ($1)", [migration.name]), console.info(`Successfully applied migration: ${migration.name}`);
+      await client.query("COMMIT"), console.info("Successfully committed all migrations");
+    }
+  } catch (e) {
+    console.error("Migration failed, rolling back:", e), await client.query("ROLLBACK"), console.info("Releasing client connection...");
+    try {
+      client.release(!1);
+    } catch (releaseErr) {
+      console.error("Error releasing connection after rollback:", releaseErr);
+    }
+    throw e;
+  }
+  onMigrationComplete && await onMigrationComplete(), console.info("Releasing client connection...");
+  try {
+    client.release(!1);
+  } catch (err) {
+    console.error("Error releasing connection gracefully, trying to destroy:", err);
+    try {
+      client.release(!0);
+    } catch (destroyErr) {
+      console.error("Error destroying connection:", destroyErr);
+    }
+  }
+  console.info("\u{1F64C} Done migrating"), exitProcess();
+}
+function exitProcess() {
+  typeof process > "u" || isServerless || process.exit(0);
+}
+export {
+  migrate
+};
+//# sourceMappingURL=migrate.js.map