feat: pg引入

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

@@ -1,7 +1,7 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { authToken } from '@/service/utils/tools';
-import { connectRedis } from '@/service/redis';
+import { connectPg } from '@/service/pg';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
   try {
@@ -21,15 +21,9 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
     // 凭证校验
     const userId = await authToken(authorization);
 
-    const redis = await connectRedis();
+    const pg = await connectPg();
+    await pg.query(`DELETE FROM modelData WHERE user_id = '${userId}' AND id = '${dataId}'`);
 
-    // 校验是否为该用户的数据
-    const dataItemUserId = await redis.hGet(dataId, 'userId');
-    if (dataItemUserId !== userId) {
-      throw new Error('无权操作');
-    }
-    // 删除
-    await redis.del(dataId);
     jsonRes(res);
   } catch (err) {
     console.log(err);

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

@@ -2,8 +2,8 @@ import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { connectToDatabase } from '@/service/mongo';
 import { authToken } from '@/service/utils/tools';
-import { connectRedis } from '@/service/redis';
-import { VecModelDataIdx } from '@/constants/redis';
+import { connectPg } from '@/service/pg';
+import type { PgModelDataItemType } from '@/types/pg';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
   try {
@@ -35,34 +35,20 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
     const userId = await authToken(authorization);
 
     await connectToDatabase();
-    const redis = await connectRedis();
+    const pg = await connectPg();
 
-    // 从 redis 中获取数据
-    const searchRes = await redis.ft.search(
-      VecModelDataIdx,
-      `@modelId:{${modelId}} @userId:{${userId}} ${searchText ? `*${searchText}*` : ''}`,
-      {
-        RETURN: ['q', 'text', 'status'],
-        LIMIT: {
-          from: (pageNum - 1) * pageSize,
-          size: pageSize
-        },
-        SORTBY: {
-          BY: 'modelId',
-          DIRECTION: 'DESC'
-        }
-      }
-    );
+    const searchRes = await pg.query<PgModelDataItemType>(`SELECT id, q, a, status
+              FROM modelData
+              WHERE user_id='${userId}' AND model_id='${modelId}'
+              LIMIT ${pageSize} OFFSET ${pageSize * (pageNum - 1)}
+            `);
 
     jsonRes(res, {
       data: {
         pageNum,
         pageSize,
-        data: searchRes.documents.map((item) => ({
-          id: item.id,
-          ...item.value
-        })),
-        total: searchRes.total
+        data: searchRes.rows,
+        total: searchRes.rowCount
       }
     });
   } catch (err) {

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, {