-import { isTestInstance } from '../../helpers/core-utils'
+import { isTestOrDevInstance } from '../../helpers/core-utils'
import { logger } from '../../helpers/logger'
-import { ActorFollowModel } from '../../models/activitypub/actor-follow'
+import { ACTOR_FOLLOW_SCORE, SCHEDULER_INTERVALS_MS } from '../../initializers/constants'
+import { ActorFollowModel } from '../../models/actor/actor-follow'
+import { ActorFollowHealthCache } from '../actor-follow-health-cache'
import { AbstractScheduler } from './abstract-scheduler'
-import { SCHEDULER_INTERVALS_MS } from '../../initializers'
-import { ActorFollowScoreCache } from '../cache'
export class ActorFollowScheduler extends AbstractScheduler {
private static instance: AbstractScheduler
- protected schedulerIntervalMs = SCHEDULER_INTERVALS_MS.actorFollowScores
+ protected schedulerIntervalMs = SCHEDULER_INTERVALS_MS.ACTOR_FOLLOW_SCORES
private constructor () {
super()
}
private async processPendingScores () {
- const pendingScores = ActorFollowScoreCache.Instance.getPendingFollowsScoreCopy()
+ const pendingScores = ActorFollowHealthCache.Instance.getPendingFollowsScore()
+ const badServerIds = ActorFollowHealthCache.Instance.getBadFollowingServerIds()
+ const goodServerIds = ActorFollowHealthCache.Instance.getGoodFollowingServerIds()
- ActorFollowScoreCache.Instance.clearPendingFollowsScore()
+ ActorFollowHealthCache.Instance.clearPendingFollowsScore()
+ ActorFollowHealthCache.Instance.clearBadFollowingServerIds()
+ ActorFollowHealthCache.Instance.clearGoodFollowingServerIds()
for (const inbox of Object.keys(pendingScores)) {
- await ActorFollowModel.updateFollowScore(inbox, pendingScores[inbox])
+ await ActorFollowModel.updateScore(inbox, pendingScores[inbox])
}
+
+ 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).')
+ if (!isTestOrDevInstance()) logger.info('Removing bad actor follows (scheduler).')
try {
await ActorFollowModel.removeBadActorFollows()