85 lines
2.9 KiB
TypeScript
85 lines
2.9 KiB
TypeScript
import Chart, { KVs } from '../core.js';
|
|
import { Followings, Instances } from '@/models/index.js';
|
|
import { name, schema } from './entities/federation.js';
|
|
import { fetchMeta } from '@/misc/fetch-meta.js';
|
|
import { In, MoreThan, Not } from 'typeorm';
|
|
|
|
/**
|
|
* フェデレーションに関するチャート
|
|
*/
|
|
// eslint-disable-next-line import/no-default-export
|
|
export default class FederationChart extends Chart<typeof schema> {
|
|
constructor() {
|
|
super(name, schema);
|
|
}
|
|
|
|
protected async tickMajor(): Promise<Partial<KVs<typeof schema>>> {
|
|
return {
|
|
};
|
|
}
|
|
|
|
protected async tickMinor(): Promise<Partial<KVs<typeof schema>>> {
|
|
const meta = await fetchMeta();
|
|
|
|
const suspendedInstancesQuery = Instances.createQueryBuilder('instance')
|
|
.select('instance.host')
|
|
.where('instance.isSuspended = true');
|
|
|
|
const pubsubSubQuery = Followings.createQueryBuilder('f')
|
|
.select('f.followerHost')
|
|
.where('f.followerHost IS NOT NULL');
|
|
|
|
const [sub, pub, pubsub, active] = await Promise.all([
|
|
Followings.createQueryBuilder('following')
|
|
.select('COUNT(DISTINCT following.followeeHost)')
|
|
.where('following.followeeHost IS NOT NULL')
|
|
.andWhere(meta.blockedHosts.length === 0 ? '1=1' : `following.followeeHost NOT IN (:...blocked)`, { blocked: meta.blockedHosts })
|
|
.andWhere(`following.followeeHost NOT IN (${ suspendedInstancesQuery.getQuery() })`)
|
|
.getRawOne()
|
|
.then(x => parseInt(x.count, 10)),
|
|
Followings.createQueryBuilder('following')
|
|
.select('COUNT(DISTINCT following.followerHost)')
|
|
.where('following.followerHost IS NOT NULL')
|
|
.andWhere(meta.blockedHosts.length === 0 ? '1=1' : `following.followerHost NOT IN (:...blocked)`, { blocked: meta.blockedHosts })
|
|
.andWhere(`following.followerHost NOT IN (${ suspendedInstancesQuery.getQuery() })`)
|
|
.getRawOne()
|
|
.then(x => parseInt(x.count, 10)),
|
|
Followings.createQueryBuilder('following')
|
|
.select('COUNT(DISTINCT following.followeeHost)')
|
|
.where('following.followeeHost IS NOT NULL')
|
|
.andWhere(meta.blockedHosts.length === 0 ? '1=1' : `following.followeeHost NOT IN (:...blocked)`, { blocked: meta.blockedHosts })
|
|
.andWhere(`following.followeeHost NOT IN (${ suspendedInstancesQuery.getQuery() })`)
|
|
.andWhere(`following.followeeHost IN (${ pubsubSubQuery.getQuery() })`)
|
|
.setParameters(pubsubSubQuery.getParameters())
|
|
.getRawOne()
|
|
.then(x => parseInt(x.count, 10)),
|
|
Instances.count({
|
|
host: Not(In(meta.blockedHosts)),
|
|
isSuspended: false,
|
|
lastCommunicatedAt: MoreThan(new Date(Date.now() - (1000 * 60 * 60 * 24 * 30))),
|
|
}),
|
|
]);
|
|
|
|
return {
|
|
'sub': sub,
|
|
'pub': pub,
|
|
'pubsub': pubsub,
|
|
'active': active,
|
|
};
|
|
}
|
|
|
|
public async deliverd(host: string, succeeded: boolean): Promise<void> {
|
|
await this.commit(succeeded ? {
|
|
'deliveredInstances': [host],
|
|
} : {
|
|
'stalled': [host],
|
|
});
|
|
}
|
|
|
|
public async inbox(host: string): Promise<void> {
|
|
await this.commit({
|
|
'inboxInstances': [host],
|
|
});
|
|
}
|
|
}
|