@elizaos/plugin-sql 1.0.0-alpha.6 → 1.0.0-alpha.60

package/README.md

@@ -9,19 +9,6 @@ A PostgreSQL database adapter built with Drizzle ORM for the ElizaOS ecosystem.
 bun add @elizaos/plugin-sql
 ```
 
-## Database Schema
-
-The adapter includes the following tables:
-- accounts
-- cache
-- embeddings
-- goals
-- logs
-- memories
-- participants
-- relationships
-- rooms
-
 ## Vector Dimensions
 
 The adapter supports the following vector dimensions:
@@ -49,25 +36,42 @@ Important Note: Once an agent is initialized with a specific embedding dimension
 - Room and participant management
 - Goal tracking system
 
+## Database Schema
+
+The plugin uses a structured schema with the following main tables:
+
+### Core Tables
+- **Agent**: Stores agent information and configurations
+- **Room**: Manages conversation rooms and their settings
+- **Participant**: Tracks participants in rooms
+- **Memory**: Stores agent memories with vector embeddings for semantic search
+- **Embedding**: Manages vector embeddings for various entities
+- **Entity**: Represents entities that agents can interact with
+- **Relationship**: Tracks relationships between entities
+- **Component**: Stores agent components and their configurations
+- **Tasks**: Manages tasks and goals for agents
+- **Log**: Stores system logs
+- **Cache**: Provides a caching mechanism for frequently accessed data
+- **World**: Manages world settings and configurations
+
+Each table is defined using Drizzle ORM schema definitions in the `src/schema` directory. The schema is designed to support the ElizaOS ecosystem's requirements for agent-based systems.
+
 ## Usage
 
 The adapter is typically used as part of the ElizaOS runtime:
 
 ```typescript
 async function findDatabaseAdapter(runtime: IAgentRuntime) {
-    const { adapters } = runtime;
-    let adapter: Adapter | undefined;
+    let adapter = runtime;
     
-    if (adapters.length === 0) {
+    if (!adapter) {
         const drizzleAdapterPlugin = await import('@elizaos/plugin-sql');
         const drizzleAdapterPluginDefault = drizzleAdapterPlugin.default;
-        adapter = drizzleAdapterPluginDefault.adapters[0];
+        adapter = drizzleAdapterPluginDefault.adapter;
         if (!adapter) {
             throw new Error("Internal error: No database adapter found for default plugin-sql");
         }
-    } else if (adapters.length === 1) {
-        adapter = adapters[0];
-    } else {
+    } else if (!adapter) {
         throw new Error("Multiple database adapters found. You must have no more than one. Adjust your plugins configuration.");
     }
     
@@ -97,6 +101,16 @@ The adapter implements the following error handling configurations:
 - PostgreSQL with vector extension installed
 - Node.js or Bun (≥1.2.2)
 
+## Environment Variables
+
+The plugin uses the following environment variables:
+
+- `POSTGRES_URL`: Connection string for PostgreSQL database (e.g., `postgresql://user:password@localhost:5432/dbname`)
+  - If not provided, the plugin will use PGlite as a fallback
+- `PGLITE_DATA_DIR`: (Optional) Directory for PGlite data storage (default: `../../pglite`)
+
+These variables should be defined in a `.env` file at the root of your project.
+
 ## Database Pool Configuration
 
 Default pool configuration:
@@ -165,6 +179,50 @@ import { migrate } from "drizzle-orm/node-postgres/migrator";
 await migrate(db, { migrationsFolder: "./drizzle" });
 ```
 
+c. Using the provided migration script:
+```bash
+npm run migrate
+# or
+pnpm migrate
+```
+
+d. Using drizzle-kit migrate command:
+```bash
+npx drizzle-kit migrate
+```
+
+This command will read the configuration from `drizzle.config.ts` and pull the PostgreSQL URI from the `.env` file. Make sure your `.env` file contains the `POSTGRES_URL` variable with the correct connection string.
+
+### Migration Configuration
+
+The plugin uses a `drizzle.config.ts` file to configure migrations:
+
+```typescript
+import { config } from "dotenv";
+import { defineConfig } from "drizzle-kit";
+
+config({ path: "../../.env" });
+
+export default defineConfig({
+  dialect: "postgresql",
+  schema: "./src/schema/index.ts",
+  out: "./drizzle/migrations",
+  dbCredentials: {
+    url: process.env.POSTGRES_URL || "file://../../pglite",
+  },
+  breakpoints: true,
+});
+```
+
+### Database Support
+
+The plugin supports two database backends:
+
+1. **PostgreSQL**: Used when `POSTGRES_URL` environment variable is provided
+2. **PGlite**: Used as a fallback when no PostgreSQL URL is provided
+
+Both backends use the same migration files, ensuring consistent schema across environments.
+
 ### Note on Vector Support
 Make sure the PostgreSQL vector extension is installed before running migrations. The adapter will validate vector setup during initialization.
 
@@ -176,3 +234,36 @@ The adapter implements cleanup handlers for:
 - beforeExit
 
 These ensure proper closing of database connections when the application shuts down.
+
+## Implementation Details
+
+### Connection Management
+
+The plugin uses a global singleton pattern to manage database connections. This approach ensures that:
+
+1. **Single Connection Per Process**: Only one connection manager instance exists per Node.js process, regardless of how many times the package is imported or initialized.
+
+2. **Resource Efficiency**: Prevents multiple connection pools to the same database, which could lead to resource exhaustion.
+
+3. **Consistent State**: Ensures all parts of the application share the same database connection state.
+
+4. **Proper Cleanup**: Facilitates proper cleanup of database connections during application shutdown, preventing connection leaks.
+
+This pattern is particularly important in monorepo setups or when the package is used by multiple modules within the same process. The implementation uses JavaScript Symbols to create a global registry that persists across module boundaries.
+
+```typescript
+// Example of the singleton pattern implementation
+const GLOBAL_SINGLETONS = Symbol.for("@elizaos/plugin-sql/global-singletons");
+
+// Store managers in a global symbol registry
+if (!globalSymbols[GLOBAL_SINGLETONS]) {
+  globalSymbols[GLOBAL_SINGLETONS] = {};
+}
+
+// Reuse existing managers or create new ones when needed
+if (!globalSingletons.postgresConnectionManager) {
+  globalSingletons.postgresConnectionManager = new PostgresConnectionManager(config.postgresUrl);
+}
+```
+
+This approach is especially critical for PGlite connections, which require careful management to ensure proper shutdown and prevent resource leaks.

package/dist/chunk-JA2K6HCO.js

@@ -0,0 +1,342 @@
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+
+// src/pglite/manager.ts
+import { dirname as pathDirname, resolve as pathResolve } from "node:path";
+import { fileURLToPath } from "node:url";
+import { PGlite } from "@electric-sql/pglite";
+import { fuzzystrmatch } from "@electric-sql/pglite/contrib/fuzzystrmatch";
+import { vector } from "@electric-sql/pglite/vector";
+import { logger } from "@elizaos/core";
+import { drizzle } from "drizzle-orm/pglite";
+import { migrate } from "drizzle-orm/pglite/migrator";
+var PGliteClientManager = class {
+  static {
+    __name(this, "PGliteClientManager");
+  }
+  client;
+  shuttingDown = false;
+  shutdownTimeout = 500;
+  /**
+   * Constructor for creating a new instance of PGlite with the provided options.
+   * Initializes the PGlite client with additional extensions.
+   * @param {PGliteOptions} options - The options to configure the PGlite client.
+   */
+  constructor(options) {
+    this.client = new PGlite({
+      ...options,
+      extensions: {
+        vector,
+        fuzzystrmatch
+      }
+    });
+    this.setupShutdownHandlers();
+  }
+  /**
+   * Retrieves the PostgreSQL lite connection.
+   *
+   * @returns {PGlite} The PostgreSQL lite connection.
+   * @throws {Error} If the client manager is currently shutting down.
+   */
+  getConnection() {
+    if (this.shuttingDown) {
+      throw new Error("Client manager is shutting down");
+    }
+    return this.client;
+  }
+  /**
+   * Initiates a graceful shutdown of the PGlite client.
+   * Checks if the client is already in the process of shutting down.
+   * Logs the start of shutdown process and sets shuttingDown flag to true.
+   * Sets a timeout for the shutdown process and forces closure of database connection if timeout is reached.
+   * Handles the shutdown process, closes the client connection, clears the timeout, and logs the completion of shutdown.
+   * Logs any errors that occur during the shutdown process.
+   */
+  async gracefulShutdown() {
+    if (this.shuttingDown) {
+      return;
+    }
+    this.shuttingDown = true;
+    logger.info("Starting graceful shutdown of PGlite client...");
+    const timeout = setTimeout(() => {
+      logger.warn(
+        "Shutdown timeout reached, forcing database connection closure..."
+      );
+      this.client.close().finally(() => {
+        process.exit(1);
+      });
+    }, this.shutdownTimeout);
+    try {
+      await this.client.close();
+      clearTimeout(timeout);
+      logger.info("PGlite client shutdown completed successfully");
+      process.exit(0);
+    } catch (error) {
+      logger.error("Error during graceful shutdown:", error);
+      process.exit(1);
+    }
+  }
+  /**
+   * Sets up shutdown handlers for SIGINT, SIGTERM, and beforeExit events to gracefully shutdown the application.
+   * @private
+   */
+  setupShutdownHandlers() {
+    process.on("SIGINT", async () => {
+      await this.gracefulShutdown();
+    });
+    process.on("SIGTERM", async () => {
+      await this.gracefulShutdown();
+    });
+    process.on("beforeExit", async () => {
+      await this.gracefulShutdown();
+    });
+  }
+  /**
+   * Initializes the client for PGlite.
+   *
+   * @returns {Promise<void>} A Promise that resolves when the client is initialized successfully
+   */
+  async initialize() {
+    try {
+      await this.client.waitReady;
+      logger.info("PGlite client initialized successfully");
+    } catch (error) {
+      logger.error("Failed to initialize PGlite client:", error);
+      throw error;
+    }
+  }
+  /**
+   * Asynchronously closes the resource. If the resource is not already shutting down,
+   * it performs a graceful shutdown before closing.
+   *
+   * @returns A promise that resolves once the resource has been closed.
+   */
+  async close() {
+    if (!this.shuttingDown) {
+      await this.gracefulShutdown();
+    }
+  }
+  /**
+   * Check if the system is currently shutting down.
+   *
+   * @returns {boolean} True if the system is shutting down, false otherwise.
+   */
+  isShuttingDown() {
+    return this.shuttingDown;
+  }
+  /**
+   * Asynchronously runs database migrations using Drizzle.
+   * 
+   * Drizzle will first check if the migrations are already applied.
+   * If there is a diff between database schema and migrations, it will apply the migrations.
+   * If they are already applied, it will skip them.
+   *
+   * @returns {Promise<void>} A Promise that resolves once the migrations are completed successfully.
+   */
+  async runMigrations() {
+    try {
+      const db = drizzle(this.client);
+      const __filename = fileURLToPath(import.meta.url);
+      const __dirname = pathDirname(__filename);
+      await migrate(db, {
+        migrationsFolder: pathResolve(__dirname, "../drizzle/migrations")
+      });
+    } catch (error) {
+      logger.error("Failed to run database migrations (pglite):", error);
+      console.trace(error);
+    }
+  }
+};
+
+// src/pg/manager.ts
+import path from "node:path";
+import { fileURLToPath as fileURLToPath2 } from "node:url";
+import { logger as logger2 } from "@elizaos/core";
+import { drizzle as drizzle2 } from "drizzle-orm/node-postgres";
+import { migrate as migrate2 } from "drizzle-orm/node-postgres/migrator";
+import pkg from "pg";
+var { Pool } = pkg;
+var PostgresConnectionManager = class {
+  static {
+    __name(this, "PostgresConnectionManager");
+  }
+  pool;
+  isShuttingDown = false;
+  connectionTimeout = 5e3;
+  /**
+   * Constructor for creating a connection pool.
+   * @param {string} connectionString - The connection string used to connect to the database.
+   */
+  constructor(connectionString) {
+    const defaultConfig = {
+      max: 20,
+      idleTimeoutMillis: 3e4,
+      connectionTimeoutMillis: this.connectionTimeout
+    };
+    this.pool = new Pool({
+      ...defaultConfig,
+      connectionString
+    });
+    this.pool.on("error", (err) => {
+      logger2.error("Unexpected pool error", err);
+      this.handlePoolError(err);
+    });
+    this.setupPoolErrorHandling();
+    this.testConnection();
+  }
+  /**
+   * Handles a pool error by attempting to reconnect the pool.
+   *
+   * @param {Error} error The error that occurred in the pool.
+   * @throws {Error} If failed to reconnect the pool.
+   */
+  async handlePoolError(error) {
+    logger2.error("Pool error occurred, attempting to reconnect", {
+      error: error.message
+    });
+    try {
+      await this.pool.end();
+      this.pool = new Pool({
+        ...this.pool.options,
+        connectionTimeoutMillis: this.connectionTimeout
+      });
+      await this.testConnection();
+      logger2.success("Pool reconnection successful");
+    } catch (reconnectError) {
+      logger2.error("Failed to reconnect pool", {
+        error: reconnectError instanceof Error ? reconnectError.message : String(reconnectError)
+      });
+      throw reconnectError;
+    }
+  }
+  /**
+   * Asynchronously tests the database connection by executing a query to get the current timestamp.
+   *
+   * @returns {Promise<boolean>} - A Promise that resolves to true if the database connection test is successful.
+   */
+  async testConnection() {
+    let client = null;
+    try {
+      client = await this.pool.connect();
+      const result = await client.query("SELECT NOW()");
+      logger2.success("Database connection test successful:", result.rows[0]);
+      return true;
+    } catch (error) {
+      logger2.error("Database connection test failed:", error);
+      throw new Error(
+        `Failed to connect to database: ${error.message}`
+      );
+    } finally {
+      if (client) client.release();
+    }
+  }
+  /**
+   * Sets up event listeners to handle pool cleanup on SIGINT, SIGTERM, and beforeExit events.
+   */
+  setupPoolErrorHandling() {
+    process.on("SIGINT", async () => {
+      await this.cleanup();
+      process.exit(0);
+    });
+    process.on("SIGTERM", async () => {
+      await this.cleanup();
+      process.exit(0);
+    });
+    process.on("beforeExit", async () => {
+      await this.cleanup();
+    });
+  }
+  /**
+   * Get the connection pool.
+   * @returns {PgPool} The connection pool
+   * @throws {Error} If the connection manager is shutting down or an error occurs when trying to get the connection from the pool
+   */
+  getConnection() {
+    if (this.isShuttingDown) {
+      throw new Error("Connection manager is shutting down");
+    }
+    try {
+      return this.pool;
+    } catch (error) {
+      logger2.error("Failed to get connection from pool:", error);
+      throw error;
+    }
+  }
+  /**
+   * Asynchronously acquires a database client from the connection pool.
+   *
+   * @returns {Promise<pkg.PoolClient>} A Promise that resolves with the acquired database client.
+   * @throws {Error} If an error occurs while acquiring the database client.
+   */
+  async getClient() {
+    try {
+      return await this.pool.connect();
+    } catch (error) {
+      logger2.error("Failed to acquire a database client:", error);
+      throw error;
+    }
+  }
+  /**
+   * Initializes the PostgreSQL connection manager by testing the connection and logging the result.
+   *
+   * @returns {Promise<void>} A Promise that resolves once the manager is successfully initialized
+   * @throws {Error} If there is an error initializing the connection manager
+   */
+  async initialize() {
+    try {
+      await this.testConnection();
+      logger2.info("PostgreSQL connection manager initialized successfully");
+    } catch (error) {
+      logger2.error("Failed to initialize connection manager:", error);
+      throw error;
+    }
+  }
+  /**
+   * Asynchronously close the current process by executing a cleanup function.
+   * @returns A promise that resolves once the cleanup is complete.
+   */
+  async close() {
+    await this.cleanup();
+  }
+  /**
+   * Cleans up and closes the database pool.
+   * @returns {Promise<void>} A Promise that resolves when the database pool is closed.
+   */
+  async cleanup() {
+    try {
+      await this.pool.end();
+      logger2.info("Database pool closed");
+    } catch (error) {
+      logger2.error("Error closing database pool:", error);
+    }
+  }
+  /**
+   * Asynchronously runs database migrations using the Drizzle library.
+   *
+   * Drizzle will first check if the migrations are already applied.
+   * If there is a diff between database schema and migrations, it will apply the migrations.
+   * If they are already applied, it will skip them.
+   *
+   * @returns {Promise<void>} A Promise that resolves once the migrations are completed successfully.
+   */
+  async runMigrations() {
+    try {
+      const db = drizzle2(this.pool);
+      const __filename = fileURLToPath2(import.meta.url);
+      const __dirname = path.dirname(__filename);
+      await migrate2(db, {
+        migrationsFolder: path.resolve(__dirname, "../drizzle/migrations")
+      });
+    } catch (error) {
+      logger2.error("Failed to run database migrations (pg):", error);
+      console.trace(error);
+    }
+  }
+};
+
+export {
+  __name,
+  PGliteClientManager,
+  PostgresConnectionManager
+};
+//# sourceMappingURL=chunk-JA2K6HCO.js.map

package/dist/chunk-JA2K6HCO.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/pglite/manager.ts","../src/pg/manager.ts"],"sourceRoot":"./","sourcesContent":["import { dirname as pathDirname, resolve as pathResolve } from \"node:path\";\nimport { fileURLToPath } from \"node:url\";\nimport { PGlite, type PGliteOptions } from \"@electric-sql/pglite\";\nimport { fuzzystrmatch } from \"@electric-sql/pglite/contrib/fuzzystrmatch\";\nimport { vector } from \"@electric-sql/pglite/vector\";\nimport { logger } from \"@elizaos/core\";\nimport { drizzle } from \"drizzle-orm/pglite\";\nimport { migrate } from \"drizzle-orm/pglite/migrator\";\nimport type { IDatabaseClientManager } from \"../types\";\n\n/**\n * Class representing a database client manager for PGlite.\n * @implements { IDatabaseClientManager }\n */\nexport class PGliteClientManager implements IDatabaseClientManager<PGlite> {\n\tprivate client: PGlite;\n\tprivate shuttingDown = false;\n\tprivate readonly shutdownTimeout = 500;\n\n\t/**\n\t * Constructor for creating a new instance of PGlite with the provided options.\n\t * Initializes the PGlite client with additional extensions.\n\t * @param {PGliteOptions} options - The options to configure the PGlite client.\n\t */\n\tconstructor(options: PGliteOptions) {\n\t\tthis.client = new PGlite({\n\t\t\t...options,\n\t\t\textensions: {\n\t\t\t\tvector,\n\t\t\t\tfuzzystrmatch,\n\t\t\t},\n\t\t});\n\t\tthis.setupShutdownHandlers();\n\t}\n\n\t/**\n\t * Retrieves the PostgreSQL lite connection.\n\t *\n\t * @returns {PGlite} The PostgreSQL lite connection.\n\t * @throws {Error} If the client manager is currently shutting down.\n\t */\n\tpublic getConnection(): PGlite {\n\t\tif (this.shuttingDown) {\n\t\t\tthrow new Error(\"Client manager is shutting down\");\n\t\t}\n\t\treturn this.client;\n\t}\n\n\t/**\n\t * Initiates a graceful shutdown of the PGlite client.\n\t * Checks if the client is already in the process of shutting down.\n\t * Logs the start of shutdown process and sets shuttingDown flag to true.\n\t * Sets a timeout for the shutdown process and forces closure of database connection if timeout is reached.\n\t * Handles the shutdown process, closes the client connection, clears the timeout, and logs the completion of shutdown.\n\t * Logs any errors that occur during the shutdown process.\n\t */\n\tprivate async gracefulShutdown() {\n\t\tif (this.shuttingDown) {\n\t\t\treturn;\n\t\t}\n\n\t\tthis.shuttingDown = true;\n\t\tlogger.info(\"Starting graceful shutdown of PGlite client...\");\n\n\t\tconst timeout = setTimeout(() => {\n\t\t\tlogger.warn(\n\t\t\t\t\"Shutdown timeout reached, forcing database connection closure...\",\n\t\t\t);\n\t\t\tthis.client.close().finally(() => {\n\t\t\t\tprocess.exit(1);\n\t\t\t});\n\t\t}, this.shutdownTimeout);\n\n\t\ttry {\n\t\t\tawait this.client.close();\n\t\t\tclearTimeout(timeout);\n\t\t\tlogger.info(\"PGlite client shutdown completed successfully\");\n\t\t\tprocess.exit(0);\n\t\t} catch (error) {\n\t\t\tlogger.error(\"Error during graceful shutdown:\", error);\n\t\t\tprocess.exit(1);\n\t\t}\n\t}\n\n\t/**\n\t * Sets up shutdown handlers for SIGINT, SIGTERM, and beforeExit events to gracefully shutdown the application.\n\t * @private\n\t */\n\tprivate setupShutdownHandlers() {\n\t\tprocess.on(\"SIGINT\", async () => {\n\t\t\tawait this.gracefulShutdown();\n\t\t});\n\n\t\tprocess.on(\"SIGTERM\", async () => {\n\t\t\tawait this.gracefulShutdown();\n\t\t});\n\n\t\tprocess.on(\"beforeExit\", async () => {\n\t\t\tawait this.gracefulShutdown();\n\t\t});\n\t}\n\n\t/**\n\t * Initializes the client for PGlite.\n\t *\n\t * @returns {Promise<void>} A Promise that resolves when the client is initialized successfully\n\t */\n\tpublic async initialize(): Promise<void> {\n\t\ttry {\n\t\t\tawait this.client.waitReady;\n\t\t\tlogger.info(\"PGlite client initialized successfully\");\n\t\t} catch (error) {\n\t\t\tlogger.error(\"Failed to initialize PGlite client:\", error);\n\t\t\tthrow error;\n\t\t}\n\t}\n\n\t/**\n\t * Asynchronously closes the resource. If the resource is not already shutting down,\n\t * it performs a graceful shutdown before closing.\n\t *\n\t * @returns A promise that resolves once the resource has been closed.\n\t */\n\tpublic async close(): Promise<void> {\n\t\tif (!this.shuttingDown) {\n\t\t\tawait this.gracefulShutdown();\n\t\t}\n\t}\n\n\t/**\n\t * Check if the system is currently shutting down.\n\t *\n\t * @returns {boolean} True if the system is shutting down, false otherwise.\n\t */\n\tpublic isShuttingDown(): boolean {\n\t\treturn this.shuttingDown;\n\t}\n\n\t/**\n\t * Asynchronously runs database migrations using Drizzle.\n\t * \n\t * Drizzle will first check if the migrations are already applied.\n\t * If there is a diff between database schema and migrations, it will apply the migrations.\n\t * If they are already applied, it will skip them.\n\t *\n\t * @returns {Promise<void>} A Promise that resolves once the migrations are completed successfully.\n\t */\n\tasync runMigrations(): Promise<void> {\n\t\ttry {\n\t\t\tconst db = drizzle(this.client);\n\n\t\t\tconst __filename = fileURLToPath(import.meta.url);\n\t\t\tconst __dirname = pathDirname(__filename);\n\n\t\t\tawait migrate(db, {\n\t\t\t\tmigrationsFolder: pathResolve(__dirname, \"../drizzle/migrations\"),\n\t\t\t});\n\t\t} catch (error) {\n\t\t\tlogger.error(\"Failed to run database migrations (pglite):\", error);\n\t\t\t// throw error;\n\t\t\tconsole.trace(error);\n\t\t}\n\t}\n}\n","import path from \"node:path\";\nimport { fileURLToPath } from \"node:url\";\nimport { logger } from \"@elizaos/core\";\nimport { drizzle } from \"drizzle-orm/node-postgres\";\nimport { migrate } from \"drizzle-orm/node-postgres/migrator\";\nimport pkg, { type Pool as PgPool } from \"pg\";\nimport type { IDatabaseClientManager } from \"../types\";\n\nconst { Pool } = pkg;\n\n/**\n * Manages connections to a PostgreSQL database using a connection pool.\n * Implements IDatabaseClientManager interface.\n */\n\nexport class PostgresConnectionManager\n\timplements IDatabaseClientManager<PgPool>\n{\n\tprivate pool: PgPool;\n\tprivate isShuttingDown = false;\n\tprivate readonly connectionTimeout: number = 5000;\n\n\t/**\n\t * Constructor for creating a connection pool.\n\t * @param {string} connectionString - The connection string used to connect to the database.\n\t */\n\tconstructor(connectionString: string) {\n\t\tconst defaultConfig = {\n\t\t\tmax: 20,\n\t\t\tidleTimeoutMillis: 30000,\n\t\t\tconnectionTimeoutMillis: this.connectionTimeout,\n\t\t};\n\n\t\tthis.pool = new Pool({\n\t\t\t...defaultConfig,\n\t\t\tconnectionString,\n\t\t});\n\n\t\tthis.pool.on(\"error\", (err) => {\n\t\t\tlogger.error(\"Unexpected pool error\", err);\n\t\t\tthis.handlePoolError(err);\n\t\t});\n\n\t\tthis.setupPoolErrorHandling();\n\t\tthis.testConnection();\n\t}\n\n\t/**\n\t * Handles a pool error by attempting to reconnect the pool.\n\t *\n\t * @param {Error} error The error that occurred in the pool.\n\t * @throws {Error} If failed to reconnect the pool.\n\t */\n\tprivate async handlePoolError(error: Error) {\n\t\tlogger.error(\"Pool error occurred, attempting to reconnect\", {\n\t\t\terror: error.message,\n\t\t});\n\n\t\ttry {\n\t\t\tawait this.pool.end();\n\n\t\t\tthis.pool = new Pool({\n\t\t\t\t...this.pool.options,\n\t\t\t\tconnectionTimeoutMillis: this.connectionTimeout,\n\t\t\t});\n\n\t\t\tawait this.testConnection();\n\t\t\tlogger.success(\"Pool reconnection successful\");\n\t\t} catch (reconnectError) {\n\t\t\tlogger.error(\"Failed to reconnect pool\", {\n\t\t\t\terror:\n\t\t\t\t\treconnectError instanceof Error\n\t\t\t\t\t\t? reconnectError.message\n\t\t\t\t\t\t: String(reconnectError),\n\t\t\t});\n\t\t\tthrow reconnectError;\n\t\t}\n\t}\n\n\t/**\n\t * Asynchronously tests the database connection by executing a query to get the current timestamp.\n\t *\n\t * @returns {Promise<boolean>} - A Promise that resolves to true if the database connection test is successful.\n\t */\n\tasync testConnection(): Promise<boolean> {\n\t\tlet client: pkg.PoolClient | null = null;\n\t\ttry {\n\t\t\tclient = await this.pool.connect();\n\t\t\tconst result = await client.query(\"SELECT NOW()\");\n\t\t\tlogger.success(\"Database connection test successful:\", result.rows[0]);\n\t\t\treturn true;\n\t\t} catch (error) {\n\t\t\tlogger.error(\"Database connection test failed:\", error);\n\t\t\tthrow new Error(\n\t\t\t\t`Failed to connect to database: ${(error as Error).message}`,\n\t\t\t);\n\t\t} finally {\n\t\t\tif (client) client.release();\n\t\t}\n\t}\n\n\t/**\n\t * Sets up event listeners to handle pool cleanup on SIGINT, SIGTERM, and beforeExit events.\n\t */\n\tprivate setupPoolErrorHandling() {\n\t\tprocess.on(\"SIGINT\", async () => {\n\t\t\tawait this.cleanup();\n\t\t\tprocess.exit(0);\n\t\t});\n\n\t\tprocess.on(\"SIGTERM\", async () => {\n\t\t\tawait this.cleanup();\n\t\t\tprocess.exit(0);\n\t\t});\n\n\t\tprocess.on(\"beforeExit\", async () => {\n\t\t\tawait this.cleanup();\n\t\t});\n\t}\n\n\t/**\n\t * Get the connection pool.\n\t * @returns {PgPool} The connection pool\n\t * @throws {Error} If the connection manager is shutting down or an error occurs when trying to get the connection from the pool\n\t */\n\tpublic getConnection(): PgPool {\n\t\tif (this.isShuttingDown) {\n\t\t\tthrow new Error(\"Connection manager is shutting down\");\n\t\t}\n\n\t\ttry {\n\t\t\treturn this.pool;\n\t\t} catch (error) {\n\t\t\tlogger.error(\"Failed to get connection from pool:\", error);\n\t\t\tthrow error;\n\t\t}\n\t}\n\n\t/**\n\t * Asynchronously acquires a database client from the connection pool.\n\t *\n\t * @returns {Promise<pkg.PoolClient>} A Promise that resolves with the acquired database client.\n\t * @throws {Error} If an error occurs while acquiring the database client.\n\t */\n\tpublic async getClient(): Promise<pkg.PoolClient> {\n\t\ttry {\n\t\t\treturn await this.pool.connect();\n\t\t} catch (error) {\n\t\t\tlogger.error(\"Failed to acquire a database client:\", error);\n\t\t\tthrow error;\n\t\t}\n\t}\n\n\t/**\n\t * Initializes the PostgreSQL connection manager by testing the connection and logging the result.\n\t *\n\t * @returns {Promise<void>} A Promise that resolves once the manager is successfully initialized\n\t * @throws {Error} If there is an error initializing the connection manager\n\t */\n\tpublic async initialize(): Promise<void> {\n\t\ttry {\n\t\t\tawait this.testConnection();\n\t\t\tlogger.info(\"PostgreSQL connection manager initialized successfully\");\n\t\t} catch (error) {\n\t\t\tlogger.error(\"Failed to initialize connection manager:\", error);\n\t\t\tthrow error;\n\t\t}\n\t}\n\n\t/**\n\t * Asynchronously close the current process by executing a cleanup function.\n\t * @returns A promise that resolves once the cleanup is complete.\n\t */\n\tpublic async close(): Promise<void> {\n\t\tawait this.cleanup();\n\t}\n\n\t/**\n\t * Cleans up and closes the database pool.\n\t * @returns {Promise<void>} A Promise that resolves when the database pool is closed.\n\t */\n\tasync cleanup(): Promise<void> {\n\t\ttry {\n\t\t\tawait this.pool.end();\n\t\t\tlogger.info(\"Database pool closed\");\n\t\t} catch (error) {\n\t\t\tlogger.error(\"Error closing database pool:\", error);\n\t\t}\n\t}\n\n\t/**\n\t * Asynchronously runs database migrations using the Drizzle library.\n\t *\n\t * Drizzle will first check if the migrations are already applied.\n\t * If there is a diff between database schema and migrations, it will apply the migrations.\n\t * If they are already applied, it will skip them.\n\t *\n\t * @returns {Promise<void>} A Promise that resolves once the migrations are completed successfully.\n\t */\n\tasync runMigrations(): Promise<void> {\n\t\ttry {\n\t\t\tconst db = drizzle(this.pool);\n\n\t\t\tconst __filename = fileURLToPath(import.meta.url);\n\t\t\tconst __dirname = path.dirname(__filename);\n\n\t\t\tawait migrate(db, {\n\t\t\t\tmigrationsFolder: path.resolve(__dirname, \"../drizzle/migrations\"),\n\t\t\t});\n\t\t} catch (error) {\n\t\t\tlogger.error(\"Failed to run database migrations (pg):\", error);\n\t\t\tconsole.trace(error);\n\t\t}\n\t}\n}\n"],"mappings":";;;;AAAA,SAAS,WAAW,aAAa,WAAW,mBAAmB;AAC/D,SAAS,qBAAqB;AAC9B,SAAS,cAAkC;AAC3C,SAAS,qBAAqB;AAC9B,SAAS,cAAc;AACvB,SAAS,cAAc;AACvB,SAAS,eAAe;AACxB,SAAS,eAAe;AAOjB,IAAM,sBAAN,MAAoE;AAAA,EAd3E,OAc2E;AAAA;AAAA;AAAA,EAClE;AAAA,EACA,eAAe;AAAA,EACN,kBAAkB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOnC,YAAY,SAAwB;AACnC,SAAK,SAAS,IAAI,OAAO;AAAA,MACxB,GAAG;AAAA,MACH,YAAY;AAAA,QACX;AAAA,QACA;AAAA,MACD;AAAA,IACD,CAAC;AACD,SAAK,sBAAsB;AAAA,EAC5B;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQO,gBAAwB;AAC9B,QAAI,KAAK,cAAc;AACtB,YAAM,IAAI,MAAM,iCAAiC;AAAA,IAClD;AACA,WAAO,KAAK;AAAA,EACb;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA,MAAc,mBAAmB;AAChC,QAAI,KAAK,cAAc;AACtB;AAAA,IACD;AAEA,SAAK,eAAe;AACpB,WAAO,KAAK,gDAAgD;AAE5D,UAAM,UAAU,WAAW,MAAM;AAChC,aAAO;AAAA,QACN;AAAA,MACD;AACA,WAAK,OAAO,MAAM,EAAE,QAAQ,MAAM;AACjC,gBAAQ,KAAK,CAAC;AAAA,MACf,CAAC;AAAA,IACF,GAAG,KAAK,eAAe;AAEvB,QAAI;AACH,YAAM,KAAK,OAAO,MAAM;AACxB,mBAAa,OAAO;AACpB,aAAO,KAAK,+CAA+C;AAC3D,cAAQ,KAAK,CAAC;AAAA,IACf,SAAS,OAAO;AACf,aAAO,MAAM,mCAAmC,KAAK;AACrD,cAAQ,KAAK,CAAC;AAAA,IACf;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA,EAMQ,wBAAwB;AAC/B,YAAQ,GAAG,UAAU,YAAY;AAChC,YAAM,KAAK,iBAAiB;AAAA,IAC7B,CAAC;AAED,YAAQ,GAAG,WAAW,YAAY;AACjC,YAAM,KAAK,iBAAiB;AAAA,IAC7B,CAAC;AAED,YAAQ,GAAG,cAAc,YAAY;AACpC,YAAM,KAAK,iBAAiB;AAAA,IAC7B,CAAC;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAa,aAA4B;AACxC,QAAI;AACH,YAAM,KAAK,OAAO;AAClB,aAAO,KAAK,wCAAwC;AAAA,IACrD,SAAS,OAAO;AACf,aAAO,MAAM,uCAAuC,KAAK;AACzD,YAAM;AAAA,IACP;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,MAAa,QAAuB;AACnC,QAAI,CAAC,KAAK,cAAc;AACvB,YAAM,KAAK,iBAAiB;AAAA,IAC7B;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOO,iBAA0B;AAChC,WAAO,KAAK;AAAA,EACb;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAWA,MAAM,gBAA+B;AACpC,QAAI;AACH,YAAM,KAAK,QAAQ,KAAK,MAAM;AAE9B,YAAM,aAAa,cAAc,YAAY,GAAG;AAChD,YAAM,YAAY,YAAY,UAAU;AAExC,YAAM,QAAQ,IAAI;AAAA,QACjB,kBAAkB,YAAY,WAAW,uBAAuB;AAAA,MACjE,CAAC;AAAA,IACF,SAAS,OAAO;AACf,aAAO,MAAM,+CAA+C,KAAK;AAEjE,cAAQ,MAAM,KAAK;AAAA,IACpB;AAAA,EACD;AACD;;;ACnKA,OAAO,UAAU;AACjB,SAAS,iBAAAA,sBAAqB;AAC9B,SAAS,UAAAC,eAAc;AACvB,SAAS,WAAAC,gBAAe;AACxB,SAAS,WAAAC,gBAAe;AACxB,OAAO,SAAkC;AAGzC,IAAM,EAAE,KAAK,IAAI;AAOV,IAAM,4BAAN,MAEP;AAAA,EAjBA,OAiBA;AAAA;AAAA;AAAA,EACS;AAAA,EACA,iBAAiB;AAAA,EACR,oBAA4B;AAAA;AAAA;AAAA;AAAA;AAAA,EAM7C,YAAY,kBAA0B;AACrC,UAAM,gBAAgB;AAAA,MACrB,KAAK;AAAA,MACL,mBAAmB;AAAA,MACnB,yBAAyB,KAAK;AAAA,IAC/B;AAEA,SAAK,OAAO,IAAI,KAAK;AAAA,MACpB,GAAG;AAAA,MACH;AAAA,IACD,CAAC;AAED,SAAK,KAAK,GAAG,SAAS,CAAC,QAAQ;AAC9B,MAAAC,QAAO,MAAM,yBAAyB,GAAG;AACzC,WAAK,gBAAgB,GAAG;AAAA,IACzB,CAAC;AAED,SAAK,uBAAuB;AAC5B,SAAK,eAAe;AAAA,EACrB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,MAAc,gBAAgB,OAAc;AAC3C,IAAAA,QAAO,MAAM,gDAAgD;AAAA,MAC5D,OAAO,MAAM;AAAA,IACd,CAAC;AAED,QAAI;AACH,YAAM,KAAK,KAAK,IAAI;AAEpB,WAAK,OAAO,IAAI,KAAK;AAAA,QACpB,GAAG,KAAK,KAAK;AAAA,QACb,yBAAyB,KAAK;AAAA,MAC/B,CAAC;AAED,YAAM,KAAK,eAAe;AAC1B,MAAAA,QAAO,QAAQ,8BAA8B;AAAA,IAC9C,SAAS,gBAAgB;AACxB,MAAAA,QAAO,MAAM,4BAA4B;AAAA,QACxC,OACC,0BAA0B,QACvB,eAAe,UACf,OAAO,cAAc;AAAA,MAC1B,CAAC;AACD,YAAM;AAAA,IACP;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAM,iBAAmC;AACxC,QAAI,SAAgC;AACpC,QAAI;AACH,eAAS,MAAM,KAAK,KAAK,QAAQ;AACjC,YAAM,SAAS,MAAM,OAAO,MAAM,cAAc;AAChD,MAAAA,QAAO,QAAQ,wCAAwC,OAAO,KAAK,CAAC,CAAC;AACrE,aAAO;AAAA,IACR,SAAS,OAAO;AACf,MAAAA,QAAO,MAAM,oCAAoC,KAAK;AACtD,YAAM,IAAI;AAAA,QACT,kCAAmC,MAAgB,OAAO;AAAA,MAC3D;AAAA,IACD,UAAE;AACD,UAAI,OAAQ,QAAO,QAAQ;AAAA,IAC5B;AAAA,EACD;AAAA;AAAA;AAAA;AAAA,EAKQ,yBAAyB;AAChC,YAAQ,GAAG,UAAU,YAAY;AAChC,YAAM,KAAK,QAAQ;AACnB,cAAQ,KAAK,CAAC;AAAA,IACf,CAAC;AAED,YAAQ,GAAG,WAAW,YAAY;AACjC,YAAM,KAAK,QAAQ;AACnB,cAAQ,KAAK,CAAC;AAAA,IACf,CAAC;AAED,YAAQ,GAAG,cAAc,YAAY;AACpC,YAAM,KAAK,QAAQ;AAAA,IACpB,CAAC;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOO,gBAAwB;AAC9B,QAAI,KAAK,gBAAgB;AACxB,YAAM,IAAI,MAAM,qCAAqC;AAAA,IACtD;AAEA,QAAI;AACH,aAAO,KAAK;AAAA,IACb,SAAS,OAAO;AACf,MAAAA,QAAO,MAAM,uCAAuC,KAAK;AACzD,YAAM;AAAA,IACP;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,MAAa,YAAqC;AACjD,QAAI;AACH,aAAO,MAAM,KAAK,KAAK,QAAQ;AAAA,IAChC,SAAS,OAAO;AACf,MAAAA,QAAO,MAAM,wCAAwC,KAAK;AAC1D,YAAM;AAAA,IACP;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,MAAa,aAA4B;AACxC,QAAI;AACH,YAAM,KAAK,eAAe;AAC1B,MAAAA,QAAO,KAAK,wDAAwD;AAAA,IACrE,SAAS,OAAO;AACf,MAAAA,QAAO,MAAM,4CAA4C,KAAK;AAC9D,YAAM;AAAA,IACP;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAa,QAAuB;AACnC,UAAM,KAAK,QAAQ;AAAA,EACpB;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAM,UAAyB;AAC9B,QAAI;AACH,YAAM,KAAK,KAAK,IAAI;AACpB,MAAAA,QAAO,KAAK,sBAAsB;AAAA,IACnC,SAAS,OAAO;AACf,MAAAA,QAAO,MAAM,gCAAgC,KAAK;AAAA,IACnD;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAWA,MAAM,gBAA+B;AACpC,QAAI;AACH,YAAM,KAAKC,SAAQ,KAAK,IAAI;AAE5B,YAAM,aAAaC,eAAc,YAAY,GAAG;AAChD,YAAM,YAAY,KAAK,QAAQ,UAAU;AAEzC,YAAMC,SAAQ,IAAI;AAAA,QACjB,kBAAkB,KAAK,QAAQ,WAAW,uBAAuB;AAAA,MAClE,CAAC;AAAA,IACF,SAAS,OAAO;AACf,MAAAH,QAAO,MAAM,2CAA2C,KAAK;AAC7D,cAAQ,MAAM,KAAK;AAAA,IACpB;AAAA,EACD;AACD;","names":["fileURLToPath","logger","drizzle","migrate","logger","drizzle","fileURLToPath","migrate"]}