forked from pinks/eris
66 lines
2.1 KiB
TypeScript
66 lines
2.1 KiB
TypeScript
|
import { UTCDateMini } from "@date-fns/utc";
|
||
|
|
import { hoursToMilliseconds, isSameDay, minutesToMilliseconds } from "date-fns";
|
||
|
|
import { getLogger } from "std/log/mod.ts";
|
||
|
|
import { JsonSchema, jsonType } from "t_rest/server";
|
||
|
|
import { db } from "./db.ts";
|
||
|
|
import { generationStore } from "./generationStore.ts";
|
||
|
|
import { kvMemoize } from "./kvMemoize.ts";
|
||
|
|
|
||
|
|
const logger = () => getLogger();
|
||
|
|
|
||
|
|
export const dailyStatsSchema = {
|
||
|
|
type: "object",
|
||
|
|
properties: {
|
||
|
|
userIds: { type: "array", items: { type: "number" } },
|
||
|
|
imageCount: { type: "number" },
|
||
|
|
pixelCount: { type: "number" },
|
||
|
|
timestamp: { type: "number" },
|
||
|
|
},
|
||
|
|
required: ["userIds", "imageCount", "pixelCount", "timestamp"],
|
||
|
|
} as const satisfies JsonSchema;
|
||
|
|
|
||
|
|
export type DailyStats = jsonType<typeof dailyStatsSchema>;
|
||
|
|
|
||
|
|
export const getDailyStats = kvMemoize(
|
||
|
|
db,
|
||
|
|
["dailyStats"],
|
||
|
|
async (year: number, month: number, day: number): Promise<DailyStats> => {
|
||
|
|
const userIdSet = new Set<number>();
|
||
|
|
let imageCount = 0;
|
||
|
|
let pixelCount = 0;
|
||
|
|
|
||
|
|
const after = new Date(Date.UTC(year, month - 1, day));
|
||
|
|
const before = new Date(Date.UTC(year, month - 1, day + 1));
|
||
|
|
|
||
|
|
logger().info(`Calculating daily stats for ${year}-${month}-${day}`);
|
||
|
|
|
||
|
|
for await (
|
||
|
|
const generation of generationStore.listAll({ after, before })
|
||
|
|
) {
|
||
|
|
userIdSet.add(generation.value.from.id);
|
||
|
|
imageCount++;
|
||
|
|
pixelCount += (generation.value.info?.width ?? 0) * (generation.value.info?.height ?? 0);
|
||
|
|
}
|
||
|
|
|
||
|
|
return {
|
||
|
|
userIds: [...userIdSet],
|
||
|
|
imageCount,
|
||
|
|
pixelCount,
|
||
|
|
timestamp: Date.now(),
|
||
|
|
};
|
||
|
|
},
|
||
|
|
{
|
||
|
|
// expire in 1 minute if was calculated on the same day, otherwise 7-14 days.
|
||
|
|
expireIn: (result, year, month, day) => {
|
||
|
|
const requestDate = new UTCDateMini(year, month - 1, day);
|
||
|
|
const calculatedDate = new UTCDateMini(result.timestamp);
|
||
|
|
return isSameDay(requestDate, calculatedDate)
|
||
|
|
? minutesToMilliseconds(1)
|
||
|
|
: hoursToMilliseconds(24 * 7 + Math.random() * 24 * 7);
|
||
|
|
},
|
||
|
|
// should cache if the stats are non-zero
|
||
|
|
shouldCache: (result) =>
|
||
|
|
result.userIds.length > 0 || result.imageCount > 0 || result.pixelCount > 0,
|
||
|
|
},
|
||
|
|
);
|