feat: 新增数据库
This commit is contained in:
@@ -22,3 +22,5 @@ logs
|
||||
.env
|
||||
.env.*
|
||||
!.env.example
|
||||
|
||||
/app/generated/prisma
|
||||
|
||||
@@ -0,0 +1,55 @@
|
||||
-- 表1：用户表
|
||||
CREATE TABLE IF NOT EXISTS `users` (
|
||||
`id` INT NOT NULL COMMENT 'NewAPI 用户 ID，作为本站用户主键',
|
||||
`username` VARCHAR(191) NOT NULL DEFAULT '' COMMENT '用户名快照',
|
||||
`display_name` VARCHAR(191) NOT NULL DEFAULT '' COMMENT '展示名称快照',
|
||||
`group` VARCHAR(191) NOT NULL DEFAULT '' COMMENT '用户分组快照',
|
||||
`role` INT NOT NULL DEFAULT 0 COMMENT '用户角色快照',
|
||||
`status` INT NOT NULL DEFAULT 0 COMMENT '用户状态快照',
|
||||
`created_at` DATETIME(3) NOT NULL DEFAULT CURRENT_TIMESTAMP(3) COMMENT '创建时间',
|
||||
`updated_at` DATETIME(3) NOT NULL DEFAULT CURRENT_TIMESTAMP(3) ON UPDATE CURRENT_TIMESTAMP(3) COMMENT '更新时间',
|
||||
PRIMARY KEY (`id`)
|
||||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci COMMENT='用户表';
|
||||
|
||||
-- 表2：生图记录表
|
||||
CREATE TABLE IF NOT EXISTS `image_generations` (
|
||||
`id` BIGINT UNSIGNED NOT NULL AUTO_INCREMENT COMMENT '生图记录 ID',
|
||||
`user_id` INT NOT NULL COMMENT 'NewAPI 用户 ID',
|
||||
`prompt` TEXT NOT NULL COMMENT '用户输入的生图提示词',
|
||||
`status` ENUM('QUEUED', 'RUNNING', 'SUCCEEDED', 'FAILED') NOT NULL DEFAULT 'RUNNING' COMMENT '生图状态',
|
||||
`model` VARCHAR(64) NOT NULL DEFAULT 'gpt-image-2' COMMENT '生图模型',
|
||||
`size` VARCHAR(32) NOT NULL DEFAULT '1024x1024' COMMENT '图片尺寸',
|
||||
`started_at` DATETIME(3) NOT NULL DEFAULT CURRENT_TIMESTAMP(3) COMMENT '生图开始时间',
|
||||
`ended_at` DATETIME(3) NULL COMMENT '生图结束时间',
|
||||
`duration_ms` INT NULL COMMENT '生图耗时，单位毫秒',
|
||||
`image_url` VARCHAR(2048) NULL COMMENT '上游返回的图片 URL',
|
||||
`image_base64` LONGTEXT NULL COMMENT '服务端下载图片后保存的 base64 原文',
|
||||
`image_mime_type` VARCHAR(191) NULL COMMENT '图片 MIME 类型',
|
||||
`revised_prompt` TEXT NULL COMMENT '上游返回的修订提示词',
|
||||
`upstream_response` JSON NULL COMMENT '完整上游生图接口响应',
|
||||
`error_message` TEXT NULL COMMENT '失败原因或图片下载失败信息',
|
||||
`created_at` DATETIME(3) NOT NULL DEFAULT CURRENT_TIMESTAMP(3) COMMENT '创建时间',
|
||||
`updated_at` DATETIME(3) NOT NULL DEFAULT CURRENT_TIMESTAMP(3) ON UPDATE CURRENT_TIMESTAMP(3) COMMENT '更新时间',
|
||||
`deleted_at` DATETIME(3) NULL COMMENT '软删除时间',
|
||||
PRIMARY KEY (`id`),
|
||||
INDEX `image_generations_user_id_created_at_idx` (`user_id`, `created_at`),
|
||||
INDEX `image_generations_status_idx` (`status`),
|
||||
INDEX `image_generations_deleted_at_idx` (`deleted_at`),
|
||||
CONSTRAINT `image_generations_user_id_fkey`
|
||||
FOREIGN KEY (`user_id`) REFERENCES `users` (`id`)
|
||||
ON DELETE CASCADE ON UPDATE CASCADE
|
||||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci COMMENT='生图记录表';
|
||||
|
||||
-- 表3：生图统计表
|
||||
CREATE TABLE IF NOT EXISTS `generation_stats` (
|
||||
`id` VARCHAR(32) NOT NULL COMMENT '统计行 ID，固定为 global',
|
||||
`total_requests` INT NOT NULL DEFAULT 0 COMMENT '总生图请求数，包含成功和失败',
|
||||
`success_requests` INT NOT NULL DEFAULT 0 COMMENT '成功生图请求数',
|
||||
`failed_requests` INT NOT NULL DEFAULT 0 COMMENT '失败生图请求数',
|
||||
`queued_requests` INT NOT NULL DEFAULT 0 COMMENT '排队中请求数，预留异步任务使用',
|
||||
`running_requests` INT NOT NULL DEFAULT 0 COMMENT '进行中请求数',
|
||||
`total_images` INT NOT NULL DEFAULT 0 COMMENT '成功生成图片总数',
|
||||
`created_at` DATETIME(3) NOT NULL DEFAULT CURRENT_TIMESTAMP(3) COMMENT '创建时间',
|
||||
`updated_at` DATETIME(3) NOT NULL DEFAULT CURRENT_TIMESTAMP(3) ON UPDATE CURRENT_TIMESTAMP(3) COMMENT '更新时间',
|
||||
PRIMARY KEY (`id`)
|
||||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci COMMENT='生图统计表';
|
||||
@@ -17,10 +17,13 @@
|
||||
"@nuxtjs/seo": "5.1.3",
|
||||
"@nuxtjs/tailwindcss": "6.14.0",
|
||||
"@pinia/nuxt": "0.11.3",
|
||||
"@prisma/adapter-mariadb": "^7.8.0",
|
||||
"@prisma/client": "^7.8.0",
|
||||
"@takumi-rs/core": "^1.0.16",
|
||||
"nuxt": "^4.4.2",
|
||||
"openai": "^6.34.0",
|
||||
"pinia": "^3.0.4",
|
||||
"prisma": "^7.8.0",
|
||||
"vue": "^3.5.32",
|
||||
"vue-router": "^5.0.4",
|
||||
"vue-sonner": "^2.0.9"
|
||||
|
||||
Generated
+753
-88
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,14 @@
|
||||
// This file was generated by Prisma, and assumes you have installed the following:
|
||||
// npm install --save-dev prisma dotenv
|
||||
import "dotenv/config";
|
||||
import { defineConfig } from "prisma/config";
|
||||
|
||||
export default defineConfig({
|
||||
schema: "prisma/schema.prisma",
|
||||
migrations: {
|
||||
path: "prisma/migrations",
|
||||
},
|
||||
datasource: {
|
||||
url: process.env["DATABASE_URL"],
|
||||
},
|
||||
});
|
||||
@@ -0,0 +1,64 @@
|
||||
-- CreateTable
|
||||
CREATE TABLE `users` (
|
||||
`id` INT NOT NULL,
|
||||
`username` VARCHAR(191) NOT NULL DEFAULT '',
|
||||
`display_name` VARCHAR(191) NOT NULL DEFAULT '',
|
||||
`group` VARCHAR(191) NOT NULL DEFAULT '',
|
||||
`role` INT NOT NULL DEFAULT 0,
|
||||
`status` INT NOT NULL DEFAULT 0,
|
||||
`created_at` DATETIME(3) NOT NULL DEFAULT CURRENT_TIMESTAMP(3),
|
||||
`updated_at` DATETIME(3) NOT NULL DEFAULT CURRENT_TIMESTAMP(3) ON UPDATE CURRENT_TIMESTAMP(3),
|
||||
PRIMARY KEY (`id`)
|
||||
) DEFAULT CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci;
|
||||
|
||||
-- CreateTable
|
||||
CREATE TABLE `image_generations` (
|
||||
`id` BIGINT UNSIGNED NOT NULL AUTO_INCREMENT,
|
||||
`user_id` INT NOT NULL,
|
||||
`prompt` TEXT NOT NULL,
|
||||
`status` ENUM('QUEUED', 'RUNNING', 'SUCCEEDED', 'FAILED') NOT NULL DEFAULT 'RUNNING',
|
||||
`model` VARCHAR(64) NOT NULL DEFAULT 'gpt-image-2',
|
||||
`size` VARCHAR(32) NOT NULL DEFAULT '1024x1024',
|
||||
`started_at` DATETIME(3) NOT NULL DEFAULT CURRENT_TIMESTAMP(3),
|
||||
`ended_at` DATETIME(3) NULL,
|
||||
`duration_ms` INT NULL,
|
||||
`image_url` VARCHAR(2048) NULL,
|
||||
`image_base64` LONGTEXT NULL,
|
||||
`image_mime_type` VARCHAR(191) NULL,
|
||||
`revised_prompt` TEXT NULL,
|
||||
`upstream_response` JSON NULL,
|
||||
`error_message` TEXT NULL,
|
||||
`created_at` DATETIME(3) NOT NULL DEFAULT CURRENT_TIMESTAMP(3),
|
||||
`updated_at` DATETIME(3) NOT NULL DEFAULT CURRENT_TIMESTAMP(3) ON UPDATE CURRENT_TIMESTAMP(3),
|
||||
`deleted_at` DATETIME(3) NULL,
|
||||
PRIMARY KEY (`id`)
|
||||
) DEFAULT CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci;
|
||||
|
||||
-- CreateTable
|
||||
CREATE TABLE `generation_stats` (
|
||||
`id` VARCHAR(32) NOT NULL,
|
||||
`total_requests` INT NOT NULL DEFAULT 0,
|
||||
`success_requests` INT NOT NULL DEFAULT 0,
|
||||
`failed_requests` INT NOT NULL DEFAULT 0,
|
||||
`queued_requests` INT NOT NULL DEFAULT 0,
|
||||
`running_requests` INT NOT NULL DEFAULT 0,
|
||||
`total_images` INT NOT NULL DEFAULT 0,
|
||||
`created_at` DATETIME(3) NOT NULL DEFAULT CURRENT_TIMESTAMP(3),
|
||||
`updated_at` DATETIME(3) NOT NULL DEFAULT CURRENT_TIMESTAMP(3) ON UPDATE CURRENT_TIMESTAMP(3),
|
||||
PRIMARY KEY (`id`)
|
||||
) DEFAULT CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci;
|
||||
|
||||
-- CreateIndex
|
||||
CREATE INDEX `image_generations_user_id_created_at_idx` ON `image_generations`(`user_id`, `created_at`);
|
||||
|
||||
-- CreateIndex
|
||||
CREATE INDEX `image_generations_status_idx` ON `image_generations`(`status`);
|
||||
|
||||
-- CreateIndex
|
||||
CREATE INDEX `image_generations_deleted_at_idx` ON `image_generations`(`deleted_at`);
|
||||
|
||||
-- AddForeignKey
|
||||
ALTER TABLE `image_generations`
|
||||
ADD CONSTRAINT `image_generations_user_id_fkey`
|
||||
FOREIGN KEY (`user_id`) REFERENCES `users`(`id`)
|
||||
ON DELETE CASCADE ON UPDATE CASCADE;
|
||||
@@ -0,0 +1,3 @@
|
||||
# Please do not edit this file manually
|
||||
# It should be added in your version-control system (e.g., Git)
|
||||
provider = "mysql"
|
||||
@@ -0,0 +1,75 @@
|
||||
// This is your Prisma schema file,
|
||||
// learn more about it in the docs: https://pris.ly/d/prisma-schema
|
||||
|
||||
// Get a free hosted Postgres database in seconds: `npx create-db`
|
||||
|
||||
generator client {
|
||||
provider = "prisma-client"
|
||||
output = "../app/generated/prisma"
|
||||
}
|
||||
|
||||
datasource db {
|
||||
provider = "mysql"
|
||||
}
|
||||
|
||||
enum ImageGenerationStatus {
|
||||
QUEUED
|
||||
RUNNING
|
||||
SUCCEEDED
|
||||
FAILED
|
||||
}
|
||||
|
||||
model User {
|
||||
id Int @id
|
||||
username String @default("") @db.VarChar(191)
|
||||
displayName String @default("") @map("display_name") @db.VarChar(191)
|
||||
group String @default("") @db.VarChar(191)
|
||||
role Int @default(0)
|
||||
status Int @default(0)
|
||||
createdAt DateTime @default(now()) @map("created_at")
|
||||
updatedAt DateTime @updatedAt @map("updated_at")
|
||||
imageGenerations ImageGeneration[]
|
||||
|
||||
@@map("users")
|
||||
}
|
||||
|
||||
model ImageGeneration {
|
||||
id BigInt @id @default(autoincrement()) @db.UnsignedBigInt
|
||||
userId Int @map("user_id")
|
||||
prompt String @db.Text
|
||||
status ImageGenerationStatus @default(RUNNING)
|
||||
model String @default("gpt-image-2") @db.VarChar(64)
|
||||
size String @default("1024x1024") @db.VarChar(32)
|
||||
startedAt DateTime @default(now()) @map("started_at")
|
||||
endedAt DateTime? @map("ended_at")
|
||||
durationMs Int? @map("duration_ms")
|
||||
imageUrl String? @map("image_url") @db.VarChar(2048)
|
||||
imageBase64 String? @map("image_base64") @db.LongText
|
||||
imageMimeType String? @map("image_mime_type") @db.VarChar(191)
|
||||
revisedPrompt String? @map("revised_prompt") @db.Text
|
||||
upstreamResponse Json? @map("upstream_response")
|
||||
errorMessage String? @map("error_message") @db.Text
|
||||
createdAt DateTime @default(now()) @map("created_at")
|
||||
updatedAt DateTime @updatedAt @map("updated_at")
|
||||
deletedAt DateTime? @map("deleted_at")
|
||||
user User @relation(fields: [userId], references: [id], onDelete: Cascade)
|
||||
|
||||
@@index([userId, createdAt])
|
||||
@@index([status])
|
||||
@@index([deletedAt])
|
||||
@@map("image_generations")
|
||||
}
|
||||
|
||||
model GenerationStats {
|
||||
id String @id @db.VarChar(32)
|
||||
totalRequests Int @default(0) @map("total_requests")
|
||||
successRequests Int @default(0) @map("success_requests")
|
||||
failedRequests Int @default(0) @map("failed_requests")
|
||||
queuedRequests Int @default(0) @map("queued_requests")
|
||||
runningRequests Int @default(0) @map("running_requests")
|
||||
totalImages Int @default(0) @map("total_images")
|
||||
createdAt DateTime @default(now()) @map("created_at")
|
||||
updatedAt DateTime @updatedAt @map("updated_at")
|
||||
|
||||
@@map("generation_stats")
|
||||
}
|
||||
@@ -9,7 +9,8 @@ import {
|
||||
extractCookieMetaFromSetCookie,
|
||||
extractCookieValueFromSetCookie,
|
||||
newApiFetchRaw,
|
||||
setNewApiAuthCookies
|
||||
setNewApiAuthCookies,
|
||||
upsertUserSnapshot
|
||||
} from "~~/server/utils";
|
||||
|
||||
interface INewApiWrappedLoginResponse {
|
||||
@@ -92,6 +93,10 @@ export default defineEventHandler(async (event) => {
|
||||
extractCookieMetaFromSetCookie(result.headers, "session")
|
||||
);
|
||||
|
||||
upsertUserSnapshot(normalized.user).catch((error) => {
|
||||
consola.error("/api/auth/login 保存用户快照失败", error);
|
||||
});
|
||||
|
||||
return createSuccessResponse<IUserLoginData>(
|
||||
normalized.user,
|
||||
normalized.message
|
||||
|
||||
@@ -3,7 +3,8 @@ import {
|
||||
clearNewApiAuthCookies,
|
||||
createErrorResponse,
|
||||
createSuccessResponse,
|
||||
newApiAuthedFetch
|
||||
newApiAuthedFetch,
|
||||
upsertUserSnapshot
|
||||
} from "~~/server/utils";
|
||||
|
||||
interface INewApiWrappedUserResponse {
|
||||
@@ -40,6 +41,8 @@ export default defineEventHandler(async (event) => {
|
||||
return createErrorResponse(401, normalized.message);
|
||||
}
|
||||
|
||||
upsertUserSnapshot(normalized.user).catch(() => {});
|
||||
|
||||
return createSuccessResponse<IUserMeData>(
|
||||
normalized.user,
|
||||
normalized.message
|
||||
|
||||
@@ -0,0 +1,22 @@
|
||||
import { consola } from "consola";
|
||||
import { prisma } from "~~/server/utils";
|
||||
|
||||
export default defineEventHandler(async () => {
|
||||
const startedAt = Date.now();
|
||||
|
||||
try {
|
||||
await prisma.$queryRaw`SELECT 1`;
|
||||
|
||||
return createSuccessResponse(
|
||||
{
|
||||
ok: true,
|
||||
elapsedMs: Date.now() - startedAt
|
||||
},
|
||||
"数据库连接正常"
|
||||
);
|
||||
} catch (error) {
|
||||
consola.error("[images.db-health] 数据库连接失败", error);
|
||||
|
||||
return createUpstreamErrorResponse(error, "数据库连接失败");
|
||||
}
|
||||
});
|
||||
@@ -1,12 +1,21 @@
|
||||
import { randomUUID } from "node:crypto";
|
||||
import { consola } from "consola";
|
||||
|
||||
import type {
|
||||
IImageGenerateData,
|
||||
IImageGenerateRequest
|
||||
} from "#shared/types/openai";
|
||||
import { isUnauthorizedError } from "~~/server/utils";
|
||||
import {
|
||||
createRunningImageGeneration,
|
||||
downloadImageAsBase64,
|
||||
finishImageGenerationFailed,
|
||||
finishImageGenerationSuccess,
|
||||
getNewApiUserIdFromCookie,
|
||||
isUnauthorizedError
|
||||
} from "~~/server/utils";
|
||||
|
||||
export default defineEventHandler(async (event) => {
|
||||
const requestId = randomUUID();
|
||||
let stage = "read_body";
|
||||
const requestBody = await readBody<Partial<IImageGenerateRequest> | null>(
|
||||
event
|
||||
);
|
||||
@@ -24,16 +33,96 @@ export default defineEventHandler(async (event) => {
|
||||
return createErrorResponse(400, "请输入图片描述");
|
||||
}
|
||||
|
||||
let record: { id: bigint; startedAt: Date } | null = null;
|
||||
|
||||
try {
|
||||
consola.info("[images.generate] 开始", {
|
||||
requestId,
|
||||
promptLength: prompt.length
|
||||
});
|
||||
|
||||
stage = "read_user_id";
|
||||
const userId = getNewApiUserIdFromCookie(event);
|
||||
consola.info("[images.generate] 读取用户成功", {
|
||||
requestId,
|
||||
userId
|
||||
});
|
||||
|
||||
stage = "create_running_record";
|
||||
record = await createRunningImageGeneration(userId, prompt);
|
||||
consola.info("[images.generate] 创建生图记录成功", {
|
||||
requestId,
|
||||
recordId: record.id.toString()
|
||||
});
|
||||
|
||||
stage = "get_api_key";
|
||||
const apiKey = await getAiArtStudioTokenKey(event);
|
||||
consola.info("[images.generate] 获取服务端 key 成功", {
|
||||
requestId,
|
||||
recordId: record.id.toString()
|
||||
});
|
||||
|
||||
stage = "call_image_api";
|
||||
const result = await askImg({
|
||||
apiKey,
|
||||
prompt
|
||||
});
|
||||
consola.info("[images.generate] 上游生图成功", {
|
||||
requestId,
|
||||
recordId: record.id.toString(),
|
||||
hasImageUrl: Boolean(result.imageUrl)
|
||||
});
|
||||
|
||||
consola.info("图片生成成功", { prompt, imageUrl: result.imageUrl });
|
||||
return createSuccessResponse<IImageGenerateData>(result, "图片生成成功");
|
||||
stage = "download_image_base64";
|
||||
const downloadedImage = await downloadGeneratedImage(result.imageUrl);
|
||||
consola.info("[images.generate] 图片归档下载完成", {
|
||||
requestId,
|
||||
recordId: record.id.toString(),
|
||||
hasBase64: Boolean(downloadedImage.base64),
|
||||
mimeType: downloadedImage.mimeType
|
||||
});
|
||||
|
||||
stage = "finish_success_record";
|
||||
await finishImageGenerationSuccess(record.id, record.startedAt, {
|
||||
imageUrl: result.imageUrl,
|
||||
imageBase64: downloadedImage.base64,
|
||||
imageMimeType: downloadedImage.mimeType,
|
||||
revisedPrompt: result.revisedPrompt,
|
||||
upstreamResponse: result.upstreamResponse,
|
||||
errorMessage: downloadedImage.errorMessage
|
||||
});
|
||||
|
||||
consola.info("[images.generate] 完成", {
|
||||
requestId,
|
||||
recordId: record.id.toString()
|
||||
});
|
||||
return createSuccessResponse<IImageGenerateData>(
|
||||
{
|
||||
imageUrl: result.imageUrl,
|
||||
revisedPrompt: result.revisedPrompt
|
||||
},
|
||||
"图片生成成功"
|
||||
);
|
||||
} catch (error) {
|
||||
consola.error("[images.generate] 失败", {
|
||||
requestId,
|
||||
stage,
|
||||
recordId: record?.id.toString() ?? null,
|
||||
error: toSafeLogError(error)
|
||||
});
|
||||
|
||||
if (record) {
|
||||
await finishImageGenerationFailed(record.id, record.startedAt, error).catch(
|
||||
(recordError) => {
|
||||
consola.error("[images.generate] 更新失败记录失败", {
|
||||
requestId,
|
||||
recordId: record?.id.toString() ?? null,
|
||||
error: toSafeLogError(recordError)
|
||||
});
|
||||
}
|
||||
);
|
||||
}
|
||||
|
||||
if (isUnauthorizedError(error)) {
|
||||
clearNewApiAuthCookies(event);
|
||||
return createErrorResponse(401, "未登录");
|
||||
@@ -42,3 +131,53 @@ export default defineEventHandler(async (event) => {
|
||||
return createUpstreamErrorResponse(error, "图片生成失败");
|
||||
}
|
||||
});
|
||||
|
||||
/** 图片 URL 仍然可用时，base64 保存失败不阻断本次生成结果 */
|
||||
const downloadGeneratedImage = async (
|
||||
imageUrl: string
|
||||
): Promise<{
|
||||
base64: string | null;
|
||||
mimeType: string | null;
|
||||
errorMessage: string | null;
|
||||
}> => {
|
||||
try {
|
||||
const image = await downloadImageAsBase64(imageUrl);
|
||||
return {
|
||||
base64: image.base64,
|
||||
mimeType: image.mimeType,
|
||||
errorMessage: null
|
||||
};
|
||||
} catch (error) {
|
||||
consola.error("[images.generate] 保存生成图片 base64 失败", {
|
||||
error: toSafeLogError(error)
|
||||
});
|
||||
return {
|
||||
base64: null,
|
||||
mimeType: null,
|
||||
errorMessage: "图片生成成功，但 base64 保存失败"
|
||||
};
|
||||
}
|
||||
};
|
||||
|
||||
const toSafeLogError = (error: unknown) => {
|
||||
const maybeError = error as {
|
||||
data?: unknown;
|
||||
message?: string;
|
||||
name?: string;
|
||||
response?: {
|
||||
_data?: unknown;
|
||||
status?: number;
|
||||
};
|
||||
stack?: string;
|
||||
status?: number;
|
||||
statusCode?: number;
|
||||
};
|
||||
|
||||
return {
|
||||
name: maybeError.name,
|
||||
message: maybeError.message,
|
||||
status: maybeError.status ?? maybeError.statusCode ?? maybeError.response?.status,
|
||||
data: maybeError.data ?? maybeError.response?._data ?? null,
|
||||
stack: maybeError.stack
|
||||
};
|
||||
};
|
||||
|
||||
@@ -0,0 +1,52 @@
|
||||
import type { IImageHistoryListData } from "#shared/types/openai";
|
||||
import {
|
||||
clearNewApiAuthCookies,
|
||||
getNewApiUserIdFromCookie,
|
||||
isUnauthorizedError,
|
||||
listImageGenerationHistory
|
||||
} from "~~/server/utils";
|
||||
|
||||
const DEFAULT_PAGE = 1;
|
||||
const DEFAULT_PAGE_SIZE = 10;
|
||||
const MAX_PAGE_SIZE = 50;
|
||||
|
||||
export default defineEventHandler(async (event) => {
|
||||
try {
|
||||
const userId = getNewApiUserIdFromCookie(event);
|
||||
const query = getQuery(event);
|
||||
const page = normalizePositiveInt(query.page, DEFAULT_PAGE);
|
||||
const pageSize = Math.min(
|
||||
normalizePositiveInt(query.size, DEFAULT_PAGE_SIZE),
|
||||
MAX_PAGE_SIZE
|
||||
);
|
||||
const result = await listImageGenerationHistory({
|
||||
userId,
|
||||
page,
|
||||
pageSize
|
||||
});
|
||||
|
||||
return createSuccessResponse<IImageHistoryListData>(
|
||||
result,
|
||||
"获取生图历史成功"
|
||||
);
|
||||
} catch (error) {
|
||||
if (isUnauthorizedError(error)) {
|
||||
clearNewApiAuthCookies(event);
|
||||
return createErrorResponse(401, "未登录");
|
||||
}
|
||||
|
||||
return createUpstreamErrorResponse(error, "获取生图历史失败");
|
||||
}
|
||||
});
|
||||
|
||||
const normalizePositiveInt = (
|
||||
value: unknown,
|
||||
fallbackValue: number
|
||||
): number => {
|
||||
const rawValue = Array.isArray(value) ? value[0] : value;
|
||||
const numberValue = Number.parseInt(String(rawValue ?? ""), 10);
|
||||
|
||||
return Number.isInteger(numberValue) && numberValue > 0
|
||||
? numberValue
|
||||
: fallbackValue;
|
||||
};
|
||||
@@ -0,0 +1,38 @@
|
||||
import {
|
||||
clearNewApiAuthCookies,
|
||||
getNewApiUserIdFromCookie,
|
||||
isUnauthorizedError,
|
||||
softDeleteImageGeneration
|
||||
} from "~~/server/utils";
|
||||
|
||||
export default defineEventHandler(async (event) => {
|
||||
try {
|
||||
const userId = getNewApiUserIdFromCookie(event);
|
||||
const recordId = parseRecordId(getRouterParam(event, "id"));
|
||||
const deleted = await softDeleteImageGeneration(userId, recordId);
|
||||
|
||||
if (!deleted) {
|
||||
return createErrorResponse(404, "生图记录不存在");
|
||||
}
|
||||
|
||||
return createSuccessResponse(null, "删除生图记录成功");
|
||||
} catch (error) {
|
||||
if (isUnauthorizedError(error)) {
|
||||
clearNewApiAuthCookies(event);
|
||||
return createErrorResponse(401, "未登录");
|
||||
}
|
||||
|
||||
return createUpstreamErrorResponse(error, "删除生图记录失败");
|
||||
}
|
||||
});
|
||||
|
||||
const parseRecordId = (value: string | undefined): bigint => {
|
||||
if (!value || !/^\d+$/.test(value)) {
|
||||
throw createError({
|
||||
statusCode: 400,
|
||||
statusMessage: "生图记录 ID 不正确"
|
||||
});
|
||||
}
|
||||
|
||||
return BigInt(value);
|
||||
};
|
||||
@@ -0,0 +1,42 @@
|
||||
import type { IImageHistoryDetail } from "#shared/types/openai";
|
||||
import {
|
||||
clearNewApiAuthCookies,
|
||||
getImageGenerationDetail,
|
||||
getNewApiUserIdFromCookie,
|
||||
isUnauthorizedError
|
||||
} from "~~/server/utils";
|
||||
|
||||
export default defineEventHandler(async (event) => {
|
||||
try {
|
||||
const userId = getNewApiUserIdFromCookie(event);
|
||||
const recordId = parseRecordId(getRouterParam(event, "id"));
|
||||
const result = await getImageGenerationDetail(userId, recordId);
|
||||
|
||||
if (!result) {
|
||||
return createErrorResponse(404, "生图记录不存在");
|
||||
}
|
||||
|
||||
return createSuccessResponse<IImageHistoryDetail>(
|
||||
result,
|
||||
"获取生图详情成功"
|
||||
);
|
||||
} catch (error) {
|
||||
if (isUnauthorizedError(error)) {
|
||||
clearNewApiAuthCookies(event);
|
||||
return createErrorResponse(401, "未登录");
|
||||
}
|
||||
|
||||
return createUpstreamErrorResponse(error, "获取生图详情失败");
|
||||
}
|
||||
});
|
||||
|
||||
const parseRecordId = (value: string | undefined): bigint => {
|
||||
if (!value || !/^\d+$/.test(value)) {
|
||||
throw createError({
|
||||
statusCode: 400,
|
||||
statusMessage: "生图记录 ID 不正确"
|
||||
});
|
||||
}
|
||||
|
||||
return BigInt(value);
|
||||
};
|
||||
@@ -0,0 +1,15 @@
|
||||
import type { IImageGenerationStatsData } from "#shared/types/openai";
|
||||
import { getGenerationStats } from "~~/server/utils";
|
||||
|
||||
export default defineEventHandler(async () => {
|
||||
try {
|
||||
const result = await getGenerationStats();
|
||||
|
||||
return createSuccessResponse<IImageGenerationStatsData>(
|
||||
result,
|
||||
"获取生图统计成功"
|
||||
);
|
||||
} catch (error) {
|
||||
return createUpstreamErrorResponse(error, "获取生图统计失败");
|
||||
}
|
||||
});
|
||||
@@ -0,0 +1,23 @@
|
||||
export interface IDownloadedImage {
|
||||
/** 图片 base64 原文，不包含 data URL 前缀 */
|
||||
base64: string;
|
||||
/** 图片 MIME 类型 */
|
||||
mimeType: string;
|
||||
}
|
||||
|
||||
/** 下载远程图片并转成 base64，用于生图结果归档 */
|
||||
export const downloadImageAsBase64 = async (
|
||||
imageUrl: string
|
||||
): Promise<IDownloadedImage> => {
|
||||
const response = await fetch(imageUrl);
|
||||
if (!response.ok) {
|
||||
throw new Error(`图片下载失败：${response.status}`);
|
||||
}
|
||||
|
||||
const arrayBuffer = await response.arrayBuffer();
|
||||
|
||||
return {
|
||||
base64: Buffer.from(arrayBuffer).toString("base64"),
|
||||
mimeType: response.headers.get("content-type") || "image/png"
|
||||
};
|
||||
};
|
||||
@@ -0,0 +1,384 @@
|
||||
import type { IUserBasicData } from "#shared/types";
|
||||
import type {
|
||||
IImageGenerationStatsData,
|
||||
IImageHistoryDetail,
|
||||
IImageHistoryItem,
|
||||
IImageHistoryListData
|
||||
} from "#shared/types/openai";
|
||||
import { consola } from "consola";
|
||||
import { ImageGenerationStatus, Prisma } from "~~/app/generated/prisma/client";
|
||||
import { prisma } from "~~/server/utils/prisma";
|
||||
|
||||
const GLOBAL_STATS_ID = "global";
|
||||
const DEFAULT_IMAGE_MODEL = "gpt-image-2";
|
||||
const DEFAULT_IMAGE_SIZE = "1024x1024";
|
||||
|
||||
interface IFinishImageGenerationSuccessInput {
|
||||
/** 生成图片 URL */
|
||||
imageUrl: string;
|
||||
/** 生成图片 base64 原文 */
|
||||
imageBase64: string | null;
|
||||
/** 图片 MIME 类型 */
|
||||
imageMimeType: string | null;
|
||||
/** 上游返回的修订提示词 */
|
||||
revisedPrompt?: string | null;
|
||||
/** 完整上游响应 */
|
||||
upstreamResponse: unknown;
|
||||
/** 成功状态下的非阻断提示，例如 base64 保存失败 */
|
||||
errorMessage?: string | null;
|
||||
}
|
||||
|
||||
/** 登录或恢复登录时保存 NewAPI 用户快照 */
|
||||
export const upsertUserSnapshot = (user: IUserBasicData) => {
|
||||
return prisma.user.upsert({
|
||||
where: {
|
||||
id: user.id
|
||||
},
|
||||
create: {
|
||||
id: user.id,
|
||||
username: user.username,
|
||||
displayName: user.display_name,
|
||||
group: user.group,
|
||||
role: user.role,
|
||||
status: user.status
|
||||
},
|
||||
update: {
|
||||
username: user.username,
|
||||
displayName: user.display_name,
|
||||
group: user.group,
|
||||
role: user.role,
|
||||
status: user.status
|
||||
}
|
||||
});
|
||||
};
|
||||
|
||||
/** 仅有 userId 时兜底创建用户记录，避免旧 cookie 用户生成时外键失败 */
|
||||
export const ensureUserRecord = (userId: number) => {
|
||||
return prisma.user.upsert({
|
||||
where: {
|
||||
id: userId
|
||||
},
|
||||
create: {
|
||||
id: userId
|
||||
},
|
||||
update: {}
|
||||
});
|
||||
};
|
||||
|
||||
/** 创建进行中的生图记录，并递增全局请求与进行中统计 */
|
||||
export const createRunningImageGeneration = async (
|
||||
userId: number,
|
||||
prompt: string
|
||||
) => {
|
||||
await ensureUserRecord(userId);
|
||||
|
||||
const record = await prisma.imageGeneration.create({
|
||||
data: {
|
||||
userId,
|
||||
prompt,
|
||||
status: ImageGenerationStatus.RUNNING,
|
||||
model: DEFAULT_IMAGE_MODEL,
|
||||
size: DEFAULT_IMAGE_SIZE
|
||||
}
|
||||
});
|
||||
|
||||
await safeUpsertGenerationStats({
|
||||
totalRequests: {
|
||||
increment: 1
|
||||
},
|
||||
runningRequests: {
|
||||
increment: 1
|
||||
}
|
||||
});
|
||||
|
||||
return record;
|
||||
};
|
||||
|
||||
/** 将生图记录标记为成功，并保存 URL、base64、完整上游响应和耗时 */
|
||||
export const finishImageGenerationSuccess = async (
|
||||
recordId: bigint,
|
||||
startedAt: Date,
|
||||
input: IFinishImageGenerationSuccessInput
|
||||
) => {
|
||||
const endedAt = new Date();
|
||||
const successData = {
|
||||
status: ImageGenerationStatus.SUCCEEDED,
|
||||
endedAt,
|
||||
durationMs: getDurationMs(startedAt, endedAt),
|
||||
imageUrl: input.imageUrl,
|
||||
imageBase64: input.imageBase64,
|
||||
imageMimeType: input.imageMimeType,
|
||||
revisedPrompt: input.revisedPrompt || null,
|
||||
upstreamResponse: input.upstreamResponse as Prisma.InputJsonValue,
|
||||
errorMessage: input.errorMessage || null
|
||||
};
|
||||
|
||||
try {
|
||||
await prisma.imageGeneration.update({
|
||||
where: {
|
||||
id: recordId
|
||||
},
|
||||
data: successData
|
||||
});
|
||||
} catch (error) {
|
||||
if (!input.imageBase64) {
|
||||
throw error;
|
||||
}
|
||||
|
||||
consola.error("[imageGenerationRecords] 保存带 base64 的成功记录失败，重试仅保存 URL", {
|
||||
recordId: recordId.toString(),
|
||||
message: error instanceof Error ? error.message : String(error)
|
||||
});
|
||||
|
||||
await prisma.imageGeneration.update({
|
||||
where: {
|
||||
id: recordId
|
||||
},
|
||||
data: {
|
||||
...successData,
|
||||
imageBase64: null,
|
||||
errorMessage: mergeRecordMessages(
|
||||
input.errorMessage,
|
||||
"图片生成成功，但 base64 归档保存失败"
|
||||
)
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
await safeUpsertGenerationStats({
|
||||
successRequests: {
|
||||
increment: 1
|
||||
},
|
||||
runningRequests: {
|
||||
decrement: 1
|
||||
},
|
||||
totalImages: {
|
||||
increment: 1
|
||||
}
|
||||
});
|
||||
};
|
||||
|
||||
/** 将生图记录标记为失败，并保存失败原因和耗时 */
|
||||
export const finishImageGenerationFailed = async (
|
||||
recordId: bigint,
|
||||
startedAt: Date,
|
||||
error: unknown
|
||||
) => {
|
||||
const endedAt = new Date();
|
||||
|
||||
await prisma.imageGeneration.update({
|
||||
where: {
|
||||
id: recordId
|
||||
},
|
||||
data: {
|
||||
status: ImageGenerationStatus.FAILED,
|
||||
endedAt,
|
||||
durationMs: getDurationMs(startedAt, endedAt),
|
||||
errorMessage: getSafeErrorMessage(error)
|
||||
}
|
||||
});
|
||||
|
||||
await safeUpsertGenerationStats({
|
||||
failedRequests: {
|
||||
increment: 1
|
||||
},
|
||||
runningRequests: {
|
||||
decrement: 1
|
||||
}
|
||||
});
|
||||
};
|
||||
|
||||
/** 查询当前用户未删除的生图历史列表，不返回 base64 和完整上游响应 */
|
||||
export const listImageGenerationHistory = async ({
|
||||
userId,
|
||||
page,
|
||||
pageSize
|
||||
}: {
|
||||
userId: number;
|
||||
page: number;
|
||||
pageSize: number;
|
||||
}): Promise<IImageHistoryListData> => {
|
||||
const where = {
|
||||
userId,
|
||||
deletedAt: null
|
||||
};
|
||||
|
||||
const [total, records] = await prisma.$transaction([
|
||||
prisma.imageGeneration.count({ where }),
|
||||
prisma.imageGeneration.findMany({
|
||||
where,
|
||||
orderBy: {
|
||||
createdAt: "desc"
|
||||
},
|
||||
skip: (page - 1) * pageSize,
|
||||
take: pageSize
|
||||
})
|
||||
]);
|
||||
|
||||
return {
|
||||
page,
|
||||
pageSize,
|
||||
total,
|
||||
items: records.map(mapImageGenerationItem)
|
||||
};
|
||||
};
|
||||
|
||||
/** 查询当前用户单条生图历史详情 */
|
||||
export const getImageGenerationDetail = async (
|
||||
userId: number,
|
||||
recordId: bigint
|
||||
): Promise<IImageHistoryDetail | null> => {
|
||||
const record = await prisma.imageGeneration.findFirst({
|
||||
where: {
|
||||
id: recordId,
|
||||
userId,
|
||||
deletedAt: null
|
||||
}
|
||||
});
|
||||
|
||||
if (!record) return null;
|
||||
|
||||
return {
|
||||
...mapImageGenerationItem(record),
|
||||
imageBase64: record.imageBase64,
|
||||
imageMimeType: record.imageMimeType,
|
||||
upstreamResponse: record.upstreamResponse
|
||||
};
|
||||
};
|
||||
|
||||
/** 软删除当前用户单条生图历史 */
|
||||
export const softDeleteImageGeneration = async (
|
||||
userId: number,
|
||||
recordId: bigint
|
||||
): Promise<boolean> => {
|
||||
const result = await prisma.imageGeneration.updateMany({
|
||||
where: {
|
||||
id: recordId,
|
||||
userId,
|
||||
deletedAt: null
|
||||
},
|
||||
data: {
|
||||
deletedAt: new Date()
|
||||
}
|
||||
});
|
||||
|
||||
return result.count > 0;
|
||||
};
|
||||
|
||||
/** 读取全局生图统计，没有记录时返回全 0 */
|
||||
export const getGenerationStats =
|
||||
async (): Promise<IImageGenerationStatsData> => {
|
||||
const stats = await prisma.generationStats.findUnique({
|
||||
where: {
|
||||
id: GLOBAL_STATS_ID
|
||||
}
|
||||
});
|
||||
|
||||
return {
|
||||
totalRequests: stats?.totalRequests ?? 0,
|
||||
successRequests: stats?.successRequests ?? 0,
|
||||
failedRequests: stats?.failedRequests ?? 0,
|
||||
queuedRequests: stats?.queuedRequests ?? 0,
|
||||
runningRequests: stats?.runningRequests ?? 0,
|
||||
totalImages: stats?.totalImages ?? 0
|
||||
};
|
||||
};
|
||||
|
||||
/** 统一创建或更新全局统计行 */
|
||||
const upsertGenerationStats = (update: Prisma.GenerationStatsUpdateInput) => {
|
||||
return prisma.generationStats.upsert({
|
||||
where: {
|
||||
id: GLOBAL_STATS_ID
|
||||
},
|
||||
create: {
|
||||
id: GLOBAL_STATS_ID,
|
||||
...buildStatsCreateInput(update)
|
||||
},
|
||||
update
|
||||
});
|
||||
};
|
||||
|
||||
const safeUpsertGenerationStats = async (
|
||||
update: Prisma.GenerationStatsUpdateInput
|
||||
) => {
|
||||
try {
|
||||
await upsertGenerationStats(update);
|
||||
} catch (error) {
|
||||
consola.error("[imageGenerationRecords] 更新生图统计失败", {
|
||||
message: error instanceof Error ? error.message : String(error)
|
||||
});
|
||||
}
|
||||
};
|
||||
|
||||
const buildStatsCreateInput = (update: Prisma.GenerationStatsUpdateInput) => {
|
||||
return {
|
||||
totalRequests: getIncrementValue(update.totalRequests),
|
||||
successRequests: getIncrementValue(update.successRequests),
|
||||
failedRequests: getIncrementValue(update.failedRequests),
|
||||
queuedRequests: getIncrementValue(update.queuedRequests),
|
||||
runningRequests: getIncrementValue(update.runningRequests),
|
||||
totalImages: getIncrementValue(update.totalImages)
|
||||
};
|
||||
};
|
||||
|
||||
const getIncrementValue = (value: unknown): number => {
|
||||
if (
|
||||
value &&
|
||||
typeof value === "object" &&
|
||||
"increment" in value &&
|
||||
typeof value.increment === "number"
|
||||
) {
|
||||
return value.increment;
|
||||
}
|
||||
|
||||
return 0;
|
||||
};
|
||||
|
||||
const mapImageGenerationItem = (record: {
|
||||
id: bigint;
|
||||
userId: number;
|
||||
prompt: string;
|
||||
status: ImageGenerationStatus;
|
||||
model: string;
|
||||
size: string;
|
||||
startedAt: Date;
|
||||
endedAt: Date | null;
|
||||
durationMs: number | null;
|
||||
imageUrl: string | null;
|
||||
revisedPrompt: string | null;
|
||||
errorMessage: string | null;
|
||||
createdAt: Date;
|
||||
}): IImageHistoryItem => {
|
||||
return {
|
||||
id: record.id.toString(),
|
||||
userId: record.userId,
|
||||
prompt: record.prompt,
|
||||
status: record.status,
|
||||
model: record.model,
|
||||
size: record.size,
|
||||
startedAt: record.startedAt.toISOString(),
|
||||
endedAt: record.endedAt?.toISOString() ?? null,
|
||||
durationMs: record.durationMs,
|
||||
imageUrl: record.imageUrl,
|
||||
revisedPrompt: record.revisedPrompt,
|
||||
errorMessage: record.errorMessage,
|
||||
createdAt: record.createdAt.toISOString()
|
||||
};
|
||||
};
|
||||
|
||||
const getDurationMs = (startedAt: Date, endedAt: Date) => {
|
||||
return Math.max(0, endedAt.getTime() - startedAt.getTime());
|
||||
};
|
||||
|
||||
const mergeRecordMessages = (
|
||||
currentMessage: string | null | undefined,
|
||||
nextMessage: string
|
||||
) => {
|
||||
return currentMessage ? `${currentMessage}; ${nextMessage}` : nextMessage;
|
||||
};
|
||||
|
||||
const getSafeErrorMessage = (error: unknown): string => {
|
||||
if (error instanceof Error) return error.message;
|
||||
if (typeof error === "string") return error;
|
||||
return "图片生成失败";
|
||||
};
|
||||
@@ -1,5 +1,8 @@
|
||||
export * from "./createApiResponse";
|
||||
export * from "./fetch";
|
||||
export * from "./imageAssets";
|
||||
export * from "./imageGenerationRecords";
|
||||
export * from "./newApiAuthCookies";
|
||||
export * from "./newApiTokens";
|
||||
export * from "./openai";
|
||||
export * from "./prisma";
|
||||
|
||||
@@ -1,5 +1,5 @@
|
||||
import type { H3Event } from "h3";
|
||||
import { deleteCookie, setCookie } from "h3";
|
||||
import { createError, deleteCookie, getCookie, setCookie } from "h3";
|
||||
|
||||
/** 上游没有声明 session 过期时间时，本地登录态默认保留 3 天 */
|
||||
const DEFAULT_AUTH_COOKIE_MAX_AGE = 60 * 60 * 24 * 3;
|
||||
@@ -100,6 +100,23 @@ export const clearNewApiAuthCookies = (event: H3Event) => {
|
||||
deleteCookie(event, NEWAPI_USER_ID_COOKIE, options);
|
||||
};
|
||||
|
||||
/** 从本项目 httpOnly cookie 中读取 NewAPI 用户 ID，缺失或非法时统一视为未登录 */
|
||||
export const getNewApiUserIdFromCookie = (event: H3Event): number => {
|
||||
const userId = Number.parseInt(
|
||||
getCookie(event, NEWAPI_USER_ID_COOKIE) ?? "",
|
||||
10
|
||||
);
|
||||
|
||||
if (!Number.isInteger(userId) || userId <= 0) {
|
||||
throw createError({
|
||||
statusCode: 401,
|
||||
statusMessage: "未登录"
|
||||
});
|
||||
}
|
||||
|
||||
return userId;
|
||||
};
|
||||
|
||||
/** 统一生成本站认证 cookie 选项，保证安全属性在两个 cookie 上一致 */
|
||||
const buildAuthCookieOptions = (meta?: INewApiCookieMeta): CookieOptions => {
|
||||
const options: CookieOptions = {
|
||||
|
||||
+10
-2
@@ -16,6 +16,13 @@ interface IImageGenerationResponse {
|
||||
/** 生成图片地址 */
|
||||
url?: string;
|
||||
}>;
|
||||
/** 上游 token 用量 */
|
||||
usage?: unknown;
|
||||
}
|
||||
|
||||
export interface IAskImageResult extends IImageGenerateData {
|
||||
/** 完整上游生图接口返回结果，仅服务端内部保存 */
|
||||
upstreamResponse: IImageGenerationResponse;
|
||||
}
|
||||
|
||||
/** 通用 AI 调用函数（支持文本 / 图文 / 多模态） */
|
||||
@@ -107,7 +114,7 @@ export const askImg = async ({
|
||||
}: {
|
||||
apiKey: string;
|
||||
prompt: string;
|
||||
}): Promise<IImageGenerateData> => {
|
||||
}): Promise<IAskImageResult> => {
|
||||
const result = await $fetch<IImageGenerationResponse>(IMAGE_GENERATION_URL, {
|
||||
method: "POST",
|
||||
headers: {
|
||||
@@ -128,7 +135,8 @@ export const askImg = async ({
|
||||
|
||||
return {
|
||||
imageUrl: image.url,
|
||||
revisedPrompt: image.revised_prompt
|
||||
revisedPrompt: image.revised_prompt,
|
||||
upstreamResponse: result
|
||||
};
|
||||
};
|
||||
|
||||
|
||||
@@ -0,0 +1,76 @@
|
||||
import { PrismaMariaDb } from "@prisma/adapter-mariadb";
|
||||
import { PrismaClient } from "~~/app/generated/prisma/client";
|
||||
|
||||
const globalForPrisma = globalThis as unknown as {
|
||||
prisma?: PrismaClient;
|
||||
prismaLogged?: boolean;
|
||||
};
|
||||
|
||||
const databaseUrl = process.env.DATABASE_URL;
|
||||
|
||||
if (!databaseUrl) {
|
||||
throw new Error("DATABASE_URL is required to initialize PrismaClient");
|
||||
}
|
||||
|
||||
type MariaDbPoolConfig = Exclude<ConstructorParameters<typeof PrismaMariaDb>[0], string>;
|
||||
|
||||
function getNumberParam(url: URL, name: string, fallback: number) {
|
||||
const value = url.searchParams.get(name);
|
||||
if (!value) {
|
||||
return fallback;
|
||||
}
|
||||
|
||||
const parsed = Number(value);
|
||||
return Number.isFinite(parsed) && parsed > 0 ? parsed : fallback;
|
||||
}
|
||||
|
||||
function createMariaDbConfig(urlString: string): MariaDbPoolConfig {
|
||||
const url = new URL(urlString);
|
||||
const database = decodeURIComponent(url.pathname.replace(/^\//, ""));
|
||||
|
||||
if (!url.hostname || !url.username || !database) {
|
||||
throw new Error("DATABASE_URL must include host, user and database");
|
||||
}
|
||||
|
||||
const port = url.port ? Number(url.port) : 3306;
|
||||
|
||||
if (!Number.isInteger(port) || port <= 0) {
|
||||
throw new Error("DATABASE_URL port is invalid");
|
||||
}
|
||||
|
||||
return {
|
||||
host: url.hostname,
|
||||
port,
|
||||
user: decodeURIComponent(url.username),
|
||||
password: decodeURIComponent(url.password),
|
||||
database,
|
||||
connectionLimit: getNumberParam(url, "connection_limit", 5),
|
||||
acquireTimeout: getNumberParam(url, "pool_timeout", 30) * 1000,
|
||||
connectTimeout: getNumberParam(url, "connect_timeout", 10) * 1000,
|
||||
};
|
||||
}
|
||||
|
||||
const mariaDbConfig = createMariaDbConfig(databaseUrl);
|
||||
const adapter = new PrismaMariaDb(mariaDbConfig, { database: mariaDbConfig.database });
|
||||
|
||||
if (!globalForPrisma.prismaLogged) {
|
||||
console.info("[prisma] initializing MariaDB pool", {
|
||||
host: mariaDbConfig.host,
|
||||
port: mariaDbConfig.port,
|
||||
database: mariaDbConfig.database,
|
||||
connectionLimit: mariaDbConfig.connectionLimit,
|
||||
});
|
||||
globalForPrisma.prismaLogged = true;
|
||||
}
|
||||
|
||||
/** Reuse PrismaClient during dev hot reloads so Nuxt does not create duplicate pools. */
|
||||
export const prisma =
|
||||
globalForPrisma.prisma ??
|
||||
new PrismaClient({
|
||||
adapter,
|
||||
log: ["warn", "error"],
|
||||
});
|
||||
|
||||
if (process.env.NODE_ENV !== "production") {
|
||||
globalForPrisma.prisma = prisma;
|
||||
}
|
||||
@@ -24,3 +24,88 @@ export interface IImageGenerateData {
|
||||
/** 上游返回的修订提示词，可能为空 */
|
||||
revisedPrompt?: string;
|
||||
}
|
||||
|
||||
/**
|
||||
* 生图任务状态。
|
||||
*/
|
||||
export type ImageGenerationStatus =
|
||||
| "QUEUED"
|
||||
| "RUNNING"
|
||||
| "SUCCEEDED"
|
||||
| "FAILED";
|
||||
|
||||
/**
|
||||
* 生图历史列表项。
|
||||
*/
|
||||
export interface IImageHistoryItem {
|
||||
/** 生图记录 ID，BigInt 会以字符串返回 */
|
||||
id: string;
|
||||
/** NewAPI 用户 ID */
|
||||
userId: number;
|
||||
/** 用户输入的生图提示词 */
|
||||
prompt: string;
|
||||
/** 生图状态 */
|
||||
status: ImageGenerationStatus;
|
||||
/** 生图模型 */
|
||||
model: string;
|
||||
/** 图片尺寸 */
|
||||
size: string;
|
||||
/** 生图开始时间，ISO 字符串 */
|
||||
startedAt: string;
|
||||
/** 生图结束时间，ISO 字符串，未结束时为 null */
|
||||
endedAt: string | null;
|
||||
/** 生图耗时，单位毫秒 */
|
||||
durationMs: number | null;
|
||||
/** 生成图片的访问地址 */
|
||||
imageUrl: string | null;
|
||||
/** 上游返回的修订提示词，可能为空 */
|
||||
revisedPrompt: string | null;
|
||||
/** 失败原因或图片保存提示 */
|
||||
errorMessage: string | null;
|
||||
/** 记录创建时间，ISO 字符串 */
|
||||
createdAt: string;
|
||||
}
|
||||
|
||||
/**
|
||||
* 生图历史详情。
|
||||
*/
|
||||
export interface IImageHistoryDetail extends IImageHistoryItem {
|
||||
/** 服务端保存的图片 base64 原文，可能为空 */
|
||||
imageBase64: string | null;
|
||||
/** 图片 MIME 类型，可能为空 */
|
||||
imageMimeType: string | null;
|
||||
/** 完整上游生图接口返回结果 */
|
||||
upstreamResponse: unknown;
|
||||
}
|
||||
|
||||
/**
|
||||
* 生图历史分页响应。
|
||||
*/
|
||||
export interface IImageHistoryListData {
|
||||
/** 当前页码 */
|
||||
page: number;
|
||||
/** 每页数量 */
|
||||
pageSize: number;
|
||||
/** 当前用户未删除记录总数 */
|
||||
total: number;
|
||||
/** 当前页历史记录 */
|
||||
items: IImageHistoryItem[];
|
||||
}
|
||||
|
||||
/**
|
||||
* 全局生图统计响应。
|
||||
*/
|
||||
export interface IImageGenerationStatsData {
|
||||
/** 总生图请求数，包含成功和失败 */
|
||||
totalRequests: number;
|
||||
/** 成功生图请求数 */
|
||||
successRequests: number;
|
||||
/** 失败生图请求数 */
|
||||
failedRequests: number;
|
||||
/** 排队中请求数，预留异步任务使用 */
|
||||
queuedRequests: number;
|
||||
/** 进行中请求数 */
|
||||
runningRequests: number;
|
||||
/** 成功生成图片总数 */
|
||||
totalImages: number;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user