324e4a0e75
* chat item table * perf: chat item save * docs * limit * docs * docs * perf: node card * docs * docs
59 lines
1.4 KiB
TypeScript
59 lines
1.4 KiB
TypeScript
// 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, Chat } from '@/service/mongo';
|
|
|
|
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);
|
|
|
|
await Promise.all(
|
|
chats.map((chat) =>
|
|
Chat.findByIdAndUpdate(chat._id, {
|
|
chatId: String(chat._id),
|
|
source: 'online'
|
|
})
|
|
)
|
|
);
|
|
}
|