perf: completion chatId

client/src/pages/api/chat/init.ts

@@ -6,8 +6,7 @@ import { authUser } from '@/service/utils/auth';
 import { ChatItemType } from '@/types/chat';
 import { authApp } from '@/service/utils/auth';
 import mongoose from 'mongoose';
-import type { AppSchema, ChatSchema } from '@/types/mongoSchema';
-import { quoteLenKey, rawSearchKey } from '@/constants/chat';
+import type { ChatSchema } from '@/types/mongoSchema';
 import { getSpecialModule } from '@/components/ChatBox';
 
 /* 初始化我的聊天框，需要身份验证 */
@@ -20,72 +19,62 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
       chatId: '' | string;
     };
 
+    if (!appId) {
+      return jsonRes(res, {
+        code: 501,
+        message: "You don't have an app yet"
+      });
+    }
+
     await connectToDatabase();
 
-    // 没有 appId 时，直接获取用户的第一个id
-    const app = await (async () => {
-      if (!appId) {
-        const myModel = await App.findOne({ userId });
-        if (!myModel) {
-          const { _id } = await App.create({
-            name: '应用1',
-            userId
-          });
-          return (await App.findById(_id)) as AppSchema;
-        } else {
-          return myModel;
-        }
-      } else {
-        // 校验使用权限
-        const authRes = await authApp({
-          appId,
-          userId,
-          authUser: false,
-          authOwner: false
-        });
-        return authRes.app;
-      }
-    })();
-
-    appId = appId || app._id;
+    // 校验使用权限
+    const app = (
+      await authApp({
+        appId,
+        userId,
+        authUser: false,
+        authOwner: false
+      })
+    ).app;
 
     // 历史记录
     const { chat, history = [] }: { chat?: ChatSchema; history?: ChatItemType[] } =
       await (async () => {
         if (chatId) {
           // auth chatId
-          const chat = await Chat.findOne({
-            _id: chatId,
-            userId
-          });
+          const [chat, history] = await Promise.all([
+            Chat.findOne({
+              chatId,
+              userId
+            }),
+            Chat.aggregate([
+              {
+                $match: {
+                  chatId,
+                  userId: new mongoose.Types.ObjectId(userId)
+                }
+              },
+              {
+                $project: {
+                  content: {
+                    $slice: ['$content', -50] // 返回 content 数组的最后50个元素
+                  }
+                }
+              },
+              { $unwind: '$content' },
+              {
+                $project: {
+                  _id: '$content._id',
+                  obj: '$content.obj',
+                  value: '$content.value'
+                }
+              }
+            ])
+          ]);
           if (!chat) {
             throw new Error('聊天框不存在');
           }
-          // 获取 chat.content 数据
-          const history = await Chat.aggregate([
-            {
-              $match: {
-                _id: new mongoose.Types.ObjectId(chatId),
-                userId: new mongoose.Types.ObjectId(userId)
-              }
-            },
-            {
-              $project: {
-                content: {
-                  $slice: ['$content', -50] // 返回 content 数组的最后50个元素
-                }
-              }
-            },
-            { $unwind: '$content' },
-            {
-              $project: {
-                _id: '$content._id',
-                obj: '$content.obj',
-                value: '$content.value',
-                [quoteLenKey]: { $size: { $ifNull: [`$content.${rawSearchKey}`, []] } }
-              }
-            }
-          ]);
           return { history, chat };
         }
         return {};