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: fix contest paginator

Browse Source
This commit is contained in:
undefined committed 2021-05-03 02:00:20 +08:00
1 parent 0d7c53b927
commit 2a3933de1a
14 files changed
+46 -40
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

packages/hydrooj/src/lib/rank.ts
+9 -8
View File
@@ -3,24 +3,25 @@ import paginate from './paginate';
type G<T> = Promise<[[number, T], () => G<T>] | []>;
interface ranked {
<T>(cursor: Cursor<T>, equ: (a: T, b: T) => boolean, page: number): Promise<[number, T][]>;
<T>(cursor: Cursor<T>, equ: (a: T, b: T) => boolean, page: number): Promise<[[number, T][], number]>;
iterate<T>(cursor: Cursor<T>, equ: (a: T, b: T) => boolean): G<T>;
all<T>(cursor: Cursor<T>, equ: (a: T, b: T) => boolean): Promise<[number, T][]>;
all<T>(cursor: Cursor<T>, equ: (a: T, b: T) => boolean): Promise<[[number, T][], 1]>;
}
const ranked: ranked = async function rankedPage(cursor, equ = (a, b) => a === b, page) {
if (!page) return await ranked.all(cursor, equ);
let last = null;
let r = (page - 1) * 100;
let count = r;
const results = [];
const [docs] = await paginate(cursor, page, 100);
const [docs, nPages] = await paginate(cursor, page, 100);
for (const doc of docs) {
count++;
if (count === 1 || !equ(last, doc)) r = count;
if (!last || !equ(last, doc)) r = count;
last = doc;
results.push([r, doc]);
}
return results;
return [results, nPages];
};
ranked.iterate = async function rankedIterate<T>(cursor: Cursor<T>, equ = (a, b) => a === b) {
let last = null;
@@ -30,7 +31,7 @@ ranked.iterate = async function rankedIterate<T>(cursor: Cursor<T>, equ = (a, b)
if (!await cursor.hasNext()) return [];
const doc = await cursor.next();
count++;
if (count === 1 || !equ(last, doc)) r = count;
if (!last || !equ(last, doc)) r = count;
last = doc;
return [[r, doc], getNext];
}
@@ -44,11 +45,11 @@ ranked.all = async function rankedAll(cursor, equ = (a, b) => a === b) {
const docs = await cursor.toArray();
for (const doc of docs) {
count++;
if (count === 1 || !equ(last, doc)) r = count;
if (!last || !equ(last, doc)) r = count;
last = doc;
results.push([r, doc]);
}
return results;
return [results, 1];
};
global.Hydro.lib.rank = ranked;