|
|
|
/* eslint-disable no-await-in-loop */
|
|
|
|
import type { DomainDoc, Udoc } from './interface';
|
|
|
|
import domain from './model/domain';
|
|
|
|
import user from './model/user';
|
|
|
|
import problem, { Field, Pdoc } from './model/problem';
|
|
|
|
|
|
|
|
export async function iterateAllDomain(cb: (ddoc: DomainDoc, current?: number, total?: number) => Promise<any>) {
|
|
|
|
const ddocs = await domain.getMulti().toArray();
|
|
|
|
for (const i in ddocs) await cb(ddocs[i], +i, ddocs.length);
|
|
|
|
}
|
|
|
|
|
|
|
|
export async function iterateAllUser(cb: (udoc: Udoc, current?: number, total?: number) => Promise<any>) {
|
|
|
|
const udocs = await user.getMulti().toArray();
|
|
|
|
for (const i in udocs) await cb(udocs[i], +i, udocs.length);
|
|
|
|
}
|
|
|
|
|
|
|
|
interface PartialPdoc extends Pdoc {
|
|
|
|
[key: string]: any,
|
|
|
|
}
|
|
|
|
|
|
|
|
export async function iterateAllProblemInDomain(
|
|
|
|
domainId: string,
|
|
|
|
fields: (Field | string)[],
|
|
|
|
cb: (pdoc: PartialPdoc, current?: number, total?: number) => Promise<any>,
|
|
|
|
) {
|
|
|
|
if (!fields.includes('domainId')) fields.push('domainId');
|
|
|
|
if (!fields.includes('docId')) fields.push('docId');
|
|
|
|
const cursor = problem.getMulti(domainId, {}, fields as any);
|
|
|
|
const total = await problem.getMulti(domainId, {}).count();
|
|
|
|
let i = 0;
|
|
|
|
while (await cursor.hasNext()) {
|
|
|
|
const doc = await cursor.next();
|
|
|
|
i++;
|
|
|
|
const res = await cb(doc, i, total);
|
|
|
|
if (res) await problem.edit(doc.domainId, doc.docId, res);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
export async function iterateAllProblem(
|
|
|
|
fields: (Field | string)[],
|
|
|
|
cb: (pdoc: PartialPdoc, current?: number, total?: number) => Promise<any>,
|
|
|
|
) {
|
|
|
|
await iterateAllDomain(async (d) => {
|
|
|
|
await iterateAllProblemInDomain(d._id, fields, cb);
|
|
|
|
});
|
|
|
|
}
|