feat: 添加图床归档任务支持,优化归档逻辑和错误处理
This commit is contained in:
@@ -1,15 +1,9 @@
|
||||
// server/api/images/generate.post.ts - 图片生成接口:创建记录、流式调用上游生图并在响应后异步归档图床。
|
||||
import { setImmediate } from "node:timers";
|
||||
// server/api/images/generate.post.ts - 图片生成接口:创建记录、流式调用上游生图并将图床归档交给数据库队列。
|
||||
import type {
|
||||
IImageGenerateData,
|
||||
IImageGenerateRequest
|
||||
} from "#shared/types/openai";
|
||||
import {
|
||||
finishImageGenerationArchiveFailed,
|
||||
finishImageGenerationArchiveSuccess
|
||||
} from "~~/server/utils";
|
||||
|
||||
type ApiLogger = ReturnType<typeof createApiLogger>;
|
||||
import { wakeImageArchiveWorker } from "~~/server/utils";
|
||||
|
||||
/**
|
||||
* POST /api/images/generate
|
||||
@@ -20,8 +14,8 @@ type ApiLogger = ReturnType<typeof createApiLogger>;
|
||||
* 3. 创建 RUNNING 生图记录,并递增全局请求/进行中统计。
|
||||
* 4. 在服务端确保并读取 AIArtStudio 完整 key,完整 key 不返回前端。
|
||||
* 5. 调用 Chat Completions 流式生图接口,累积 SSE delta content 并提取最终图片 URL。
|
||||
* 6. 上游返回图片 URL 后立刻把生图结果落库为 SUCCEEDED,并马上返回前端。
|
||||
* 7. 响应返回后再用后台异步任务上传 Lsky;归档失败只补写记录和日志,不影响本次响应。
|
||||
* 6. 上游返回图片 URL 后立刻把生图结果落库为 SUCCEEDED,并把图床归档任务标记为待执行。
|
||||
* 7. 唤醒数据库归档队列 worker;归档失败只补写记录和日志,不影响本次响应。
|
||||
* 8. 主链路失败时把记录标记为 FAILED;鉴权失败会清理本地登录态并返回 401。
|
||||
*/
|
||||
export default defineEventHandler(async (event) => {
|
||||
@@ -102,18 +96,7 @@ export default defineEventHandler(async (event) => {
|
||||
archiveScheduled: true
|
||||
});
|
||||
|
||||
const finishedRecord = record;
|
||||
// TODO: 归档任务现在仍跑在当前 Node 进程里;如果并发继续升高,应该迁移到独立队列/worker,
|
||||
// 避免大量响应后任务在同一进程内叠加,拖慢其他请求的调度与内存回收。
|
||||
setImmediate(() => {
|
||||
void archiveGeneratedImageInBackground({
|
||||
imageUrl: result.imageUrl,
|
||||
userId,
|
||||
recordId: finishedRecord.id,
|
||||
createdAt: finishedRecord.startedAt,
|
||||
logger
|
||||
});
|
||||
});
|
||||
wakeImageArchiveWorker();
|
||||
|
||||
return createSuccessResponse<IImageGenerateData>(
|
||||
{
|
||||
@@ -152,60 +135,6 @@ export default defineEventHandler(async (event) => {
|
||||
}
|
||||
});
|
||||
|
||||
/** 响应返回后异步归档图片,成功则补写图床字段,失败则补写本地错误文案 */
|
||||
const archiveGeneratedImageInBackground = async ({
|
||||
imageUrl,
|
||||
userId,
|
||||
recordId,
|
||||
createdAt,
|
||||
logger
|
||||
}: {
|
||||
imageUrl: string;
|
||||
userId: number;
|
||||
recordId: bigint;
|
||||
createdAt: Date;
|
||||
logger: ApiLogger;
|
||||
}) => {
|
||||
logger.info("后台归档开始", {
|
||||
recordId: recordId.toString()
|
||||
});
|
||||
|
||||
try {
|
||||
const identity = await getUserArchiveIdentity(userId);
|
||||
const uploaded = await uploadImageFromUrl({
|
||||
imageUrl,
|
||||
userId,
|
||||
username: identity.username,
|
||||
recordId,
|
||||
createdAt
|
||||
});
|
||||
|
||||
await finishImageGenerationArchiveSuccess(recordId, {
|
||||
hostedImageUrl: uploaded.publicUrl,
|
||||
imageMimeType: uploaded.mimetype,
|
||||
hostedResponse: uploaded.response
|
||||
});
|
||||
|
||||
logger.info("后台归档成功", {
|
||||
recordId: recordId.toString(),
|
||||
mimeType: uploaded.mimetype,
|
||||
hosted: true
|
||||
});
|
||||
} catch (error) {
|
||||
logger.error("后台归档失败", {
|
||||
recordId: recordId.toString(),
|
||||
error: toSafeLogError(error)
|
||||
});
|
||||
|
||||
await finishImageGenerationArchiveFailed(recordId).catch((recordError) => {
|
||||
logger.error("更新归档失败记录失败", {
|
||||
recordId: recordId.toString(),
|
||||
error: toSafeLogError(recordError)
|
||||
});
|
||||
});
|
||||
}
|
||||
};
|
||||
|
||||
// 日志只读取流式聚合结果的元信息,不记录完整提示词、key 或图片内容。
|
||||
const getStreamContentLength = (upstreamResponse: unknown) => {
|
||||
if (
|
||||
|
||||
Reference in New Issue
Block a user