perf: completion chatId

client/src/pages/api/admin/initChat.ts

@@ -0,0 +1,61 @@
+// Next.js API route support: https://nextjs.org/docs/api-routes/introduction
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { authUser } from '@/service/utils/auth';
+import { connectToDatabase, TrainingData, User, promotionRecord, Chat } from '@/service/mongo';
+import { PRICE_SCALE } from '@/constants/common';
+
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  try {
+    await authUser({ req, authRoot: true });
+    await connectToDatabase();
+
+    const { limit = 1000 } = req.body as { limit: number };
+    let skip = 0;
+    const total = await Chat.countDocuments({
+      chatId: { $exists: false }
+    });
+    let promise = Promise.resolve();
+    console.log(total);
+
+    for (let i = 0; i < total; i += limit) {
+      const skipVal = skip;
+      skip += limit;
+      promise = promise
+        .then(() => init(limit, skipVal))
+        .then(() => {
+          console.log(skipVal);
+        });
+    }
+
+    await promise;
+
+    jsonRes(res, {});
+  } catch (error) {
+    jsonRes(res, {
+      code: 500,
+      error
+    });
+  }
+}
+
+async function init(limit: number, skip: number) {
+  // 遍历 app
+  const chats = await Chat.find(
+    {
+      chatId: { $exists: false }
+    },
+    '_id'
+  )
+    .limit(limit)
+    .skip(skip);
+
+  await Promise.all(
+    chats.map((chat) =>
+      Chat.findByIdAndUpdate(chat._id, {
+        chatId: String(chat._id),
+        source: 'online'
+      })
+    )
+  );
+}