feat: log sql queries (#444)

This commit is contained in:
Manuel
2024-05-06 20:14:51 +02:00
committed by GitHub
parent 3cda718474
commit dc0184af07
3 changed files with 18 additions and 1 deletions

View File

@@ -1,9 +1,12 @@
import Database from "better-sqlite3";
import type { Logger } from "drizzle-orm";
import type { BetterSQLite3Database } from "drizzle-orm/better-sqlite3";
import { drizzle as drizzleSqlite } from "drizzle-orm/better-sqlite3";
import { drizzle as drizzleMysql } from "drizzle-orm/mysql2";
import mysql from "mysql2";
import { logger } from "@homarr/log";
import * as mysqlSchema from "./schema/mysql";
import * as sqliteSchema from "./schema/sqlite";
@@ -25,9 +28,18 @@ const init = () => {
export let connection: Database.Database | mysql.Connection;
export let database: HomarrDatabase;
class WinstonDrizzleLogger implements Logger {
logQuery(query: string, _: unknown[]): void {
logger.debug(`Executed SQL query: ${query}`);
}
}
const initBetterSqlite = () => {
connection = new Database(process.env.DB_URL);
database = drizzleSqlite(connection, { schema: sqliteSchema });
database = drizzleSqlite(connection, {
schema: sqliteSchema,
logger: new WinstonDrizzleLogger(),
});
};
const initMySQL2 = () => {
@@ -46,6 +58,7 @@ const initMySQL2 = () => {
database = drizzleMysql(connection, {
schema: mysqlSchema,
mode: "default",
logger: new WinstonDrizzleLogger(),
}) as unknown as HomarrDatabase;
};

View File

@@ -27,6 +27,7 @@
"dependencies": {
"@homarr/common": "workspace:^0.1.0",
"@homarr/definitions": "workspace:^0.1.0",
"@homarr/log": "workspace:^0.1.0",
"@paralleldrive/cuid2": "^2.2.2",
"better-sqlite3": "^9.6.0",
"drizzle-orm": "^0.30.10",

3
pnpm-lock.yaml generated
View File

@@ -485,6 +485,9 @@ importers:
'@homarr/definitions':
specifier: workspace:^0.1.0
version: link:../definitions
'@homarr/log':
specifier: workspace:^0.1.0
version: link:../log
'@paralleldrive/cuid2':
specifier: ^2.2.2
version: 2.2.2