Rewrite the data access layer onto Prisma Client

2026-07-31 01:27:18 +02:00
parent 439453be38
commit 3a04da3180
4 changed files with 119 additions and 147 deletions
@@ -1,28 +1,23 @@
-import mariadb from 'mariadb';
+import { PrismaClient } from '@prisma/client';

-let pool;
+let prisma;

 export function buildDatabaseUrl(config) {
  const { host, user, password, database } = config.db;
  return `mysql://${encodeURIComponent(user)}:${encodeURIComponent(password)}@${host}:3306/${database}?connection_limit=10`;
 }

-export function getPool(config) {
-  if (!pool) {
-    pool = mariadb.createPool({
-      host: config.db.host,
-      user: config.db.user,
-      password: config.db.password,
-      database: config.db.database,
-      connectionLimit: 10,
-    });
+export function getPrismaClient(config) {
+  if (!prisma) {
+    process.env.DATABASE_URL = buildDatabaseUrl(config);
+    prisma = new PrismaClient();
  }
-  return pool;
+  return prisma;
 }

-export async function closePool() {
-  if (pool) {
-    await pool.end();
-    pool = undefined;
+export async function closePrismaClient() {
+  if (prisma) {
+    await prisma.$disconnect();
+    prisma = undefined;
  }
 }
@@ -1,109 +1,86 @@
-function toCamelJob(row) {
-  if (!row) return null;
-  return {
-    id: row.id,
-    uuid: row.uuid,
-    status: row.status,
-    family: row.family,
-    sourceFormat: row.source_format,
-    targetFormat: row.target_format,
-    originalFilename: row.original_filename,
-    inputPath: row.input_path,
-    outputPath: row.output_path,
-    inputMimeType: row.input_mime_type,
-    outputMimeType: row.output_mime_type,
-    inputSizeBytes: row.input_size_bytes,
-    outputSizeBytes: row.output_size_bytes,
-    quality: row.quality,
-    conversionDurationSeconds: row.conversion_duration_seconds,
-    errorMessage: row.error_message,
-    createdAt: row.created_at,
-    updatedAt: row.updated_at,
-    expiresAt: row.expires_at,
-    cleanedAt: row.cleaned_at,
-  };
+const jobSelect = {
+  id: true,
+  uuid: true,
+  status: true,
+  family: true,
+  sourceFormat: true,
+  targetFormat: true,
+  originalFilename: true,
+  inputPath: true,
+  outputPath: true,
+  inputMimeType: true,
+  outputMimeType: true,
+  inputSizeBytes: true,
+  outputSizeBytes: true,
+  quality: true,
+  conversionDurationSeconds: true,
+  errorMessage: true,
+  createdAt: true,
+  updatedAt: true,
+  expiresAt: true,
+  cleanedAt: true,
+};
+
+export async function createJob(prisma, job) {
+  await prisma.conversionJob.create({
+    data: {
+      uuid: job.uuid,
+      family: job.family,
+      sourceFormat: job.sourceFormat,
+      targetFormat: job.targetFormat,
+      originalFilename: job.originalFilename,
+      inputPath: job.inputPath,
+      inputMimeType: job.inputMimeType,
+      inputSizeBytes: job.inputSizeBytes,
+      expiresAt: job.expiresAt,
+      quality: job.quality ?? null,
+    },
+  });
 }

-export async function createJob(pool, job) {
-  await pool.query(
-    `INSERT INTO conversion_jobs
-      (uuid, status, family, source_format, target_format, original_filename, input_path, input_mime_type, input_size_bytes, expires_at, quality)
-     VALUES (?, 'pending', ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
-    [
-      job.uuid,
-      job.family,
-      job.sourceFormat,
-      job.targetFormat,
-      job.originalFilename,
-      job.inputPath,
-      job.inputMimeType,
-      job.inputSizeBytes,
-      job.expiresAt,
-      job.quality ?? null,
-    ]
-  );
+export async function getJobByUuid(prisma, uuid) {
+  return prisma.conversionJob.findUnique({ where: { uuid }, select: jobSelect });
 }

-export async function getJobByUuid(pool, uuid) {
-  const rows = await pool.query(
-    `SELECT id, uuid, status, family, source_format, target_format, original_filename,
-            input_path, output_path, input_mime_type, output_mime_type,
-            input_size_bytes, output_size_bytes, quality, conversion_duration_seconds,
-            error_message, created_at, updated_at, expires_at, cleaned_at
-     FROM conversion_jobs WHERE uuid = ?`,
-    [uuid]
-  );
-  return toCamelJob(rows[0]);
+export async function getJobErrorLog(prisma, id) {
+  const job = await prisma.conversionJob.findUnique({ where: { id }, select: { errorLog: true } });
+  return job?.errorLog ?? null;
 }

-export async function getJobErrorLog(pool, id) {
-  const rows = await pool.query('SELECT error_log FROM conversion_jobs WHERE id = ?', [id]);
-  return rows[0]?.error_log ?? null;
+export async function markProcessing(prisma, id) {
+  await prisma.conversionJob.update({ where: { id }, data: { status: 'processing' } });
 }

-export async function markProcessing(pool, id) {
-  await pool.query("UPDATE conversion_jobs SET status = 'processing' WHERE id = ?", [id]);
+export async function markDone(prisma, id, { outputPath, outputMimeType, outputSizeBytes, conversionDurationSeconds }) {
+  await prisma.conversionJob.update({
+    where: { id },
+    data: { status: 'done', outputPath, outputMimeType, outputSizeBytes, conversionDurationSeconds },
+  });
 }

-export async function markDone(pool, id, { outputPath, outputMimeType, outputSizeBytes, conversionDurationSeconds }) {
-  await pool.query(
-    `UPDATE conversion_jobs
-     SET status = 'done', output_path = ?, output_mime_type = ?, output_size_bytes = ?, conversion_duration_seconds = ?
-     WHERE id = ?`,
-    [outputPath, outputMimeType, outputSizeBytes, conversionDurationSeconds, id]
-  );
+export async function markFailed(prisma, id, { errorMessage, errorLog }) {
+  await prisma.conversionJob.update({
+    where: { id },
+    data: { status: 'failed', errorMessage, errorLog },
+  });
 }

-export async function markFailed(pool, id, { errorMessage, errorLog }) {
-  await pool.query(
-    "UPDATE conversion_jobs SET status = 'failed', error_message = ?, error_log = ? WHERE id = ?",
-    [errorMessage, errorLog, id]
-  );
+export async function findPendingJobs(prisma, limit) {
+  return prisma.conversionJob.findMany({
+    where: { status: 'pending' },
+    orderBy: { createdAt: 'asc' },
+    take: limit,
+    select: jobSelect,
+  });
 }

-export async function findPendingJobs(pool, limit) {
-  const rows = await pool.query(
-    `SELECT id, uuid, status, family, source_format, target_format, original_filename,
-            input_path, output_path, input_mime_type, output_mime_type,
-            input_size_bytes, output_size_bytes, quality, conversion_duration_seconds,
-            error_message, created_at, updated_at, expires_at, cleaned_at
-     FROM conversion_jobs WHERE status = 'pending' ORDER BY created_at ASC LIMIT ?`,
-    [limit]
-  );
-  return rows.map(toCamelJob);
+export async function findExpiredJobs(prisma) {
+  return prisma.conversionJob.findMany({
+    where: { expiresAt: { lt: new Date() }, cleanedAt: null },
+    select: jobSelect,
+  });
 }

-export async function findExpiredJobs(pool) {
-  const rows = await pool.query(
-    `SELECT id, uuid, status, family, source_format, target_format, original_filename,
-            input_path, output_path, input_mime_type, output_mime_type,
-            input_size_bytes, output_size_bytes, quality, conversion_duration_seconds,
-            error_message, created_at, updated_at, expires_at, cleaned_at
-     FROM conversion_jobs WHERE expires_at < NOW() AND cleaned_at IS NULL`
-  );
-  return rows.map(toCamelJob);
-}
-
-export async function markCleaned(pool, id) {
-  await pool.query('UPDATE conversion_jobs SET cleaned_at = NOW() WHERE id = ?', [id]);
+export async function markCleaned(prisma, id) {
+  await prisma.conversionJob.update({ where: { id }, data: { cleanedAt: new Date() } });
 }