@databricks/appkit 0.0.2

package/dist/cache/storage/persistent.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"persistent.js","names":[],"sources":["../../../src/cache/storage/persistent.ts"],"sourcesContent":["import { createHash } from \"node:crypto\";\nimport type { CacheConfig, CacheEntry, CacheStorage } from \"shared\";\nimport type { LakebaseConnector } from \"../../connectors\";\nimport { lakebaseStorageDefaults } from \"./defaults\";\n\n/**\n * Persistent cache storage implementation. Uses a least recently used (LRU) eviction policy\n * to manage memory usage and ensure efficient cache operations.\n *\n * @example\n * const persistentStorage = new PersistentStorage(config, connector);\n * await persistentStorage.initialize();\n * await persistentStorage.get(\"my-key\");\n * await persistentStorage.set(\"my-key\", \"my-value\");\n * await persistentStorage.delete(\"my-key\");\n * await persistentStorage.clear();\n * await persistentStorage.has(\"my-key\");\n *\n */\nexport class PersistentStorage implements CacheStorage {\n  private readonly connector: LakebaseConnector;\n  private readonly tableName: string;\n  private readonly maxBytes: number;\n  private readonly maxEntryBytes: number;\n  private readonly evictionBatchSize: number;\n  private readonly evictionCheckProbability: number;\n  private initialized: boolean;\n\n  constructor(config: CacheConfig, connector: LakebaseConnector) {\n    this.connector = connector;\n    this.maxBytes = config.maxBytes ?? lakebaseStorageDefaults.maxBytes;\n    this.maxEntryBytes =\n      config.maxEntryBytes ?? lakebaseStorageDefaults.maxEntryBytes;\n    this.evictionBatchSize = lakebaseStorageDefaults.evictionBatchSize;\n    this.evictionCheckProbability =\n      config.evictionCheckProbability ??\n      lakebaseStorageDefaults.evictionCheckProbability;\n    this.tableName = lakebaseStorageDefaults.tableName; // hardcoded, safe for now\n    this.initialized = false;\n  }\n\n  /** Initialize the persistent storage and run migrations if necessary */\n  async initialize(): Promise<void> {\n    if (this.initialized) return;\n\n    try {\n      await this.runMigrations();\n      this.initialized = true;\n    } catch (error) {\n      console.error(\"Error in persistent storage initialization:\", error);\n      throw error;\n    }\n  }\n\n  /**\n   * Get a cached value from the persistent storage\n   * @param key - Cache key\n   * @returns Promise of the cached value or null if not found\n   */\n  async get<T>(key: string): Promise<CacheEntry<T> | null> {\n    await this.ensureInitialized();\n\n    const keyHash = this.hashKey(key);\n\n    const result = await this.connector.query<{\n      value: Buffer;\n      expiry: string;\n    }>(`SELECT value, expiry FROM ${this.tableName} WHERE key_hash = $1`, [\n      keyHash,\n    ]);\n\n    if (result.rows.length === 0) return null;\n\n    const entry = result.rows[0];\n\n    // fire-and-forget update\n    this.connector\n      .query(\n        `UPDATE ${this.tableName} SET last_accessed = NOW() WHERE key_hash = $1`,\n        [keyHash],\n      )\n      .catch(() => {\n        console.debug(\"Error updating last_accessed time for key:\", key);\n      });\n\n    return {\n      value: this.deserializeValue<T>(entry.value),\n      expiry: Number(entry.expiry),\n    };\n  }\n\n  /**\n   * Set a value in the persistent storage\n   * @param key - Cache key\n   * @param entry - Cache entry\n   * @returns Promise of the result\n   */\n  async set<T>(key: string, entry: CacheEntry<T>): Promise<void> {\n    await this.ensureInitialized();\n\n    const keyHash = this.hashKey(key);\n    const keyBytes = Buffer.from(key, \"utf-8\");\n    const valueBytes = this.serializeValue(entry.value);\n    const byteSize = keyBytes.length + valueBytes.length;\n\n    if (byteSize > this.maxEntryBytes) {\n      throw new Error(\n        `Cache entry too large: ${byteSize} bytes exceeds maximum of ${this.maxEntryBytes} bytes`,\n      );\n    }\n\n    // probabilistic eviction check\n    if (Math.random() < this.evictionCheckProbability) {\n      const totalBytes = await this.totalBytes();\n      if (totalBytes + byteSize > this.maxBytes) {\n        await this.evictBySize(byteSize);\n      }\n    }\n\n    await this.connector.query(\n      `INSERT INTO ${this.tableName} (key_hash, key, value, byte_size, expiry, created_at, last_accessed)\n      VALUES ($1, $2, $3, $4, $5, NOW(), NOW())\n      ON CONFLICT (key_hash)\n      DO UPDATE SET value = $3, byte_size = $4, expiry = $5, last_accessed = NOW()\n      `,\n      [keyHash, keyBytes, valueBytes, byteSize, entry.expiry],\n    );\n  }\n\n  /**\n   * Delete a value from the persistent storage\n   * @param key - Cache key\n   * @returns Promise of the result\n   */\n  async delete(key: string): Promise<void> {\n    await this.ensureInitialized();\n    const keyHash = this.hashKey(key);\n    await this.connector.query(\n      `DELETE FROM ${this.tableName} WHERE key_hash = $1`,\n      [keyHash],\n    );\n  }\n\n  /** Clear the persistent storage */\n  async clear(): Promise<void> {\n    await this.ensureInitialized();\n    await this.connector.query(`TRUNCATE TABLE ${this.tableName}`);\n  }\n\n  /**\n   * Check if a value exists in the persistent storage\n   * @param key - Cache key\n   * @returns Promise of true if the value exists, false otherwise\n   */\n  async has(key: string): Promise<boolean> {\n    await this.ensureInitialized();\n    const keyHash = this.hashKey(key);\n\n    const result = await this.connector.query<{ exists: boolean }>(\n      `SELECT EXISTS(SELECT 1 FROM ${this.tableName} WHERE key_hash = $1) as exists`,\n      [keyHash],\n    );\n\n    return result.rows[0]?.exists ?? false;\n  }\n\n  /**\n   * Get the size of the persistent storage\n   * @returns Promise of the size of the storage\n   */\n  async size(): Promise<number> {\n    await this.ensureInitialized();\n\n    const result = await this.connector.query<{ count: string }>(\n      `SELECT COUNT(*) as count FROM ${this.tableName}`,\n    );\n    return parseInt(result.rows[0]?.count ?? \"0\", 10);\n  }\n\n  /** Get the total number of bytes in the persistent storage */\n  async totalBytes(): Promise<number> {\n    await this.ensureInitialized();\n\n    const result = await this.connector.query<{ total: string }>(\n      `SELECT COALESCE(SUM(byte_size), 0) as total FROM ${this.tableName}`,\n    );\n    return parseInt(result.rows[0]?.total ?? \"0\", 10);\n  }\n\n  /**\n   * Check if the persistent storage is persistent\n   * @returns true if the storage is persistent, false otherwise\n   */\n  isPersistent(): boolean {\n    return true;\n  }\n\n  /**\n   * Check if the persistent storage is healthy\n   * @returns Promise of true if the storage is healthy, false otherwise\n   */\n  async healthCheck(): Promise<boolean> {\n    try {\n      return await this.connector.healthCheck();\n    } catch {\n      return false;\n    }\n  }\n\n  /** Close the persistent storage */\n  async close(): Promise<void> {\n    await this.connector.close();\n  }\n\n  /**\n   * Cleanup expired entries from the persistent storage\n   * @returns Promise of the number of expired entries\n   */\n  async cleanupExpired(): Promise<number> {\n    await this.ensureInitialized();\n    const result = await this.connector.query<{ count: string }>(\n      `WITH deleted as (DELETE FROM ${this.tableName} WHERE expiry < $1 RETURNING *) SELECT COUNT(*) as count FROM deleted`,\n      [Date.now()],\n    );\n    return parseInt(result.rows[0]?.count ?? \"0\", 10);\n  }\n\n  /** Evict entries from the persistent storage by size */\n  private async evictBySize(requiredBytes: number): Promise<void> {\n    const freedByExpiry = await this.cleanupExpired();\n    if (freedByExpiry > 0) {\n      const currentBytes = await this.totalBytes();\n      if (currentBytes + requiredBytes <= this.maxBytes) {\n        return;\n      }\n    }\n\n    await this.connector.query(\n      `DELETE FROM ${this.tableName} WHERE key_hash IN\n      (SELECT key_hash FROM ${this.tableName} ORDER BY last_accessed ASC LIMIT $1)`,\n      [this.evictionBatchSize],\n    );\n  }\n\n  /** Ensure the persistent storage is initialized */\n  private async ensureInitialized(): Promise<void> {\n    if (!this.initialized) {\n      await this.initialize();\n    }\n  }\n\n  /** Generate a 64-bit hash for the cache key using SHA256 */\n  private hashKey(key: string): bigint {\n    if (!key) throw new Error(\"Cache key cannot be empty\");\n    const hash = createHash(\"sha256\").update(key).digest();\n    return hash.readBigInt64BE(0);\n  }\n\n  /** Serialize a value to a buffer */\n  private serializeValue<T>(value: T): Buffer {\n    return Buffer.from(JSON.stringify(value), \"utf-8\");\n  }\n\n  /** Deserialize a value from a buffer */\n  private deserializeValue<T>(buffer: Buffer): T {\n    return JSON.parse(buffer.toString(\"utf-8\")) as T;\n  }\n\n  /** Run migrations for the persistent storage */\n  private async runMigrations(): Promise<void> {\n    try {\n      await this.connector.query(`\n            CREATE TABLE IF NOT EXISTS ${this.tableName} (\n                id BIGSERIAL PRIMARY KEY,\n                key_hash BIGINT NOT NULL,\n                key BYTEA NOT NULL,\n                value BYTEA NOT NULL,\n                byte_size INTEGER NOT NULL,\n                expiry BIGINT NOT NULL,\n                created_at TIMESTAMP NOT NULL DEFAULT NOW(),\n                last_accessed TIMESTAMP NOT NULL DEFAULT NOW()\n            )\n            `);\n\n      // unique index on key_hash for fast lookups\n      await this.connector.query(\n        `CREATE UNIQUE INDEX IF NOT EXISTS idx_${this.tableName}_key_hash ON ${this.tableName} (key_hash);`,\n      );\n\n      // index on expiry for cleanup queries\n      await this.connector.query(\n        `CREATE INDEX IF NOT EXISTS idx_${this.tableName}_expiry ON ${this.tableName} (expiry); `,\n      );\n\n      // index on last_accessed for LRU eviction\n      await this.connector.query(\n        `CREATE INDEX IF NOT EXISTS idx_${this.tableName}_last_accessed ON ${this.tableName} (last_accessed); `,\n      );\n\n      // index on byte_size for monitoring\n      await this.connector.query(\n        `CREATE INDEX IF NOT EXISTS idx_${this.tableName}_byte_size ON ${this.tableName} (byte_size); `,\n      );\n    } catch (error) {\n      console.error(\n        \"Error in running migrations for persistent storage:\",\n        error,\n      );\n      throw error;\n    }\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;AAmBA,IAAa,oBAAb,MAAuD;CASrD,YAAY,QAAqB,WAA8B;AAC7D,OAAK,YAAY;AACjB,OAAK,WAAW,OAAO,YAAY,wBAAwB;AAC3D,OAAK,gBACH,OAAO,iBAAiB,wBAAwB;AAClD,OAAK,oBAAoB,wBAAwB;AACjD,OAAK,2BACH,OAAO,4BACP,wBAAwB;AAC1B,OAAK,YAAY,wBAAwB;AACzC,OAAK,cAAc;;;CAIrB,MAAM,aAA4B;AAChC,MAAI,KAAK,YAAa;AAEtB,MAAI;AACF,SAAM,KAAK,eAAe;AAC1B,QAAK,cAAc;WACZ,OAAO;AACd,WAAQ,MAAM,+CAA+C,MAAM;AACnE,SAAM;;;;;;;;CASV,MAAM,IAAO,KAA4C;AACvD,QAAM,KAAK,mBAAmB;EAE9B,MAAM,UAAU,KAAK,QAAQ,IAAI;EAEjC,MAAM,SAAS,MAAM,KAAK,UAAU,MAGjC,6BAA6B,KAAK,UAAU,uBAAuB,CACpE,QACD,CAAC;AAEF,MAAI,OAAO,KAAK,WAAW,EAAG,QAAO;EAErC,MAAM,QAAQ,OAAO,KAAK;AAG1B,OAAK,UACF,MACC,UAAU,KAAK,UAAU,iDACzB,CAAC,QAAQ,CACV,CACA,YAAY;AACX,WAAQ,MAAM,8CAA8C,IAAI;IAChE;AAEJ,SAAO;GACL,OAAO,KAAK,iBAAoB,MAAM,MAAM;GAC5C,QAAQ,OAAO,MAAM,OAAO;GAC7B;;;;;;;;CASH,MAAM,IAAO,KAAa,OAAqC;AAC7D,QAAM,KAAK,mBAAmB;EAE9B,MAAM,UAAU,KAAK,QAAQ,IAAI;EACjC,MAAM,WAAW,OAAO,KAAK,KAAK,QAAQ;EAC1C,MAAM,aAAa,KAAK,eAAe,MAAM,MAAM;EACnD,MAAM,WAAW,SAAS,SAAS,WAAW;AAE9C,MAAI,WAAW,KAAK,cAClB,OAAM,IAAI,MACR,0BAA0B,SAAS,4BAA4B,KAAK,cAAc,QACnF;AAIH,MAAI,KAAK,QAAQ,GAAG,KAAK,0BAEvB;OADmB,MAAM,KAAK,YAAY,GACzB,WAAW,KAAK,SAC/B,OAAM,KAAK,YAAY,SAAS;;AAIpC,QAAM,KAAK,UAAU,MACnB,eAAe,KAAK,UAAU;;;;SAK9B;GAAC;GAAS;GAAU;GAAY;GAAU,MAAM;GAAO,CACxD;;;;;;;CAQH,MAAM,OAAO,KAA4B;AACvC,QAAM,KAAK,mBAAmB;EAC9B,MAAM,UAAU,KAAK,QAAQ,IAAI;AACjC,QAAM,KAAK,UAAU,MACnB,eAAe,KAAK,UAAU,uBAC9B,CAAC,QAAQ,CACV;;;CAIH,MAAM,QAAuB;AAC3B,QAAM,KAAK,mBAAmB;AAC9B,QAAM,KAAK,UAAU,MAAM,kBAAkB,KAAK,YAAY;;;;;;;CAQhE,MAAM,IAAI,KAA+B;AACvC,QAAM,KAAK,mBAAmB;EAC9B,MAAM,UAAU,KAAK,QAAQ,IAAI;AAOjC,UALe,MAAM,KAAK,UAAU,MAClC,+BAA+B,KAAK,UAAU,kCAC9C,CAAC,QAAQ,CACV,EAEa,KAAK,IAAI,UAAU;;;;;;CAOnC,MAAM,OAAwB;AAC5B,QAAM,KAAK,mBAAmB;EAE9B,MAAM,SAAS,MAAM,KAAK,UAAU,MAClC,iCAAiC,KAAK,YACvC;AACD,SAAO,SAAS,OAAO,KAAK,IAAI,SAAS,KAAK,GAAG;;;CAInD,MAAM,aAA8B;AAClC,QAAM,KAAK,mBAAmB;EAE9B,MAAM,SAAS,MAAM,KAAK,UAAU,MAClC,oDAAoD,KAAK,YAC1D;AACD,SAAO,SAAS,OAAO,KAAK,IAAI,SAAS,KAAK,GAAG;;;;;;CAOnD,eAAwB;AACtB,SAAO;;;;;;CAOT,MAAM,cAAgC;AACpC,MAAI;AACF,UAAO,MAAM,KAAK,UAAU,aAAa;UACnC;AACN,UAAO;;;;CAKX,MAAM,QAAuB;AAC3B,QAAM,KAAK,UAAU,OAAO;;;;;;CAO9B,MAAM,iBAAkC;AACtC,QAAM,KAAK,mBAAmB;EAC9B,MAAM,SAAS,MAAM,KAAK,UAAU,MAClC,gCAAgC,KAAK,UAAU,wEAC/C,CAAC,KAAK,KAAK,CAAC,CACb;AACD,SAAO,SAAS,OAAO,KAAK,IAAI,SAAS,KAAK,GAAG;;;CAInD,MAAc,YAAY,eAAsC;AAE9D,MADsB,MAAM,KAAK,gBAAgB,GAC7B,GAElB;OADqB,MAAM,KAAK,YAAY,GACzB,iBAAiB,KAAK,SACvC;;AAIJ,QAAM,KAAK,UAAU,MACnB,eAAe,KAAK,UAAU;8BACN,KAAK,UAAU,wCACvC,CAAC,KAAK,kBAAkB,CACzB;;;CAIH,MAAc,oBAAmC;AAC/C,MAAI,CAAC,KAAK,YACR,OAAM,KAAK,YAAY;;;CAK3B,AAAQ,QAAQ,KAAqB;AACnC,MAAI,CAAC,IAAK,OAAM,IAAI,MAAM,4BAA4B;AAEtD,SADa,WAAW,SAAS,CAAC,OAAO,IAAI,CAAC,QAAQ,CAC1C,eAAe,EAAE;;;CAI/B,AAAQ,eAAkB,OAAkB;AAC1C,SAAO,OAAO,KAAK,KAAK,UAAU,MAAM,EAAE,QAAQ;;;CAIpD,AAAQ,iBAAoB,QAAmB;AAC7C,SAAO,KAAK,MAAM,OAAO,SAAS,QAAQ,CAAC;;;CAI7C,MAAc,gBAA+B;AAC3C,MAAI;AACF,SAAM,KAAK,UAAU,MAAM;yCACQ,KAAK,UAAU;;;;;;;;;;cAU1C;AAGR,SAAM,KAAK,UAAU,MACnB,yCAAyC,KAAK,UAAU,eAAe,KAAK,UAAU,cACvF;AAGD,SAAM,KAAK,UAAU,MACnB,kCAAkC,KAAK,UAAU,aAAa,KAAK,UAAU,aAC9E;AAGD,SAAM,KAAK,UAAU,MACnB,kCAAkC,KAAK,UAAU,oBAAoB,KAAK,UAAU,oBACrF;AAGD,SAAM,KAAK,UAAU,MACnB,kCAAkC,KAAK,UAAU,gBAAgB,KAAK,UAAU,gBACjF;WACM,OAAO;AACd,WAAQ,MACN,uDACA,MACD;AACD,SAAM"}

package/dist/connectors/index.js

@@ -0,0 +1,6 @@
+import { LakebaseConnector } from "./lakebase/client.js";
+import "./lakebase/index.js";
+import { SQLWarehouseConnector } from "./sql-warehouse/client.js";
+import "./sql-warehouse/index.js";
+
+export {  };

package/dist/connectors/lakebase/client.js

@@ -0,0 +1,348 @@
+import { __toCommonJS } from "../../_virtual/rolldown_runtime.js";
+import { TelemetryManager } from "../../telemetry/telemetry-manager.js";
+import { SpanStatusCode } from "../../telemetry/index.js";
+import { deepMerge } from "../../utils/merge.js";
+import { init_utils, utils_exports } from "../../utils/index.js";
+import { lakebaseDefaults } from "./defaults.js";
+import { randomUUID } from "node:crypto";
+import { ApiClient, Config } from "@databricks/sdk-experimental";
+import pg from "pg";
+
+//#region src/connectors/lakebase/client.ts
+init_utils();
+/**
+* Enterprise-grade connector for Databricks Lakebase
+* @example Simplest - everything from env/context
+* ```typescript
+* const connector = new LakebaseConnector();
+* await connector.query('SELECT * FROM users');
+* ```
+*
+* @example With explicit connection string
+* ```typescript
+* const connector = new LakebaseConnector({
+*   connectionString: 'postgresql://...'
+* });
+* ```
+*/
+var LakebaseConnector = class {
+	constructor(userConfig) {
+		this.name = "lakebase";
+		this.CACHE_BUFFER_MS = 120 * 1e3;
+		this.pool = null;
+		this.credentials = null;
+		this.config = deepMerge(lakebaseDefaults, userConfig);
+		this.connectionConfig = this.parseConnectionConfig();
+		this.telemetry = TelemetryManager.getProvider(this.name, this.config.telemetry);
+		this.telemetryMetrics = {
+			queryCount: this.telemetry.getMeter().createCounter("lakebase.query.count", {
+				description: "Total number of queries executed",
+				unit: "1"
+			}),
+			queryDuration: this.telemetry.getMeter().createHistogram("lakebase.query.duration", {
+				description: "Duration of queries executed",
+				unit: "ms"
+			})
+		};
+		if (this.config.maxPoolSize < 1) throw new Error("maxPoolSize must be at least 1");
+	}
+	/**
+	* Execute a SQL query
+	*
+	* @example
+	* ```typescript
+	* const users = await connector.query('SELECT * FROM users');
+	* const user = await connector.query('SELECT * FROM users WHERE id = $1', [123]);
+	* ```
+	*/
+	async query(sql, params, retryCount = 0) {
+		const startTime = Date.now();
+		return this.telemetry.startActiveSpan("lakebase.query", { attributes: {
+			"db.system": "lakebase",
+			"db.statement": sql.substring(0, 500),
+			"db.retry_count": retryCount
+		} }, async (span) => {
+			try {
+				const result = await (await this.getPool()).query(sql, params);
+				span.setAttribute("db.rows_affected", result.rowCount ?? 0);
+				span.setStatus({ code: SpanStatusCode.OK });
+				return result;
+			} catch (error) {
+				if (this.isAuthError(error)) {
+					span.addEvent("auth_error_retry");
+					await this.rotateCredentials();
+					const result = await (await this.getPool()).query(sql, params);
+					span.setAttribute("db.rows_affected", result.rowCount ?? 0);
+					span.setStatus({ code: SpanStatusCode.OK });
+					return result;
+				}
+				if (this.isTransientError(error) && retryCount < 1) {
+					span.addEvent("transient_error_retry");
+					await new Promise((resolve) => setTimeout(resolve, 100));
+					return await this.query(sql, params, retryCount + 1);
+				}
+				span.recordException(error);
+				span.setStatus({ code: SpanStatusCode.ERROR });
+				throw error;
+			} finally {
+				const duration = Date.now() - startTime;
+				this.telemetryMetrics.queryCount.add(1);
+				this.telemetryMetrics.queryDuration.record(duration);
+				span.end();
+			}
+		});
+	}
+	/**
+	* Execute a transaction
+	*
+	* COMMIT and ROLLBACK are automatically managed by the transaction function.
+	*
+	* @param callback - Callback function to execute within the transaction context
+	* @example
+	* ```typescript
+	* await connector.transaction(async (client) => {
+	*   await client.query('INSERT INTO accounts (name) VALUES ($1)', ['Alice']);
+	*   await client.query('INSERT INTO logs (action) VALUES ($1)', ['Created Alice']);
+	* });
+	* ```
+	*/
+	async transaction(callback, retryCount = 0) {
+		const startTime = Date.now();
+		return this.telemetry.startActiveSpan("lakebase.transaction", { attributes: {
+			"db.system": "lakebase",
+			"db.retry_count": retryCount
+		} }, async (span) => {
+			const client = await (await this.getPool()).connect();
+			try {
+				await client.query("BEGIN");
+				const result = await callback(client);
+				await client.query("COMMIT");
+				span.setStatus({ code: SpanStatusCode.OK });
+				return result;
+			} catch (error) {
+				try {
+					await client.query("ROLLBACK");
+				} catch {}
+				if (this.isAuthError(error)) {
+					span.addEvent("auth_error_retry");
+					client.release();
+					await this.rotateCredentials();
+					const retryClient = await (await this.getPool()).connect();
+					try {
+						await client.query("BEGIN");
+						const result = await callback(retryClient);
+						await client.query("COMMIT");
+						span.setStatus({ code: SpanStatusCode.OK });
+						return result;
+					} catch (retryError) {
+						try {
+							await retryClient.query("ROLLBACK");
+						} catch {}
+						throw retryError;
+					} finally {
+						retryClient.release();
+					}
+				}
+				if (this.isTransientError(error) && retryCount < 1) {
+					span.addEvent("transaction_error_retry");
+					client.release();
+					await new Promise((resolve) => setTimeout(resolve, 100));
+					return await this.transaction(callback, retryCount + 1);
+				}
+				span.recordException(error);
+				span.setStatus({ code: SpanStatusCode.ERROR });
+				throw error;
+			} finally {
+				client.release();
+				const duration = Date.now() - startTime;
+				this.telemetryMetrics.queryCount.add(1);
+				this.telemetryMetrics.queryDuration.record(duration);
+				span.end();
+			}
+		});
+	}
+	/** Check if database connection is healthy */
+	async healthCheck() {
+		return this.telemetry.startActiveSpan("lakebase.healthCheck", {}, async (span) => {
+			try {
+				const healthy = (await this.query("SELECT 1 as result")).rows[0]?.result === 1;
+				span.setAttribute("db.healthy", healthy);
+				span.setStatus({ code: SpanStatusCode.OK });
+				return healthy;
+			} catch {
+				span.setAttribute("db.healthy", false);
+				span.setStatus({ code: SpanStatusCode.ERROR });
+				return false;
+			} finally {
+				span.end();
+			}
+		});
+	}
+	/** Close connection pool (call on shutdown) */
+	async close() {
+		if (this.pool) {
+			await this.pool.end().catch((error) => {
+				console.error("Error closing connection pool:", error);
+			});
+			this.pool = null;
+		}
+		this.credentials = null;
+	}
+	/** Setup graceful shutdown to close connection pools */
+	shutdown() {
+		process.on("SIGTERM", () => this.close());
+		process.on("SIGINT", () => this.close());
+		this.close();
+	}
+	/** Get Databricks workspace client - from config or request context */
+	getWorkspaceClient() {
+		if (this.config.workspaceClient) return this.config.workspaceClient;
+		try {
+			const { getRequestContext } = (init_utils(), __toCommonJS(utils_exports));
+			const { serviceDatabricksClient } = getRequestContext();
+			this.config.workspaceClient = serviceDatabricksClient;
+			return serviceDatabricksClient;
+		} catch (_error) {
+			throw new Error("Databricks workspace client not available. Either pass it in config or use within AppKit request context.");
+		}
+	}
+	/** Get or create connection pool */
+	async getPool() {
+		if (!this.connectionConfig) throw new Error("Lakebase connection not configured. Set PGHOST, PGDATABASE, PGAPPNAME env vars, provide a connectionString, or pass explicit config.");
+		if (!this.pool) {
+			const creds = await this.getCredentials();
+			this.pool = this.createPool(creds);
+		}
+		return this.pool;
+	}
+	/** Create PostgreSQL pool */
+	createPool(credentials) {
+		const { host, database, port, sslMode } = this.connectionConfig;
+		const pool = new pg.Pool({
+			host,
+			port,
+			database,
+			user: credentials.username,
+			password: credentials.password,
+			max: this.config.maxPoolSize,
+			idleTimeoutMillis: this.config.idleTimeoutMs,
+			connectionTimeoutMillis: this.config.connectionTimeoutMs,
+			ssl: sslMode === "require" ? { rejectUnauthorized: true } : false
+		});
+		pool.on("error", (error) => {
+			console.error("Connection pool error:", error.message, { code: error.code });
+		});
+		return pool;
+	}
+	/** Get or fetch credentials with caching */
+	async getCredentials() {
+		const now = Date.now();
+		if (this.credentials && now < this.credentials.expiresAt - this.CACHE_BUFFER_MS) return this.credentials;
+		const username = await this.fetchUsername();
+		const { token, expiresAt } = await this.fetchPassword();
+		this.credentials = {
+			username,
+			password: token,
+			expiresAt
+		};
+		return {
+			username,
+			password: token
+		};
+	}
+	/** Rotate credentials and recreate pool */
+	async rotateCredentials() {
+		this.credentials = null;
+		if (this.pool) {
+			const oldPool = this.pool;
+			this.pool = null;
+			oldPool.end().catch((error) => {
+				console.error("Error closing old connection pool during rotation:", error);
+			});
+		}
+	}
+	/** Fetch username from Databricks */
+	async fetchUsername() {
+		const user = await this.getWorkspaceClient().currentUser.me();
+		if (!user.userName) throw new Error("Failed to get current user from Databricks workspace");
+		return user.userName;
+	}
+	/** Fetch password (OAuth token) from Databricks */
+	async fetchPassword() {
+		const apiClient = new ApiClient(new Config({ host: this.getWorkspaceClient().config.host }));
+		if (!this.connectionConfig.appName) throw new Error(`Database app name not found in connection config`);
+		const credentials = await apiClient.request({
+			path: `/api/2.0/database/credentials`,
+			method: "POST",
+			headers: new Headers(),
+			raw: false,
+			payload: {
+				instance_names: [this.connectionConfig.appName],
+				request_id: randomUUID()
+			}
+		});
+		if (!this.validateCredentials(credentials)) throw new Error(`Failed to generate credentials for instance: ${this.connectionConfig.appName}`);
+		const expiresAt = new Date(credentials.expiration_time).getTime();
+		return {
+			token: credentials.token,
+			expiresAt
+		};
+	}
+	/** Check if error is auth failure */
+	isAuthError(error) {
+		return typeof error === "object" && error !== null && "code" in error && error.code === "28P01";
+	}
+	/** Check if error is transient */
+	isTransientError(error) {
+		if (typeof error !== "object" || error === null || !("code" in error)) return false;
+		const code = error.code;
+		return code === "ECONNRESET" || code === "ECONNREFUSED" || code === "ETIMEDOUT" || code === "57P01" || code === "57P03" || code === "08006" || code === "08003" || code === "08000";
+	}
+	/** Type guard for credentials */
+	validateCredentials(value) {
+		if (typeof value !== "object" || value === null) return false;
+		const credentials = value;
+		return "token" in credentials && typeof credentials.token === "string" && "expiration_time" in credentials && typeof credentials.expiration_time === "string" && new Date(credentials.expiration_time).getTime() > Date.now();
+	}
+	/** Parse connection configuration from config or environment */
+	parseConnectionConfig() {
+		if (this.config.connectionString) return this.parseConnectionString(this.config.connectionString);
+		if (this.config.host && this.config.database && this.config.appName) return {
+			host: this.config.host,
+			database: this.config.database,
+			port: this.config.port ?? 5432,
+			sslMode: this.config.sslMode ?? "require",
+			appName: this.config.appName
+		};
+		const pgHost = process.env.PGHOST;
+		const pgDatabase = process.env.PGDATABASE;
+		const pgAppName = process.env.PGAPPNAME;
+		if (!pgHost || !pgDatabase || !pgAppName) throw new Error("Lakebase connection not configured. Required env vars: PGHOST, PGDATABASE, PGAPPNAME. Optional: PGPORT (default: 5432), PGSSLMODE (default: require).");
+		const pgPort = process.env.PGPORT;
+		const port = pgPort ? parseInt(pgPort, 10) : 5432;
+		if (Number.isNaN(port)) throw new Error(`Invalid port: ${pgPort}. Must be a number.`);
+		return {
+			host: pgHost,
+			database: pgDatabase,
+			port,
+			sslMode: process.env.PGSSLMODE || "require",
+			appName: pgAppName
+		};
+	}
+	parseConnectionString(connectionString) {
+		const url = new URL(connectionString);
+		const appName = url.searchParams.get("appName");
+		if (!appName) throw new Error("Connection string must include appName parameter");
+		return {
+			host: url.hostname,
+			database: url.pathname.slice(1),
+			port: url.port ? parseInt(url.port, 10) : 5432,
+			sslMode: url.searchParams.get("sslmode") ?? "require",
+			appName
+		};
+	}
+};
+
+//#endregion
+export { LakebaseConnector };
+//# sourceMappingURL=client.js.map

package/dist/connectors/lakebase/client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"client.js","names":[],"sources":["../../../src/connectors/lakebase/client.ts"],"sourcesContent":["import { randomUUID } from \"node:crypto\";\nimport type { WorkspaceClient } from \"@databricks/sdk-experimental\";\nimport { ApiClient, Config } from \"@databricks/sdk-experimental\";\nimport pg from \"pg\";\nimport {\n  type Counter,\n  type Histogram,\n  SpanStatusCode,\n  TelemetryManager,\n  type TelemetryProvider,\n} from \"@/telemetry\";\nimport { deepMerge } from \"../../utils\";\nimport { lakebaseDefaults } from \"./defaults\";\nimport type {\n  LakebaseConfig,\n  LakebaseConnectionConfig,\n  LakebaseCredentials,\n} from \"./types\";\n\n/**\n * Enterprise-grade connector for Databricks Lakebase\n * @example Simplest - everything from env/context\n * ```typescript\n * const connector = new LakebaseConnector();\n * await connector.query('SELECT * FROM users');\n * ```\n *\n * @example With explicit connection string\n * ```typescript\n * const connector = new LakebaseConnector({\n *   connectionString: 'postgresql://...'\n * });\n * ```\n */\nexport class LakebaseConnector {\n  private readonly name: string = \"lakebase\";\n  private readonly CACHE_BUFFER_MS = 2 * 60 * 1000;\n  private readonly config: LakebaseConfig;\n  private readonly connectionConfig: LakebaseConnectionConfig;\n  private pool: pg.Pool | null = null;\n  private credentials: LakebaseCredentials | null = null;\n\n  // telemetry\n  private readonly telemetry: TelemetryProvider;\n  private readonly telemetryMetrics: {\n    queryCount: Counter;\n    queryDuration: Histogram;\n  };\n\n  constructor(userConfig?: Partial<LakebaseConfig>) {\n    this.config = deepMerge(lakebaseDefaults, userConfig);\n    this.connectionConfig = this.parseConnectionConfig();\n\n    this.telemetry = TelemetryManager.getProvider(\n      this.name,\n      this.config.telemetry,\n    );\n    this.telemetryMetrics = {\n      queryCount: this.telemetry\n        .getMeter()\n        .createCounter(\"lakebase.query.count\", {\n          description: \"Total number of queries executed\",\n          unit: \"1\",\n        }),\n      queryDuration: this.telemetry\n        .getMeter()\n        .createHistogram(\"lakebase.query.duration\", {\n          description: \"Duration of queries executed\",\n          unit: \"ms\",\n        }),\n    };\n\n    // validate configuration\n    if (this.config.maxPoolSize < 1) {\n      throw new Error(\"maxPoolSize must be at least 1\");\n    }\n  }\n\n  /**\n   * Execute a SQL query\n   *\n   * @example\n   * ```typescript\n   * const users = await connector.query('SELECT * FROM users');\n   * const user = await connector.query('SELECT * FROM users WHERE id = $1', [123]);\n   * ```\n   */\n  async query<T extends pg.QueryResultRow>(\n    sql: string,\n    params?: any[],\n    retryCount: number = 0,\n  ): Promise<pg.QueryResult<T>> {\n    const startTime = Date.now();\n\n    return this.telemetry.startActiveSpan(\n      \"lakebase.query\",\n      {\n        attributes: {\n          \"db.system\": \"lakebase\",\n          \"db.statement\": sql.substring(0, 500),\n          \"db.retry_count\": retryCount,\n        },\n      },\n      async (span) => {\n        try {\n          const pool = await this.getPool();\n          const result = await pool.query<T>(sql, params);\n          span.setAttribute(\"db.rows_affected\", result.rowCount ?? 0);\n          span.setStatus({ code: SpanStatusCode.OK });\n          return result;\n        } catch (error) {\n          // retry on auth failure\n          if (this.isAuthError(error)) {\n            span.addEvent(\"auth_error_retry\");\n            await this.rotateCredentials();\n            const newPool = await this.getPool();\n            const result = await newPool.query<T>(sql, params);\n            span.setAttribute(\"db.rows_affected\", result.rowCount ?? 0);\n            span.setStatus({ code: SpanStatusCode.OK });\n            return result;\n          }\n\n          // retry on transient errors, but only once\n          if (this.isTransientError(error) && retryCount < 1) {\n            span.addEvent(\"transient_error_retry\");\n            await new Promise((resolve) => setTimeout(resolve, 100));\n            return await this.query<T>(sql, params, retryCount + 1);\n          }\n\n          span.recordException(error as Error);\n          span.setStatus({ code: SpanStatusCode.ERROR });\n\n          throw error;\n        } finally {\n          const duration = Date.now() - startTime;\n          this.telemetryMetrics.queryCount.add(1);\n          this.telemetryMetrics.queryDuration.record(duration);\n          span.end();\n        }\n      },\n    );\n  }\n\n  /**\n   * Execute a transaction\n   *\n   * COMMIT and ROLLBACK are automatically managed by the transaction function.\n   *\n   * @param callback - Callback function to execute within the transaction context\n   * @example\n   * ```typescript\n   * await connector.transaction(async (client) => {\n   *   await client.query('INSERT INTO accounts (name) VALUES ($1)', ['Alice']);\n   *   await client.query('INSERT INTO logs (action) VALUES ($1)', ['Created Alice']);\n   * });\n   * ```\n   */\n  async transaction<T>(\n    callback: (client: pg.PoolClient) => Promise<T>,\n    retryCount: number = 0,\n  ): Promise<T> {\n    const startTime = Date.now();\n    return this.telemetry.startActiveSpan(\n      \"lakebase.transaction\",\n      {\n        attributes: {\n          \"db.system\": \"lakebase\",\n          \"db.retry_count\": retryCount,\n        },\n      },\n      async (span) => {\n        const pool = await this.getPool();\n        const client = await pool.connect();\n        try {\n          await client.query(\"BEGIN\");\n          const result = await callback(client);\n          await client.query(\"COMMIT\");\n          span.setStatus({ code: SpanStatusCode.OK });\n          return result;\n        } catch (error) {\n          try {\n            await client.query(\"ROLLBACK\");\n          } catch {}\n          // retry on auth failure\n          if (this.isAuthError(error)) {\n            span.addEvent(\"auth_error_retry\");\n            client.release();\n            await this.rotateCredentials();\n            const newPool = await this.getPool();\n            const retryClient = await newPool.connect();\n            try {\n              await client.query(\"BEGIN\");\n              const result = await callback(retryClient);\n              await client.query(\"COMMIT\");\n              span.setStatus({ code: SpanStatusCode.OK });\n              return result;\n            } catch (retryError) {\n              try {\n                await retryClient.query(\"ROLLBACK\");\n              } catch {}\n              throw retryError;\n            } finally {\n              retryClient.release();\n            }\n          }\n\n          // retry on transient errors, but only once\n          if (this.isTransientError(error) && retryCount < 1) {\n            span.addEvent(\"transaction_error_retry\");\n            client.release();\n            await new Promise((resolve) => setTimeout(resolve, 100));\n            return await this.transaction<T>(callback, retryCount + 1);\n          }\n          span.recordException(error as Error);\n          span.setStatus({ code: SpanStatusCode.ERROR });\n          throw error;\n        } finally {\n          client.release();\n          const duration = Date.now() - startTime;\n          this.telemetryMetrics.queryCount.add(1);\n          this.telemetryMetrics.queryDuration.record(duration);\n          span.end();\n        }\n      },\n    );\n  }\n\n  /** Check if database connection is healthy */\n  async healthCheck(): Promise<boolean> {\n    return this.telemetry.startActiveSpan(\n      \"lakebase.healthCheck\",\n      {},\n      async (span) => {\n        try {\n          const result = await this.query<{ result: number }>(\n            \"SELECT 1 as result\",\n          );\n          const healthy = result.rows[0]?.result === 1;\n          span.setAttribute(\"db.healthy\", healthy);\n          span.setStatus({ code: SpanStatusCode.OK });\n          return healthy;\n        } catch {\n          span.setAttribute(\"db.healthy\", false);\n          span.setStatus({ code: SpanStatusCode.ERROR });\n          return false;\n        } finally {\n          span.end();\n        }\n      },\n    );\n  }\n\n  /** Close connection pool (call on shutdown) */\n  async close(): Promise<void> {\n    if (this.pool) {\n      await this.pool.end().catch((error: unknown) => {\n        console.error(\"Error closing connection pool:\", error);\n      });\n      this.pool = null;\n    }\n    this.credentials = null;\n  }\n\n  /** Setup graceful shutdown to close connection pools */\n  shutdown(): void {\n    process.on(\"SIGTERM\", () => this.close());\n    process.on(\"SIGINT\", () => this.close());\n    this.close();\n  }\n\n  /** Get Databricks workspace client - from config or request context */\n  private getWorkspaceClient(): WorkspaceClient {\n    if (this.config.workspaceClient) {\n      return this.config.workspaceClient;\n    }\n\n    try {\n      const { getRequestContext } = require(\"../../utils\");\n      const { serviceDatabricksClient } = getRequestContext();\n\n      // cache it for subsequent calls\n      this.config.workspaceClient = serviceDatabricksClient;\n      return serviceDatabricksClient;\n    } catch (_error) {\n      throw new Error(\n        \"Databricks workspace client not available. Either pass it in config or use within AppKit request context.\",\n      );\n    }\n  }\n\n  /** Get or create connection pool */\n  private async getPool(): Promise<pg.Pool> {\n    if (!this.connectionConfig) {\n      throw new Error(\n        \"Lakebase connection not configured. \" +\n          \"Set PGHOST, PGDATABASE, PGAPPNAME env vars, provide a connectionString, or pass explicit config.\",\n      );\n    }\n\n    if (!this.pool) {\n      const creds = await this.getCredentials();\n      this.pool = this.createPool(creds);\n    }\n    return this.pool;\n  }\n\n  /** Create PostgreSQL pool */\n  private createPool(credentials: {\n    username: string;\n    password: string;\n  }): pg.Pool {\n    const { host, database, port, sslMode } = this.connectionConfig;\n\n    const pool = new pg.Pool({\n      host,\n      port,\n      database,\n      user: credentials.username,\n      password: credentials.password,\n      max: this.config.maxPoolSize,\n      idleTimeoutMillis: this.config.idleTimeoutMs,\n      connectionTimeoutMillis: this.config.connectionTimeoutMs,\n      ssl: sslMode === \"require\" ? { rejectUnauthorized: true } : false,\n    });\n\n    pool.on(\"error\", (error: Error & { code?: string }) => {\n      console.error(\"Connection pool error:\", error.message, {\n        code: error.code,\n      });\n    });\n\n    return pool;\n  }\n\n  /** Get or fetch credentials with caching */\n  private async getCredentials(): Promise<{\n    username: string;\n    password: string;\n  }> {\n    const now = Date.now();\n\n    // return cached if still valid\n    if (\n      this.credentials &&\n      now < this.credentials.expiresAt - this.CACHE_BUFFER_MS\n    ) {\n      return this.credentials;\n    }\n\n    // fetch new credentials\n    const username = await this.fetchUsername();\n    const { token, expiresAt } = await this.fetchPassword();\n\n    this.credentials = {\n      username,\n      password: token,\n      expiresAt,\n    };\n\n    return { username, password: token };\n  }\n\n  /** Rotate credentials and recreate pool */\n  private async rotateCredentials(): Promise<void> {\n    // clear cached credentials\n    this.credentials = null;\n\n    if (this.pool) {\n      const oldPool = this.pool;\n      this.pool = null;\n      oldPool.end().catch((error: unknown) => {\n        console.error(\n          \"Error closing old connection pool during rotation:\",\n          error,\n        );\n      });\n    }\n  }\n\n  /** Fetch username from Databricks */\n  private async fetchUsername(): Promise<string> {\n    const workspaceClient = this.getWorkspaceClient();\n    const user = await workspaceClient.currentUser.me();\n    if (!user.userName) {\n      throw new Error(\"Failed to get current user from Databricks workspace\");\n    }\n    return user.userName;\n  }\n\n  /** Fetch password (OAuth token) from Databricks */\n  private async fetchPassword(): Promise<{ token: string; expiresAt: number }> {\n    const workspaceClient = this.getWorkspaceClient();\n    const config = new Config({ host: workspaceClient.config.host });\n    const apiClient = new ApiClient(config);\n\n    if (!this.connectionConfig.appName) {\n      throw new Error(`Database app name not found in connection config`);\n    }\n\n    const credentials = await apiClient.request({\n      path: `/api/2.0/database/credentials`,\n      method: \"POST\",\n      headers: new Headers(),\n      raw: false,\n      payload: {\n        instance_names: [this.connectionConfig.appName],\n        request_id: randomUUID(),\n      },\n    });\n\n    if (!this.validateCredentials(credentials)) {\n      throw new Error(\n        `Failed to generate credentials for instance: ${this.connectionConfig.appName}`,\n      );\n    }\n\n    const expiresAt = new Date(credentials.expiration_time).getTime();\n\n    return { token: credentials.token, expiresAt };\n  }\n\n  /** Check if error is auth failure */\n  private isAuthError(error: unknown): boolean {\n    return (\n      typeof error === \"object\" &&\n      error !== null &&\n      \"code\" in error &&\n      (error as any).code === \"28P01\"\n    );\n  }\n\n  /** Check if error is transient */\n  private isTransientError(error: unknown): boolean {\n    if (typeof error !== \"object\" || error === null || !(\"code\" in error)) {\n      return false;\n    }\n\n    const code = (error as any).code;\n    return (\n      code === \"ECONNRESET\" ||\n      code === \"ECONNREFUSED\" ||\n      code === \"ETIMEDOUT\" ||\n      code === \"57P01\" || // admin_shutdown\n      code === \"57P03\" || // cannot_connect_now\n      code === \"08006\" || // connection_failure\n      code === \"08003\" || // connection_does_not_exist\n      code === \"08000\" // connection_exception\n    );\n  }\n\n  /** Type guard for credentials */\n  private validateCredentials(\n    value: unknown,\n  ): value is { token: string; expiration_time: string } {\n    if (typeof value !== \"object\" || value === null) {\n      return false;\n    }\n\n    const credentials = value as { token: string; expiration_time: string };\n    return (\n      \"token\" in credentials &&\n      typeof credentials.token === \"string\" &&\n      \"expiration_time\" in credentials &&\n      typeof credentials.expiration_time === \"string\" &&\n      new Date(credentials.expiration_time).getTime() > Date.now()\n    );\n  }\n\n  /** Parse connection configuration from config or environment */\n  private parseConnectionConfig(): LakebaseConnectionConfig {\n    if (this.config.connectionString) {\n      return this.parseConnectionString(this.config.connectionString);\n    }\n\n    // get connection from config\n    if (this.config.host && this.config.database && this.config.appName) {\n      return {\n        host: this.config.host,\n        database: this.config.database,\n        port: this.config.port ?? 5432,\n        sslMode: this.config.sslMode ?? \"require\",\n        appName: this.config.appName,\n      };\n    }\n\n    // get connection from environment variables\n    const pgHost = process.env.PGHOST;\n    const pgDatabase = process.env.PGDATABASE;\n    const pgAppName = process.env.PGAPPNAME;\n    if (!pgHost || !pgDatabase || !pgAppName) {\n      throw new Error(\n        \"Lakebase connection not configured. Required env vars: PGHOST, PGDATABASE, PGAPPNAME. \" +\n          \"Optional: PGPORT (default: 5432), PGSSLMODE (default: require).\",\n      );\n    }\n    const pgPort = process.env.PGPORT;\n    const port = pgPort ? parseInt(pgPort, 10) : 5432;\n\n    if (Number.isNaN(port)) {\n      throw new Error(`Invalid port: ${pgPort}. Must be a number.`);\n    }\n\n    const pgSSLMode = process.env.PGSSLMODE;\n    const sslMode =\n      (pgSSLMode as \"require\" | \"disable\" | \"prefer\") || \"require\";\n\n    return {\n      host: pgHost,\n      database: pgDatabase,\n      port,\n      sslMode,\n      appName: pgAppName,\n    };\n  }\n\n  private parseConnectionString(\n    connectionString: string,\n  ): LakebaseConnectionConfig {\n    const url = new URL(connectionString);\n    const appName = url.searchParams.get(\"appName\");\n    if (!appName) {\n      throw new Error(\"Connection string must include appName parameter\");\n    }\n\n    return {\n      host: url.hostname,\n      database: url.pathname.slice(1), // remove leading slash\n      port: url.port ? parseInt(url.port, 10) : 5432,\n      sslMode:\n        (url.searchParams.get(\"sslmode\") as \"require\" | \"disable\" | \"prefer\") ??\n        \"require\",\n      appName: appName,\n    };\n  }\n}\n"],"mappings":";;;;;;;;;;;YAWwC;;;;;;;;;;;;;;;;AAuBxC,IAAa,oBAAb,MAA+B;CAe7B,YAAY,YAAsC;cAdlB;yBACG,MAAS;cAGb;qBACmB;AAUhD,OAAK,SAAS,UAAU,kBAAkB,WAAW;AACrD,OAAK,mBAAmB,KAAK,uBAAuB;AAEpD,OAAK,YAAY,iBAAiB,YAChC,KAAK,MACL,KAAK,OAAO,UACb;AACD,OAAK,mBAAmB;GACtB,YAAY,KAAK,UACd,UAAU,CACV,cAAc,wBAAwB;IACrC,aAAa;IACb,MAAM;IACP,CAAC;GACJ,eAAe,KAAK,UACjB,UAAU,CACV,gBAAgB,2BAA2B;IAC1C,aAAa;IACb,MAAM;IACP,CAAC;GACL;AAGD,MAAI,KAAK,OAAO,cAAc,EAC5B,OAAM,IAAI,MAAM,iCAAiC;;;;;;;;;;;CAarD,MAAM,MACJ,KACA,QACA,aAAqB,GACO;EAC5B,MAAM,YAAY,KAAK,KAAK;AAE5B,SAAO,KAAK,UAAU,gBACpB,kBACA,EACE,YAAY;GACV,aAAa;GACb,gBAAgB,IAAI,UAAU,GAAG,IAAI;GACrC,kBAAkB;GACnB,EACF,EACD,OAAO,SAAS;AACd,OAAI;IAEF,MAAM,SAAS,OADF,MAAM,KAAK,SAAS,EACP,MAAS,KAAK,OAAO;AAC/C,SAAK,aAAa,oBAAoB,OAAO,YAAY,EAAE;AAC3D,SAAK,UAAU,EAAE,MAAM,eAAe,IAAI,CAAC;AAC3C,WAAO;YACA,OAAO;AAEd,QAAI,KAAK,YAAY,MAAM,EAAE;AAC3B,UAAK,SAAS,mBAAmB;AACjC,WAAM,KAAK,mBAAmB;KAE9B,MAAM,SAAS,OADC,MAAM,KAAK,SAAS,EACP,MAAS,KAAK,OAAO;AAClD,UAAK,aAAa,oBAAoB,OAAO,YAAY,EAAE;AAC3D,UAAK,UAAU,EAAE,MAAM,eAAe,IAAI,CAAC;AAC3C,YAAO;;AAIT,QAAI,KAAK,iBAAiB,MAAM,IAAI,aAAa,GAAG;AAClD,UAAK,SAAS,wBAAwB;AACtC,WAAM,IAAI,SAAS,YAAY,WAAW,SAAS,IAAI,CAAC;AACxD,YAAO,MAAM,KAAK,MAAS,KAAK,QAAQ,aAAa,EAAE;;AAGzD,SAAK,gBAAgB,MAAe;AACpC,SAAK,UAAU,EAAE,MAAM,eAAe,OAAO,CAAC;AAE9C,UAAM;aACE;IACR,MAAM,WAAW,KAAK,KAAK,GAAG;AAC9B,SAAK,iBAAiB,WAAW,IAAI,EAAE;AACvC,SAAK,iBAAiB,cAAc,OAAO,SAAS;AACpD,SAAK,KAAK;;IAGf;;;;;;;;;;;;;;;;CAiBH,MAAM,YACJ,UACA,aAAqB,GACT;EACZ,MAAM,YAAY,KAAK,KAAK;AAC5B,SAAO,KAAK,UAAU,gBACpB,wBACA,EACE,YAAY;GACV,aAAa;GACb,kBAAkB;GACnB,EACF,EACD,OAAO,SAAS;GAEd,MAAM,SAAS,OADF,MAAM,KAAK,SAAS,EACP,SAAS;AACnC,OAAI;AACF,UAAM,OAAO,MAAM,QAAQ;IAC3B,MAAM,SAAS,MAAM,SAAS,OAAO;AACrC,UAAM,OAAO,MAAM,SAAS;AAC5B,SAAK,UAAU,EAAE,MAAM,eAAe,IAAI,CAAC;AAC3C,WAAO;YACA,OAAO;AACd,QAAI;AACF,WAAM,OAAO,MAAM,WAAW;YACxB;AAER,QAAI,KAAK,YAAY,MAAM,EAAE;AAC3B,UAAK,SAAS,mBAAmB;AACjC,YAAO,SAAS;AAChB,WAAM,KAAK,mBAAmB;KAE9B,MAAM,cAAc,OADJ,MAAM,KAAK,SAAS,EACF,SAAS;AAC3C,SAAI;AACF,YAAM,OAAO,MAAM,QAAQ;MAC3B,MAAM,SAAS,MAAM,SAAS,YAAY;AAC1C,YAAM,OAAO,MAAM,SAAS;AAC5B,WAAK,UAAU,EAAE,MAAM,eAAe,IAAI,CAAC;AAC3C,aAAO;cACA,YAAY;AACnB,UAAI;AACF,aAAM,YAAY,MAAM,WAAW;cAC7B;AACR,YAAM;eACE;AACR,kBAAY,SAAS;;;AAKzB,QAAI,KAAK,iBAAiB,MAAM,IAAI,aAAa,GAAG;AAClD,UAAK,SAAS,0BAA0B;AACxC,YAAO,SAAS;AAChB,WAAM,IAAI,SAAS,YAAY,WAAW,SAAS,IAAI,CAAC;AACxD,YAAO,MAAM,KAAK,YAAe,UAAU,aAAa,EAAE;;AAE5D,SAAK,gBAAgB,MAAe;AACpC,SAAK,UAAU,EAAE,MAAM,eAAe,OAAO,CAAC;AAC9C,UAAM;aACE;AACR,WAAO,SAAS;IAChB,MAAM,WAAW,KAAK,KAAK,GAAG;AAC9B,SAAK,iBAAiB,WAAW,IAAI,EAAE;AACvC,SAAK,iBAAiB,cAAc,OAAO,SAAS;AACpD,SAAK,KAAK;;IAGf;;;CAIH,MAAM,cAAgC;AACpC,SAAO,KAAK,UAAU,gBACpB,wBACA,EAAE,EACF,OAAO,SAAS;AACd,OAAI;IAIF,MAAM,WAHS,MAAM,KAAK,MACxB,qBACD,EACsB,KAAK,IAAI,WAAW;AAC3C,SAAK,aAAa,cAAc,QAAQ;AACxC,SAAK,UAAU,EAAE,MAAM,eAAe,IAAI,CAAC;AAC3C,WAAO;WACD;AACN,SAAK,aAAa,cAAc,MAAM;AACtC,SAAK,UAAU,EAAE,MAAM,eAAe,OAAO,CAAC;AAC9C,WAAO;aACC;AACR,SAAK,KAAK;;IAGf;;;CAIH,MAAM,QAAuB;AAC3B,MAAI,KAAK,MAAM;AACb,SAAM,KAAK,KAAK,KAAK,CAAC,OAAO,UAAmB;AAC9C,YAAQ,MAAM,kCAAkC,MAAM;KACtD;AACF,QAAK,OAAO;;AAEd,OAAK,cAAc;;;CAIrB,WAAiB;AACf,UAAQ,GAAG,iBAAiB,KAAK,OAAO,CAAC;AACzC,UAAQ,GAAG,gBAAgB,KAAK,OAAO,CAAC;AACxC,OAAK,OAAO;;;CAId,AAAQ,qBAAsC;AAC5C,MAAI,KAAK,OAAO,gBACd,QAAO,KAAK,OAAO;AAGrB,MAAI;GACF,MAAM,EAAE;GACR,MAAM,EAAE,4BAA4B,mBAAmB;AAGvD,QAAK,OAAO,kBAAkB;AAC9B,UAAO;WACA,QAAQ;AACf,SAAM,IAAI,MACR,4GACD;;;;CAKL,MAAc,UAA4B;AACxC,MAAI,CAAC,KAAK,iBACR,OAAM,IAAI,MACR,uIAED;AAGH,MAAI,CAAC,KAAK,MAAM;GACd,MAAM,QAAQ,MAAM,KAAK,gBAAgB;AACzC,QAAK,OAAO,KAAK,WAAW,MAAM;;AAEpC,SAAO,KAAK;;;CAId,AAAQ,WAAW,aAGP;EACV,MAAM,EAAE,MAAM,UAAU,MAAM,YAAY,KAAK;EAE/C,MAAM,OAAO,IAAI,GAAG,KAAK;GACvB;GACA;GACA;GACA,MAAM,YAAY;GAClB,UAAU,YAAY;GACtB,KAAK,KAAK,OAAO;GACjB,mBAAmB,KAAK,OAAO;GAC/B,yBAAyB,KAAK,OAAO;GACrC,KAAK,YAAY,YAAY,EAAE,oBAAoB,MAAM,GAAG;GAC7D,CAAC;AAEF,OAAK,GAAG,UAAU,UAAqC;AACrD,WAAQ,MAAM,0BAA0B,MAAM,SAAS,EACrD,MAAM,MAAM,MACb,CAAC;IACF;AAEF,SAAO;;;CAIT,MAAc,iBAGX;EACD,MAAM,MAAM,KAAK,KAAK;AAGtB,MACE,KAAK,eACL,MAAM,KAAK,YAAY,YAAY,KAAK,gBAExC,QAAO,KAAK;EAId,MAAM,WAAW,MAAM,KAAK,eAAe;EAC3C,MAAM,EAAE,OAAO,cAAc,MAAM,KAAK,eAAe;AAEvD,OAAK,cAAc;GACjB;GACA,UAAU;GACV;GACD;AAED,SAAO;GAAE;GAAU,UAAU;GAAO;;;CAItC,MAAc,oBAAmC;AAE/C,OAAK,cAAc;AAEnB,MAAI,KAAK,MAAM;GACb,MAAM,UAAU,KAAK;AACrB,QAAK,OAAO;AACZ,WAAQ,KAAK,CAAC,OAAO,UAAmB;AACtC,YAAQ,MACN,sDACA,MACD;KACD;;;;CAKN,MAAc,gBAAiC;EAE7C,MAAM,OAAO,MADW,KAAK,oBAAoB,CACd,YAAY,IAAI;AACnD,MAAI,CAAC,KAAK,SACR,OAAM,IAAI,MAAM,uDAAuD;AAEzE,SAAO,KAAK;;;CAId,MAAc,gBAA+D;EAG3E,MAAM,YAAY,IAAI,UADP,IAAI,OAAO,EAAE,MADJ,KAAK,oBAAoB,CACC,OAAO,MAAM,CAAC,CACzB;AAEvC,MAAI,CAAC,KAAK,iBAAiB,QACzB,OAAM,IAAI,MAAM,mDAAmD;EAGrE,MAAM,cAAc,MAAM,UAAU,QAAQ;GAC1C,MAAM;GACN,QAAQ;GACR,SAAS,IAAI,SAAS;GACtB,KAAK;GACL,SAAS;IACP,gBAAgB,CAAC,KAAK,iBAAiB,QAAQ;IAC/C,YAAY,YAAY;IACzB;GACF,CAAC;AAEF,MAAI,CAAC,KAAK,oBAAoB,YAAY,CACxC,OAAM,IAAI,MACR,gDAAgD,KAAK,iBAAiB,UACvE;EAGH,MAAM,YAAY,IAAI,KAAK,YAAY,gBAAgB,CAAC,SAAS;AAEjE,SAAO;GAAE,OAAO,YAAY;GAAO;GAAW;;;CAIhD,AAAQ,YAAY,OAAyB;AAC3C,SACE,OAAO,UAAU,YACjB,UAAU,QACV,UAAU,SACT,MAAc,SAAS;;;CAK5B,AAAQ,iBAAiB,OAAyB;AAChD,MAAI,OAAO,UAAU,YAAY,UAAU,QAAQ,EAAE,UAAU,OAC7D,QAAO;EAGT,MAAM,OAAQ,MAAc;AAC5B,SACE,SAAS,gBACT,SAAS,kBACT,SAAS,eACT,SAAS,WACT,SAAS,WACT,SAAS,WACT,SAAS,WACT,SAAS;;;CAKb,AAAQ,oBACN,OACqD;AACrD,MAAI,OAAO,UAAU,YAAY,UAAU,KACzC,QAAO;EAGT,MAAM,cAAc;AACpB,SACE,WAAW,eACX,OAAO,YAAY,UAAU,YAC7B,qBAAqB,eACrB,OAAO,YAAY,oBAAoB,YACvC,IAAI,KAAK,YAAY,gBAAgB,CAAC,SAAS,GAAG,KAAK,KAAK;;;CAKhE,AAAQ,wBAAkD;AACxD,MAAI,KAAK,OAAO,iBACd,QAAO,KAAK,sBAAsB,KAAK,OAAO,iBAAiB;AAIjE,MAAI,KAAK,OAAO,QAAQ,KAAK,OAAO,YAAY,KAAK,OAAO,QAC1D,QAAO;GACL,MAAM,KAAK,OAAO;GAClB,UAAU,KAAK,OAAO;GACtB,MAAM,KAAK,OAAO,QAAQ;GAC1B,SAAS,KAAK,OAAO,WAAW;GAChC,SAAS,KAAK,OAAO;GACtB;EAIH,MAAM,SAAS,QAAQ,IAAI;EAC3B,MAAM,aAAa,QAAQ,IAAI;EAC/B,MAAM,YAAY,QAAQ,IAAI;AAC9B,MAAI,CAAC,UAAU,CAAC,cAAc,CAAC,UAC7B,OAAM,IAAI,MACR,wJAED;EAEH,MAAM,SAAS,QAAQ,IAAI;EAC3B,MAAM,OAAO,SAAS,SAAS,QAAQ,GAAG,GAAG;AAE7C,MAAI,OAAO,MAAM,KAAK,CACpB,OAAM,IAAI,MAAM,iBAAiB,OAAO,qBAAqB;AAO/D,SAAO;GACL,MAAM;GACN,UAAU;GACV;GACA,SARgB,QAAQ,IAAI,aAEuB;GAOnD,SAAS;GACV;;CAGH,AAAQ,sBACN,kBAC0B;EAC1B,MAAM,MAAM,IAAI,IAAI,iBAAiB;EACrC,MAAM,UAAU,IAAI,aAAa,IAAI,UAAU;AAC/C,MAAI,CAAC,QACH,OAAM,IAAI,MAAM,mDAAmD;AAGrE,SAAO;GACL,MAAM,IAAI;GACV,UAAU,IAAI,SAAS,MAAM,EAAE;GAC/B,MAAM,IAAI,OAAO,SAAS,IAAI,MAAM,GAAG,GAAG;GAC1C,SACG,IAAI,aAAa,IAAI,UAAU,IAChC;GACO;GACV"}

package/dist/connectors/lakebase/defaults.js

@@ -0,0 +1,13 @@
+//#region src/connectors/lakebase/defaults.ts
+/** Default configuration for Lakebase connector */
+const lakebaseDefaults = {
+	port: 5432,
+	sslMode: "require",
+	maxPoolSize: 10,
+	idleTimeoutMs: 3e4,
+	connectionTimeoutMs: 1e4
+};
+
+//#endregion
+export { lakebaseDefaults };
+//# sourceMappingURL=defaults.js.map

package/dist/connectors/lakebase/defaults.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"defaults.js","names":["lakebaseDefaults: LakebaseConfig"],"sources":["../../../src/connectors/lakebase/defaults.ts"],"sourcesContent":["import type { LakebaseConfig } from \"./types\";\n\n/** Default configuration for Lakebase connector */\nexport const lakebaseDefaults: LakebaseConfig = {\n  port: 5432,\n  sslMode: \"require\",\n  maxPoolSize: 10,\n  idleTimeoutMs: 30_000,\n  connectionTimeoutMs: 10_000,\n};\n"],"mappings":";;AAGA,MAAaA,mBAAmC;CAC9C,MAAM;CACN,SAAS;CACT,aAAa;CACb,eAAe;CACf,qBAAqB;CACtB"}

package/dist/connectors/lakebase/index.js

@@ -0,0 +1,3 @@
+import { LakebaseConnector } from "./client.js";
+
+export {  };