feat: 替换redis搜索

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

@@ -1,13 +1,13 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { authToken } from '@/service/utils/tools';
-import { connectRedis } from '@/service/redis';
 import { ModelDataStatusEnum } from '@/constants/redis';
 import { generateVector } from '@/service/events/generateVector';
+import { PgClient } from '@/service/pg';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
   try {
-    const { dataId, text, q } = req.body as { dataId: string; text: string; q?: string };
+    const { dataId, a, q } = req.body as { dataId: string; a: string; q?: string };
     const { authorization } = req.headers;
 
     if (!authorization) {
@@ -21,26 +21,21 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
     // 凭证校验
     const userId = await authToken(authorization);
 
-    const redis = await connectRedis();
+    // 更新 pg 内容
+    await PgClient.update('modelData', {
+      where: [['id', dataId], 'AND', ['user_id', userId]],
+      values: [
+        { key: 'a', value: a },
+        ...(q
+          ? [
+              { key: 'q', value: q },
+              { key: 'status', value: ModelDataStatusEnum.waiting }
+            ]
+          : [])
+      ]
+    });
 
-    // 校验是否为该用户的数据
-    const dataItemUserId = await redis.hGet(dataId, 'userId');
-    if (dataItemUserId !== userId) {
-      throw new Error('无权操作');
-    }
-
-    // 更新
-    await redis.sendCommand([
-      'HMSET',
-      dataId,
-      ...(q ? ['q', q, 'status', ModelDataStatusEnum.waiting] : []),
-      'text',
-      text
-    ]);
-
-    if (q) {
-      generateVector();
-    }
+    q && generateVector();
 
     jsonRes(res);
   } catch (err) {