homeio

1import "server-only";
2
3import type { QueryResult, QueryResultRow } from "pg";
4import { pgPool } from "@/lib/server/db/postgres";
5import { withServerTiming } from "@/lib/server/logging/logger";
6
7function compactSql(sql: string) {
8  return sql.replace(/\s+/g, " ").trim();
9}
10
11export async function timedPgQuery<T extends QueryResultRow>(
12  text: string,
13  values?: readonly unknown[],
14): Promise<QueryResult<T>> {
15  const compact = compactSql(text);
16
17  return withServerTiming(
18    {
19      level: "debug",
20      layer: "db",
21      action: "pg.query",
22      meta: {
23        query: compact,
24        paramCount: values?.length ?? 0,
25      },
26      onSuccessMeta: (result) => ({
27        rowCount:
28          typeof result === "object" &&
29          result !== null &&
30          "rowCount" in result &&
31          typeof (result as { rowCount?: number | null }).rowCount === "number"
32            ? (result as { rowCount: number }).rowCount
33            : 0,
34      }),
35    },
36    () => pgPool.query<T>(text, values as unknown[] | undefined),
37  );
38}
39