first-update

2026-03-17 14:36:31 +08:00
parent 72f08aee7c
commit 4eddf05e79
516 changed files with 115270 additions and 1 deletions
--- a/easy-dataset-main/app/api/projects/[projectId]/chunks/batch-edit/route.js
+++ b/easy-dataset-main/app/api/projects/[projectId]/chunks/batch-edit/route.js
@@ -0,0 +1,102 @@
+import { NextRequest, NextResponse } from 'next/server';
+import { PrismaClient } from '@prisma/client';
+
+const prisma = new PrismaClient();
+
+/**
+ * 批量编辑文本块内容
+ * POST /api/projects/[projectId]/chunks/batch-edit
+ */
+export async function POST(request, { params }) {
+  try {
+    const { projectId } = params;
+    const body = await request.json();
+    const { position, content, chunkIds } = body;
+
+    // 验证参数
+    if (!position || !content || !chunkIds || !Array.isArray(chunkIds) || chunkIds.length === 0) {
+      return NextResponse.json({ error: 'Missing required parameters: position, content, chunkIds' }, { status: 400 });
+    }
+
+    if (!['start', 'end'].includes(position)) {
+      return NextResponse.json({ error: 'Position must be "start" or "end"' }, { status: 400 });
+    }
+
+    // 验证项目权限（获取要编辑的文本块）
+    const chunksToUpdate = await prisma.chunks.findMany({
+      where: {
+        id: { in: chunkIds },
+        projectId: projectId
+      },
+      select: {
+        id: true,
+        content: true,
+        name: true
+      }
+    });
+
+    if (chunksToUpdate.length === 0) {
+      return NextResponse.json({ error: 'Not found' }, { status: 404 });
+    }
+
+    if (chunksToUpdate.length !== chunkIds.length) {
+      return NextResponse.json({ error: 'Some chunks not found' }, { status: 400 });
+    }
+
+    // 准备更新数据
+    const updates = chunksToUpdate.map(chunk => {
+      let newContent;
+
+      if (position === 'start') {
+        // 在开头添加内容
+        newContent = content + '\n\n' + chunk.content;
+      } else {
+        // 在结尾添加内容
+        newContent = chunk.content + '\n\n' + content;
+      }
+
+      return {
+        where: { id: chunk.id },
+        data: {
+          content: newContent,
+          size: newContent.length,
+          updateAt: new Date()
+        }
+      };
+    });
+
+    async function processBatches(items, batchSize, processFn) {
+      const results = [];
+      for (let i = 0; i < items.length; i += batchSize) {
+        const batch = items.slice(i, i + batchSize);
+        const batchResults = await Promise.all(batch.map(processFn));
+        results.push(...batchResults);
+      }
+      return results;
+    }
+
+    const BATCH_SIZE = 50; // 每批处理 50 个
+    await processBatches(updates, BATCH_SIZE, update => prisma.chunks.update(update));
+
+    // 记录操作日志（可选）
+    console.log(`Successfully updated ${chunksToUpdate.length} chunks`);
+
+    return NextResponse.json({
+      success: true,
+      updatedCount: chunksToUpdate.length,
+      message: `Successfully updated ${chunksToUpdate.length} chunks`
+    });
+  } catch (error) {
+    console.error('批量编辑文本块失败:', error);
+
+    return NextResponse.json(
+      {
+        error: 'Batch edit chunks failed',
+        details: error.message
+      },
+      { status: 500 }
+    );
+  } finally {
+    await prisma.$disconnect();
+  }
+}