github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Move to a single prisma client connection for the ui backend to prevent competing connections.

Browse Source
This commit is contained in:
Jaret Burkett
2026-07-21 11:21:28 -06:00
parent 9e9439015e
commit e54a0fe78c
18 changed files with 33 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
ui/src/app/api/jobs/[jobID]/delete/route.ts
View File

@@ -1,11 +1,9 @@
import { NextRequest, NextResponse } from 'next/server';
import { PrismaClient } from '@prisma/client';
import prisma from '@/server/prisma';
import { getTrainingFolder } from '@/server/settings';
import path from 'path';
import fs from 'fs';
const prisma = new PrismaClient();
export async function GET(request: NextRequest, { params }: { params: { jobID: string } }) {
const { jobID } = await params;

4
ui/src/app/api/jobs/[jobID]/files/route.ts
View File

@@ -1,11 +1,9 @@
import { NextRequest, NextResponse } from 'next/server';
import { PrismaClient } from '@prisma/client';
import prisma from '@/server/prisma';
import path from 'path';
import fs from 'fs';
import { getTrainingFolder } from '@/server/settings';
const prisma = new PrismaClient();
export async function GET(request: NextRequest, { params }: { params: { jobID: string } }) {
const { jobID } = await params;

4
ui/src/app/api/jobs/[jobID]/log/route.ts
View File

@@ -1,12 +1,10 @@
import { NextRequest, NextResponse } from 'next/server';
import { PrismaClient } from '@prisma/client';
import prisma from '@/server/prisma';
import path from 'path';
import fs from 'fs';
import type { FileHandle } from 'fs/promises';
import { getTrainingFolder } from '@/server/settings';
const prisma = new PrismaClient();
export async function GET(request: NextRequest, { params }: { params: { jobID: string } }) {
const { jobID } = await params;

4
ui/src/app/api/jobs/[jobID]/loss/route.ts
View File

@@ -1,5 +1,5 @@
import { NextRequest, NextResponse } from 'next/server';
import { PrismaClient } from '@prisma/client';
import prisma from '@/server/prisma';
import path from 'path';
import fs from 'fs';
import { getTrainingFolder } from '@/server/settings';
@@ -8,8 +8,6 @@ import sqlite3 from 'sqlite3';
export const runtime = 'nodejs';
const prisma = new PrismaClient();
function openDb(filename: string) {
const db = new sqlite3.Database(filename);
db.configure('busyTimeout', 30_000);

4
ui/src/app/api/jobs/[jobID]/mark_stopped/route.ts
View File

@@ -1,7 +1,5 @@
import { NextRequest, NextResponse } from 'next/server';
import { PrismaClient } from '@prisma/client';
const prisma = new PrismaClient();
import prisma from '@/server/prisma';
export async function GET(request: NextRequest, { params }: { params: { jobID: string } }) {
const { jobID } = await params;

4
ui/src/app/api/jobs/[jobID]/plugin/route.ts
View File

@@ -1,11 +1,9 @@
import { NextRequest, NextResponse } from 'next/server';
import { PrismaClient } from '@prisma/client';
import prisma from '@/server/prisma';
import path from 'path';
import fs from 'fs';
import { getTrainingFolder } from '@/server/settings';
const prisma = new PrismaClient();
export async function GET(request: NextRequest, { params }: { params: { jobID: string } }) {
const { jobID } = await params;

4
ui/src/app/api/jobs/[jobID]/sample_now/route.ts
View File

@@ -1,7 +1,5 @@
import { NextRequest, NextResponse } from 'next/server';
import { PrismaClient } from '@prisma/client';
const prisma = new PrismaClient();
import prisma from '@/server/prisma';
export async function GET(request: NextRequest, { params }: { params: { jobID: string } }) {
const { jobID } = await params;

4
ui/src/app/api/jobs/[jobID]/samples/route.ts
View File

@@ -1,11 +1,9 @@
import { NextRequest, NextResponse } from 'next/server';
import { PrismaClient } from '@prisma/client';
import prisma from '@/server/prisma';
import path from 'path';
import fs from 'fs';
import { getTrainingFolder } from '@/server/settings';
const prisma = new PrismaClient();
export async function GET(request: NextRequest, { params }: { params: { jobID: string } }) {
const { jobID } = await params;

4
ui/src/app/api/jobs/[jobID]/save_now/route.ts
View File

@@ -1,7 +1,5 @@
import { NextRequest, NextResponse } from 'next/server';
import { PrismaClient } from '@prisma/client';
const prisma = new PrismaClient();
import prisma from '@/server/prisma';
export async function GET(request: NextRequest, { params }: { params: { jobID: string } }) {
const { jobID } = await params;

4
ui/src/app/api/jobs/[jobID]/start/route.ts
View File

@@ -1,7 +1,5 @@
import { NextRequest, NextResponse } from 'next/server';
import { PrismaClient } from '@prisma/client';
const prisma = new PrismaClient();
import prisma from '@/server/prisma';
export async function GET(request: NextRequest, { params }: { params: { jobID: string } }) {
const { jobID } = await params;

3
ui/src/app/api/jobs/[jobID]/stop/route.ts
View File

@@ -1,10 +1,9 @@
import { NextRequest, NextResponse } from 'next/server';
import { PrismaClient } from '@prisma/client';
import prisma from '@/server/prisma';
import { exec } from 'child_process';
import { promisify } from 'util';
const execAsync = promisify(exec);
const prisma = new PrismaClient();
const isWindows = process.platform === 'win32';
export async function GET(request: NextRequest, { params }: { params: { jobID: string } }) {

4
ui/src/app/api/jobs/route.ts
View File

@@ -1,10 +1,8 @@
import { NextResponse } from 'next/server';
import { PrismaClient } from '@prisma/client';
import prisma from '@/server/prisma';
import { isMac } from '@/helpers/basic';
import { cached } from '@/server/apiCache';
const prisma = new PrismaClient();
export async function GET(request: Request) {
const { searchParams } = new URL(request.url);
const id = searchParams.get('id');

4
ui/src/app/api/queue/[queueID]/start/route.ts
View File

@@ -1,7 +1,5 @@
import { NextRequest, NextResponse } from 'next/server';
import { PrismaClient } from '@prisma/client';
const prisma = new PrismaClient();
import prisma from '@/server/prisma';
export async function GET(request: NextRequest, { params }: { params: { queueID: string } }) {
const { queueID } = await params;

4
ui/src/app/api/queue/[queueID]/stop/route.ts
View File

@@ -1,7 +1,5 @@
import { NextRequest, NextResponse } from 'next/server';
import { PrismaClient } from '@prisma/client';
const prisma = new PrismaClient();
import prisma from '@/server/prisma';
export async function GET(request: NextRequest, { params }: { params: { queueID: string } }) {
const { queueID } = await params;

4
ui/src/app/api/queue/route.ts
View File

@@ -1,7 +1,5 @@
import { NextResponse } from 'next/server';
import { PrismaClient } from '@prisma/client';
const prisma = new PrismaClient();
import prisma from '@/server/prisma';
export async function GET(request: Request) {
const { searchParams } = new URL(request.url);

4
ui/src/app/api/settings/route.ts
View File

@@ -1,10 +1,8 @@
import { NextResponse } from 'next/server';
import { PrismaClient } from '@prisma/client';
import prisma from '@/server/prisma';
import { defaultTrainFolder, defaultDatasetsFolder } from '@/paths';
import { flushCache } from '@/server/settings';
const prisma = new PrismaClient();
export async function GET() {
try {
const settings = await prisma.settings.findMany();

16
ui/src/server/prisma.ts Normal file
View File

@@ -0,0 +1,16 @@
import { PrismaClient } from '@prisma/client';
// Single shared PrismaClient for the whole server. Route files must import
// this instead of constructing their own client — each PrismaClient spawns
// its own query engine + connection pool, and a pile of them contending for
// the same sqlite file just adds lock pressure. The globalThis stash keeps
// dev hot-reload from leaking a new client on every recompile.
const globalForPrisma = globalThis as unknown as { prisma?: PrismaClient };
const prisma = globalForPrisma.prisma ?? new PrismaClient();
if (process.env.NODE_ENV !== 'production') {
globalForPrisma.prisma = prisma;
}
export default prisma;

3
ui/src/server/settings.ts
View File

@@ -1,10 +1,9 @@
import { PrismaClient } from '@prisma/client';
import prisma from '@/server/prisma';
import { defaultDatasetsFolder, defaultDataRoot } from '@/paths';
import { defaultTrainFolder } from '@/paths';
import NodeCache from 'node-cache';
const myCache = new NodeCache();
const prisma = new PrismaClient();
export const flushCache = () => {
myCache.flushAll();