feat: add PostgreSQL client and migration runner

2026-05-14 14:45:21 +07:00
parent 818a059121
commit caf90ea9e6
2 changed files with 301 additions and 0 deletions
@@ -0,0 +1,171 @@
+import { Pool, PoolClient, QueryResult, QueryResultRow } from "pg";
+import { createChildLogger } from "../logger";
+import { config } from "../config";
+
+const logger = createChildLogger("postgres");
+
+export interface PostgresConfig {
+  host: string;
+  port: number;
+  user?: string;
+  password?: string;
+  database?: string;
+  min: number;
+  max: number;
+}
+
+let pool: Pool | null = null;
+
+/**
+ * Parse DATABASE_URL or build config from individual POSTGRES_* variables
+ */
+export function buildConfig(): PostgresConfig {
+  if (config.DATABASE_URL) {
+    try {
+      const url = new URL(config.DATABASE_URL);
+      return {
+        host: url.hostname || "localhost",
+        port: url.port ? parseInt(url.port, 10) : 5432,
+        user: url.username || undefined,
+        password: url.password || undefined,
+        database: url.pathname?.slice(1) || undefined,
+        min: config.POSTGRES_POOL_MIN,
+        max: config.POSTGRES_POOL_MAX,
+      };
+    } catch (error) {
+      logger.warn(
+        { error: error instanceof Error ? error.message : String(error) },
+        "Failed to parse DATABASE_URL, falling back to individual variables",
+      );
+    }
+  }
+
+  return {
+    host: config.POSTGRES_HOST,
+    port: config.POSTGRES_PORT,
+    user: config.POSTGRES_USER,
+    password: config.POSTGRES_PASSWORD,
+    database: config.POSTGRES_DB,
+    min: config.POSTGRES_POOL_MIN,
+    max: config.POSTGRES_POOL_MAX,
+  };
+}
+
+/**
+ * Initialize PostgreSQL connection pool
+ */
+export function initializePool(): Pool {
+  const pgConfig = buildConfig();
+
+  logger.info(
+    {
+      host: pgConfig.host,
+      port: pgConfig.port,
+      database: pgConfig.database,
+      min: pgConfig.min,
+      max: pgConfig.max,
+    },
+    "Initializing PostgreSQL connection pool",
+  );
+
+  const newPool = new Pool({
+    host: pgConfig.host,
+    port: pgConfig.port,
+    user: pgConfig.user,
+    password: pgConfig.password,
+    database: pgConfig.database,
+    min: pgConfig.min,
+    max: pgConfig.max,
+    idleTimeoutMillis: 30000,
+    connectionTimeoutMillis: 2000,
+  });
+
+  newPool.on("error", (error) => {
+    logger.error(
+      { error: error instanceof Error ? error.message : String(error) },
+      "Unexpected error on idle client in pool",
+    );
+  });
+
+  newPool.on("connect", () => {
+    logger.debug("New client connected to pool");
+  });
+
+  newPool.on("remove", () => {
+    logger.debug("Client removed from pool");
+  });
+
+  return newPool;
+}
+
+/**
+ * Get or initialize the connection pool
+ */
+export function getPool(): Pool {
+  if (!pool) {
+    pool = initializePool();
+  }
+  return pool;
+}
+
+/**
+ * Execute a query with type safety
+ */
+export async function query<T extends QueryResultRow = QueryResultRow>(
+  text: string,
+  values?: unknown[],
+): Promise<QueryResult<T>> {
+  const client = await getPool().connect();
+  try {
+    logger.debug({ text, values: values?.length || 0 }, "Executing query");
+    return await client.query<T>(text, values);
+  } catch (error) {
+    logger.error(
+      {
+        text,
+        error: error instanceof Error ? error.message : String(error),
+      },
+      "Query execution failed",
+    );
+    throw error;
+  } finally {
+    client.release();
+  }
+}
+
+/**
+ * Get a client from the pool for transaction or batch operations
+ */
+export async function getClient(): Promise<PoolClient> {
+  try {
+    const client = await getPool().connect();
+    logger.debug("Client acquired from pool");
+    return client;
+  } catch (error) {
+    logger.error(
+      { error: error instanceof Error ? error.message : String(error) },
+      "Failed to acquire client from pool",
+    );
+    throw error;
+  }
+}
+
+/**
+ * Close the connection pool
+ */
+export async function closePool(): Promise<void> {
+  if (pool) {
+    try {
+      logger.info("Closing PostgreSQL connection pool");
+      await pool.end();
+      pool = null;
+      logger.info("PostgreSQL connection pool closed");
+    } catch (error) {
+      logger.error(
+        { error: error instanceof Error ? error.message : String(error) },
+        "Error closing connection pool",
+      );
+      throw error;
+    }
+  }
+}