forked from pinks/eris
1
0
Fork 0
nyx/api/statsRoute.ts

126 lines
4.0 KiB
TypeScript
Raw Normal View History

2023-10-21 10:40:42 +00:00
import { subMinutes } from "date-fns";
2023-10-09 19:03:31 +00:00
import { createEndpoint, createMethodFilter, createPathFilter } from "t_rest/server";
import { getDailyStats } from "../app/dailyStatsStore.ts";
2023-10-13 11:47:57 +00:00
import { generationStore } from "../app/generationStore.ts";
2023-10-21 10:40:42 +00:00
import { globalStats } from "../app/globalStats.ts";
import { getUserDailyStats } from "../app/userDailyStatsStore.ts";
import { getUserStats } from "../app/userStatsStore.ts";
2023-10-13 11:47:57 +00:00
const STATS_INTERVAL_MIN = 3;
2023-10-09 19:03:31 +00:00
export const statsRoute = createPathFilter({
2023-10-10 16:21:25 +00:00
"": createMethodFilter({
2023-10-09 19:03:31 +00:00
GET: createEndpoint(
{ query: null, body: null },
async () => {
2023-10-13 11:47:57 +00:00
const after = subMinutes(new Date(), STATS_INTERVAL_MIN);
const generations = await generationStore.getAll({ after });
const imagesPerMinute = generations.length / STATS_INTERVAL_MIN;
const stepsPerMinute = generations
.map((generation) => generation.value.info?.steps ?? 0)
.reduce((sum, steps) => sum + steps, 0) / STATS_INTERVAL_MIN;
const pixelsPerMinute = generations
.map((generation) =>
(generation.value.info?.width ?? 0) * (generation.value.info?.height ?? 0)
)
.reduce((sum, pixels) => sum + pixels, 0) / STATS_INTERVAL_MIN;
const pixelStepsPerMinute = generations
.map((generation) =>
(generation.value.info?.width ?? 0) * (generation.value.info?.height ?? 0) *
(generation.value.info?.steps ?? 0)
)
.reduce((sum, pixelSteps) => sum + pixelSteps, 0) / STATS_INTERVAL_MIN;
2023-10-09 19:03:31 +00:00
return {
status: 200,
body: {
type: "application/json",
2023-10-10 16:21:25 +00:00
data: {
2023-10-13 11:47:57 +00:00
imageCount: globalStats.imageCount,
stepCount: globalStats.stepCount,
pixelCount: globalStats.pixelCount,
pixelStepCount: globalStats.pixelStepCount,
userCount: globalStats.userIds.length,
imagesPerMinute,
stepsPerMinute,
pixelsPerMinute,
pixelStepsPerMinute,
2023-10-10 16:21:25 +00:00
},
2023-10-09 19:03:31 +00:00
},
};
},
),
}),
"daily/{year}/{month}/{day}": createMethodFilter({
GET: createEndpoint(
{ query: null, body: null },
async ({ params }) => {
const year = Number(params.year);
const month = Number(params.month);
const day = Number(params.day);
2023-10-10 16:21:25 +00:00
const stats = await getDailyStats(year, month, day);
2023-10-09 19:03:31 +00:00
return {
status: 200,
body: {
type: "application/json",
2023-10-10 16:21:25 +00:00
data: {
imageCount: stats.imageCount,
pixelCount: stats.pixelCount,
userCount: stats.userIds.length,
timestamp: stats.timestamp,
},
},
};
},
),
}),
"users/{userId}": createMethodFilter({
GET: createEndpoint(
{ query: null, body: null },
async ({ params }) => {
const userId = Number(params.userId);
const stats = await getUserStats(userId);
return {
status: 200,
body: {
type: "application/json",
data: {
imageCount: stats.imageCount,
pixelCount: stats.pixelCount,
tagCountMap: stats.tagCountMap,
timestamp: stats.timestamp,
},
},
};
},
),
}),
"users/{userId}/daily/{year}/{month}/{day}": createMethodFilter({
GET: createEndpoint(
{ query: null, body: null },
async ({ params }) => {
const userId = Number(params.userId);
const year = Number(params.year);
const month = Number(params.month);
const day = Number(params.day);
const stats = await getUserDailyStats(userId, year, month, day);
return {
status: 200,
body: {
type: "application/json",
data: {
imageCount: stats.imageCount,
pixelCount: stats.pixelCount,
timestamp: stats.timestamp,
},
2023-10-09 19:03:31 +00:00
},
};
},
),
}),
});