PeerTube/server/lib/schedulers/actor-follow-scheduler.ts

55 lines
1.9 KiB
TypeScript
Raw Normal View History

import { isTestInstance } from '../../helpers/core-utils'
import { logger } from '../../helpers/logger'
2019-08-06 17:19:53 +02:00
import { ACTOR_FOLLOW_SCORE, SCHEDULER_INTERVALS_MS } from '../../initializers/constants'
2021-05-11 11:15:29 +02:00
import { ActorFollowModel } from '../../models/actor/actor-follow'
import { ActorFollowHealthCache } from '../actor-follow-health-cache'
2021-05-11 11:15:29 +02:00
import { AbstractScheduler } from './abstract-scheduler'
export class ActorFollowScheduler extends AbstractScheduler {
private static instance: AbstractScheduler
2021-10-22 10:28:00 +02:00
protected schedulerIntervalMs = SCHEDULER_INTERVALS_MS.ACTOR_FOLLOW_SCORES
private constructor () {
super()
}
protected async internalExecute () {
await this.processPendingScores()
await this.removeBadActorFollows()
}
private async processPendingScores () {
const pendingScores = ActorFollowHealthCache.Instance.getPendingFollowsScore()
const badServerIds = ActorFollowHealthCache.Instance.getBadFollowingServerIds()
const goodServerIds = ActorFollowHealthCache.Instance.getGoodFollowingServerIds()
ActorFollowHealthCache.Instance.clearPendingFollowsScore()
ActorFollowHealthCache.Instance.clearBadFollowingServerIds()
ActorFollowHealthCache.Instance.clearGoodFollowingServerIds()
for (const inbox of Object.keys(pendingScores)) {
2019-08-06 17:19:53 +02:00
await ActorFollowModel.updateScore(inbox, pendingScores[inbox])
}
2019-08-06 17:19:53 +02:00
await ActorFollowModel.updateScoreByFollowingServers(badServerIds, ACTOR_FOLLOW_SCORE.PENALTY)
await ActorFollowModel.updateScoreByFollowingServers(goodServerIds, ACTOR_FOLLOW_SCORE.BONUS)
}
private async removeBadActorFollows () {
if (!isTestInstance()) logger.info('Removing bad actor follows (scheduler).')
2018-02-27 11:08:59 +01:00
try {
await ActorFollowModel.removeBadActorFollows()
} catch (err) {
2018-03-26 15:54:13 +02:00
logger.error('Error in bad actor follows scheduler.', { err })
}
}
static get Instance () {
return this.instance || (this.instance = new this())
}
}