76 lines
1.3 KiB
TypeScript
76 lines
1.3 KiB
TypeScript
|
import autobind from 'autobind-decorator';
|
||
|
import Chart, { Obj } from './';
|
||
|
import User, { IUser, isLocalUser } from '../models/user';
|
||
|
|
||
|
/**
|
||
|
* ユーザーに関するチャート
|
||
|
*/
|
||
|
type UsersLog = {
|
||
|
local: {
|
||
|
/**
|
||
|
* 集計期間時点での、全ユーザー数
|
||
|
*/
|
||
|
total: number;
|
||
|
|
||
|
/**
|
||
|
* 増加したユーザー数
|
||
|
*/
|
||
|
inc: number;
|
||
|
|
||
|
/**
|
||
|
* 減少したユーザー数
|
||
|
*/
|
||
|
dec: number;
|
||
|
};
|
||
|
|
||
|
remote: UsersLog['local'];
|
||
|
};
|
||
|
|
||
|
class UsersChart extends Chart<UsersLog> {
|
||
|
constructor() {
|
||
|
super('users');
|
||
|
}
|
||
|
|
||
|
@autobind
|
||
|
protected async getTemplate(init: boolean, latest?: UsersLog): Promise<UsersLog> {
|
||
|
const [localCount, remoteCount] = init ? await Promise.all([
|
||
|
User.count({ host: null }),
|
||
|
User.count({ host: { $ne: null } })
|
||
|
]) : [
|
||
|
latest ? latest.local.total : 0,
|
||
|
latest ? latest.remote.total : 0
|
||
|
];
|
||
|
|
||
|
return {
|
||
|
local: {
|
||
|
total: localCount,
|
||
|
inc: 0,
|
||
|
dec: 0
|
||
|
},
|
||
|
remote: {
|
||
|
total: remoteCount,
|
||
|
inc: 0,
|
||
|
dec: 0
|
||
|
}
|
||
|
};
|
||
|
}
|
||
|
|
||
|
@autobind
|
||
|
public async update(user: IUser, isAdditional: boolean) {
|
||
|
const update: Obj = {};
|
||
|
|
||
|
update.total = isAdditional ? 1 : -1;
|
||
|
if (isAdditional) {
|
||
|
update.inc = 1;
|
||
|
} else {
|
||
|
update.dec = 1;
|
||
|
}
|
||
|
|
||
|
await this.inc({
|
||
|
[isLocalUser(user) ? 'local' : 'remote']: update
|
||
|
});
|
||
|
}
|
||
|
}
|
||
|
|
||
|
export default new UsersChart();
|