Infrastructure/zipline
1
Fork 0
Code Issues Pull requests Activity

fix: attempt to fix #243

Browse source
This commit is contained in:
diced 2022-12-09 19:31:35 -08:00
parent 9648856052
commit 76267c00d5
No known key found for this signature in database
GPG key ID: 370BD1BA142842D1
2 changed files with 31 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
src/server/index.ts
View file

@ -184,15 +184,16 @@ async function start() {
}`
);
clearInvites.bind(server)();
stats.bind(server)();
await clearInvites.bind(server)();
await stats.bind(server)();
setInterval(() => clearInvites.bind(server)(), config.core.invites_interval * 1000);
setInterval(() => stats.bind(server)(), config.core.stats_interval * 1000);
}
async function stats(this: FastifyInstance) {
const stats = await getStats(this.prisma, this.datasource);
const stats = await getStats.bind(this)();
await this.prisma.stats.create({
data: {
data: stats,

35
src/server/util.ts
View file

@ -1,6 +1,7 @@
import { PrismaClient } from '@prisma/client';
import { Migrate } from '@prisma/migrate/dist/Migrate';
import { ensureDatabaseExists } from '@prisma/migrate/dist/utils/ensureDatabaseExists';
import { FastifyInstance } from 'fastify';
import { ServerResponse } from 'http';
import { Datasource } from '../lib/datasources';
import Logger from '../lib/logger';
@ -57,19 +58,31 @@ export function redirect(res: ServerResponse, url: string) {
res.end();
}
export async function getStats(prisma: PrismaClient, datasource: Datasource) {
const size = await datasource.fullSize();
const byUser = await prisma.image.groupBy({
export async function getStats(this: FastifyInstance) {
const logger = this.logger.child('stats');
const size = await this.datasource.fullSize();
logger.debug(`full size: ${size}`);
const byUser = await this.prisma.image.groupBy({
by: ['userId'],
_count: {
_all: true,
},
});
const count_users = await prisma.user.count();
logger.debug(`by user: ${JSON.stringify(byUser)}`);
const count_users = await this.prisma.user.count();
logger.debug(`count users: ${count_users}`);
const count_by_user = [];
for (let i = 0, L = byUser.length; i !== L; ++i) {
const user = await prisma.user.findFirst({
if (!byUser[i].userId) {
logger.debug(`skipping user ${byUser[i]}`);
continue;
}
const user = await this.prisma.user.findFirst({
where: {
id: byUser[i].userId,
},
@ -80,21 +93,25 @@ export async function getStats(prisma: PrismaClient, datasource: Datasource) {
count: byUser[i]._count._all,
});
}
logger.debug(`count by user: ${JSON.stringify(count_by_user)}`);
const count = await prisma.image.count();
const count = await this.prisma.image.count();
logger.debug(`count files: ${JSON.stringify(count)}`);
const views = await prisma.image.aggregate({
const views = await this.prisma.image.aggregate({
_sum: {
views: true,
},
});
logger.debug(`sum views: ${JSON.stringify(views)}`);
const typesCount = await prisma.image.groupBy({
const typesCount = await this.prisma.image.groupBy({
by: ['mimetype'],
_count: {
mimetype: true,
},
});
logger.debug(`types count: ${JSON.stringify(typesCount)}`);
const types_count = [];
for (let i = 0, L = typesCount.length; i !== L; ++i)
types_count.push({
@ -102,6 +119,8 @@ export async function getStats(prisma: PrismaClient, datasource: Datasource) {
count: typesCount[i]._count.mimetype,
});
logger.debug(`types count: ${JSON.stringify(types_count)}`);
return {
size: bytesToHuman(size),
size_num: size,