sharkey/packages/backend/src/server/api/endpoints/stats.ts
2022-09-24 06:45:44 +09:00

103 lines
2.5 KiB
TypeScript

import { Inject, Injectable } from '@nestjs/common';
import { IsNull } from 'typeorm';
import type { InstancesRepository, NoteReactionsRepository, NotesRepository, UsersRepository } from '@/models/index.js';
import { Endpoint } from '@/server/api/endpoint-base.js';
import { DI } from '@/di-symbols.js';
export const meta = {
requireCredential: false,
tags: ['meta'],
res: {
type: 'object',
optional: false, nullable: false,
properties: {
notesCount: {
type: 'number',
optional: false, nullable: false,
},
originalNotesCount: {
type: 'number',
optional: false, nullable: false,
},
usersCount: {
type: 'number',
optional: false, nullable: false,
},
originalUsersCount: {
type: 'number',
optional: false, nullable: false,
},
instances: {
type: 'number',
optional: false, nullable: false,
},
driveUsageLocal: {
type: 'number',
optional: false, nullable: false,
},
driveUsageRemote: {
type: 'number',
optional: false, nullable: false,
},
},
},
} as const;
export const paramDef = {
type: 'object',
properties: {},
required: [],
} as const;
// eslint-disable-next-line import/no-default-export
@Injectable()
export default class extends Endpoint<typeof meta, typeof paramDef> {
constructor(
@Inject(DI.usersRepository)
private usersRepository: UsersRepository,
@Inject(DI.notesRepository)
private notesRepository: NotesRepository,
@Inject(DI.instancesRepository)
private instancesRepository: InstancesRepository,
@Inject(DI.noteReactionsRepository)
private noteReactionsRepository: NoteReactionsRepository,
) {
super(meta, paramDef, async () => {
const [
notesCount,
originalNotesCount,
usersCount,
originalUsersCount,
reactionsCount,
//originalReactionsCount,
instances,
] = await Promise.all([
this.notesRepository.count({ cache: 3600000 }), // 1 hour
this.notesRepository.count({ where: { userHost: IsNull() }, cache: 3600000 }),
this.usersRepository.count({ cache: 3600000 }),
this.usersRepository.count({ where: { host: IsNull() }, cache: 3600000 }),
this.noteReactionsRepository.count({ cache: 3600000 }), // 1 hour
//this.noteReactionsRepository.count({ where: { userHost: IsNull() }, cache: 3600000 }),
this.instancesRepository.count({ cache: 3600000 }),
]);
return {
notesCount,
originalNotesCount,
usersCount,
originalUsersCount,
reactionsCount,
//originalReactionsCount,
instances,
driveUsageLocal: 0,
driveUsageRemote: 0,
};
});
}
}