Weishen/Hydro
2
0 Fork 0
mirror of https://gitcode.com/gh_mirrors/hy/Hydro.git synced 2026-10-20 01:38:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

core: problem_prefix_list: support index by pid

Browse Source
This commit is contained in:
undefined committed 2022-12-03 15:48:45 +08:00
1 parent d92f3f3ed6
commit bba42d5936
1 file changed
+4 -2
Download Patch File Download Diff File Expand all files Collapse all files
packages/hydrooj/src/handler/problem.ts
+4 -2
View File
@@ -2,7 +2,7 @@ import AdmZip from 'adm-zip';
import { readFile, statSync } from 'fs-extra'; import { readFile, statSync } from 'fs-extra';
import { isBinaryFile } from 'isbinaryfile'; import { isBinaryFile } from 'isbinaryfile';
import { import {
escapeRegExp, flattenDeep, intersection, isSafeInteger, pick, escapeRegExp, flattenDeep, intersection, isSafeInteger, pick, uniqBy,
} from 'lodash'; } from 'lodash';
import { FilterQuery, ObjectID } from 'mongodb'; import { FilterQuery, ObjectID } from 'mongodb';
import { nanoid } from 'nanoid'; import { nanoid } from 'nanoid';
@@ -997,13 +997,15 @@ export class ProblemPrefixListHandler extends Handler {
const pdoc = await problem.get(domainId, +prefix, ['domainId', 'docId', 'pid', 'title']); const pdoc = await problem.get(domainId, +prefix, ['domainId', 'docId', 'pid', 'title']);
if (pdoc) pdocs.unshift(pdoc); if (pdoc) pdocs.unshift(pdoc);
} }
const pdoc = await problem.get(domainId, prefix, ['domainId', 'docId', 'pid', 'title']);
if (pdoc) pdocs.unshift(pdoc);
if (pdocs.length < 20) { if (pdocs.length < 20) {
const search = global.Hydro.lib.problemSearch || defaultSearch; const search = global.Hydro.lib.problemSearch || defaultSearch;
const result = await search(domainId, prefix, { limit: 20 - pdocs.length }); const result = await search(domainId, prefix, { limit: 20 - pdocs.length });
const docs = await problem.getMulti(domainId, { docId: { $in: result.hits.map((i) => +i.split('/')[1]) } }).toArray(); const docs = await problem.getMulti(domainId, { docId: { $in: result.hits.map((i) => +i.split('/')[1]) } }).toArray();
pdocs.push(...docs); pdocs.push(...docs);
} }
this.response.body = pdocs; this.response.body = uniqBy(pdocs, 'docId');
} }
} }