feat: pg引入

src/pages/api/model/data/pushModelDataInput.ts

@@ -4,14 +4,13 @@ import { connectToDatabase, Model } from '@/service/mongo';
 import { authToken } from '@/service/utils/tools';
 import { ModelDataSchema } from '@/types/mongoSchema';
 import { generateVector } from '@/service/events/generateVector';
-import { connectRedis } from '@/service/redis';
-import { VecModelDataPrefix, ModelDataStatusEnum } from '@/constants/redis';
+import { connectPg } from '@/service/pg';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
   try {
     const { modelId, data } = req.body as {
       modelId: string;
-      data: { text: ModelDataSchema['text']; q: ModelDataSchema['q'] }[];
+      data: { a: ModelDataSchema['a']; q: ModelDataSchema['q'] }[];
     };
     const { authorization } = req.headers;
 
@@ -27,7 +26,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
     const userId = await authToken(authorization);
 
     await connectToDatabase();
-    const redis = await connectRedis();
+    const pg = await connectPg();
 
     // 验证是否是该用户的 model
     const model = await Model.findOne({
@@ -39,29 +38,23 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
       throw new Error('无权操作该模型');
     }
 
-    const insertRes = await Promise.allSettled(
-      data.map((item) => {
-        return redis.sendCommand([
-          'HMSET',
-          `${VecModelDataPrefix}:${item.q.id}`,
-          'userId',
-          userId,
-          'modelId',
-          modelId,
-          'q',
-          item.q.text,
-          'text',
-          item.text,
-          'status',
-          ModelDataStatusEnum.waiting
-        ]);
-      })
+    // 插入记录
+    await pg.query(
+      `INSERT INTO modelData (user_id, model_id, q, a, status) VALUES ${data
+        .map(
+          (item) =>
+            `('${userId}', '${modelId}', '${item.q.replace(/\'/g, '"')}', '${item.a.replace(
+              /\'/g,
+              '"'
+            )}', 'waiting')`
+        )
+        .join(',')}`
     );
 
     generateVector();
 
     jsonRes(res, {
-      data: insertRes.filter((item) => item.status === 'rejected').length
+      data: 0
     });
   } catch (err) {
     jsonRes(res, {