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

每日签到功能

Browse Source
This commit is contained in:
undefined committed 2020-08-13 13:03:33 +08:00
1 parent 06847d9b19
commit 9887deca6a
7 files changed
+52 -11
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

packages/hydrooj/src/script/rating.ts
+13 -3
View File
@@ -84,7 +84,8 @@ async function runContest(...arg: any[]) {
}
export async function calcLevel(domainId: string, report: Function) {
const dudocs = await domain.getMultiInDomain(domainId).sort('rp', -1).toArray();
const dudocs = await domain.getMultiUserInDomain(domainId).sort('rp', -1).toArray();
if (!dudocs.length) return;
let last = { rp: null };
let rank = 0;
let count = 0;
@@ -97,8 +98,8 @@ export async function calcLevel(domainId: string, report: Function) {
bulk.find({ _id: dudoc._id }).updateOne({ $set: { rank } });
last = dudoc;
if (count % 1000 === 0) report({ message: `#${count}: Rank ${rank}` });
await bulk.execute();
}
await bulk.execute();
if (!rank) {
report({ message: 'No one has rp' });
return;
@@ -120,6 +121,11 @@ export async function calcLevel(domainId: string, report: Function) {
async function runInDomain(domainId: string, isSub: boolean, report: Function) {
const udict: ND = {};
const deltaudict: ND = {};
const dudocs = await domain.getMultiUserInDomain(domainId).toArray();
for (const dudoc of dudocs) {
deltaudict[dudoc.uid] = dudoc.rpdelta || 0;
}
const contests = await contest.getMulti(domainId, { rated: true }, -1).sort('endAt', -1).toArray();
await report({ message: `Found ${contests.length} contests in ${domainId}` });
for (const i in contests) {
@@ -146,7 +152,11 @@ async function runInDomain(domainId: string, isSub: boolean, report: Function) {
await domain.setMultiUserInDomain(domainId, {}, { rp: 1500 });
const tasks = [];
for (const uid in udict) {
tasks.push(domain.setUserInDomain(domainId, parseInt(uid, 10), { rp: udict[uid] }));
tasks.push(
domain.setUserInDomain(
domainId, parseInt(uid, 10), { rp: udict[uid] + (deltaudict[uid] || 0) },
),
);
}
await Promise.all(tasks);
await calcLevel(domainId, report);