monkeytype/backend/jobs/update-leaderboards.ts

import { CronJob } from "cron";
import BotDAO from "../dao/bot";
import George from "../tasks/george";
import { Document, WithId } from "mongodb";
import LeaderboardsDAO from "../dao/leaderboards";
import ConfigurationClient from "../init/configuration";

const CRON_SCHEDULE = "30 14/15 * * * *";
const RECENT_AGE_MINUTES = 10;
const RECENT_AGE_MILLISECONDS = RECENT_AGE_MINUTES * 60 * 1000;

async function getTop10(leaderboardTime: string): Promise<WithId<Document>[]> {
  return (await LeaderboardsDAO.get(
    "time",
    leaderboardTime,
    "english",
    0,
    10
  )) as any[];
}

async function updateLeaderboardAndNotifyChanges(
  leaderboardTime: string
): Promise<void> {
  const top10BeforeUpdate = await getTop10(leaderboardTime);

  const previousRecordsMap = Object.fromEntries(
    top10BeforeUpdate.map((record) => {
      return [record.uid, record];
    })
  );

  await LeaderboardsDAO.update("time", leaderboardTime, "english");

  const top10AfterUpdate = await getTop10(leaderboardTime);

  const newRecords = top10AfterUpdate.filter((record) => {
    const userId = record.uid;

    const userImprovedRank =
      userId in previousRecordsMap &&
      previousRecordsMap[userId].rank > record.rank;

    const newUserInTop10 = !(userId in previousRecordsMap);

    const isRecentRecord =
      record.timestamp > Date.now() - RECENT_AGE_MILLISECONDS;

    return (userImprovedRank || newUserInTop10) && isRecentRecord;
  });

  if (newRecords.length > 0) {
    const cachedConfig = await ConfigurationClient.getCachedConfiguration();

    const leaderboardId = `time ${leaderboardTime} english`;

    if (cachedConfig.useRedisForBotTasks.enabled) {
      await George.announceLbUpdate(newRecords, leaderboardId);
    }

    await BotDAO.announceLbUpdate(newRecords, leaderboardId);
  }
}

async function updateLeaderboards(): Promise<void> {
  await updateLeaderboardAndNotifyChanges("15");
  await updateLeaderboardAndNotifyChanges("60");
}

export default new CronJob(CRON_SCHEDULE, updateLeaderboards);