@bytebase/dbhub 1.3.1 → 1.5.0

package/dist/{chunk-ABTVMHD6.js → chunk-55X2VY5D.js}

@@ -60,7 +60,7 @@ var sqlServerPassThroughPattern = new RegExp(
 var escapeHatchFunctionKeywords = {
   mysql: ["load_file", "get_lock", "release_lock", "release_all_locks"],
   mariadb: ["load_file", "get_lock", "release_lock", "release_all_locks"],
-  postgres: ["pg_read_file", "pg_read_binary_file", "pg_ls_dir"],
+  postgres: ["pg_read_file", "pg_read_binary_file", "pg_ls_dir", "set_config"],
   sqlserver: sqlServerPassThroughKeywords,
   // Oracle packages a SELECT can call to reach outside the database, matched
   // as `package.member(` (see escapeHatchCallSuffix).

package/dist/chunk-5XSBOK3W.js

@@ -0,0 +1,63 @@
+// src/utils/query-timeout.ts
+var CLIENT_QUERY_TIMEOUT_GRACE_MS = 5e3;
+var CANCEL_QUERY_TIMEOUT_MS = 5e3;
+var MAX_TIMER_MS = 2147483647;
+var MAX_QUERY_TIMEOUT_SECONDS = Math.floor(
+  (MAX_TIMER_MS - CLIENT_QUERY_TIMEOUT_GRACE_MS) / 1e3
+);
+function clientQueryTimeoutMs(queryTimeoutMs) {
+  const deadlineMs = queryTimeoutMs + CLIENT_QUERY_TIMEOUT_GRACE_MS;
+  if (!Number.isFinite(deadlineMs) || deadlineMs > MAX_TIMER_MS) {
+    throw new RangeError(
+      `query_timeout of ${queryTimeoutMs}ms exceeds the client-side deadline range (at most ${MAX_QUERY_TIMEOUT_SECONDS} seconds)`
+    );
+  }
+  return deadlineMs;
+}
+var CLIENT_QUERY_TIMEOUT_CODE = "DBHUB_CLIENT_QUERY_TIMEOUT";
+var PG_CLIENT_TIMEOUT_MESSAGE = "Query read timeout";
+var MYSQL2_CLIENT_TIMEOUT_CODE = "PROTOCOL_SEQUENCE_TIMEOUT";
+var ClientQueryTimeoutError = class extends Error {
+  constructor(timeoutMs) {
+    super(`Query timed out on the client after ${timeoutMs}ms`);
+    this.code = CLIENT_QUERY_TIMEOUT_CODE;
+    this.name = "ClientQueryTimeoutError";
+  }
+};
+function isClientSideTimeout(error) {
+  if (typeof error !== "object" || error === null) {
+    return false;
+  }
+  const { code, message } = error;
+  return code === MYSQL2_CLIENT_TIMEOUT_CODE || code === CLIENT_QUERY_TIMEOUT_CODE || code === void 0 && message === PG_CLIENT_TIMEOUT_MESSAGE;
+}
+function withClientQueryDeadline(query, timeoutMs) {
+  if (timeoutMs === void 0) {
+    return query;
+  }
+  return new Promise((resolve, reject) => {
+    const timer = setTimeout(() => {
+      query.catch(() => {
+      });
+      reject(new ClientQueryTimeoutError(timeoutMs));
+    }, timeoutMs);
+    query.then(
+      (value) => {
+        clearTimeout(timer);
+        resolve(value);
+      },
+      (error) => {
+        clearTimeout(timer);
+        reject(error);
+      }
+    );
+  });
+}
+
+export {
+  CANCEL_QUERY_TIMEOUT_MS,
+  MAX_QUERY_TIMEOUT_SECONDS,
+  clientQueryTimeoutMs,
+  isClientSideTimeout,
+  withClientQueryDeadline
+};