]>
Commit | Line | Data |
---|---|---|
1 | import { isTestInstance } from '../../helpers/core-utils' | |
2 | import { logger } from '../../helpers/logger' | |
3 | import { ActorFollowModel } from '../../models/activitypub/actor-follow' | |
4 | import { AbstractScheduler } from './abstract-scheduler' | |
5 | import { ACTOR_FOLLOW_SCORE, SCHEDULER_INTERVALS_MS } from '../../initializers/constants' | |
6 | import { ActorFollowScoreCache } from '../files-cache' | |
7 | ||
8 | export class ActorFollowScheduler extends AbstractScheduler { | |
9 | ||
10 | private static instance: AbstractScheduler | |
11 | ||
12 | protected schedulerIntervalMs = SCHEDULER_INTERVALS_MS.actorFollowScores | |
13 | ||
14 | private constructor () { | |
15 | super() | |
16 | } | |
17 | ||
18 | protected async internalExecute () { | |
19 | await this.processPendingScores() | |
20 | ||
21 | await this.removeBadActorFollows() | |
22 | } | |
23 | ||
24 | private async processPendingScores () { | |
25 | const pendingScores = ActorFollowScoreCache.Instance.getPendingFollowsScore() | |
26 | const badServerIds = ActorFollowScoreCache.Instance.getBadFollowingServerIds() | |
27 | const goodServerIds = ActorFollowScoreCache.Instance.getGoodFollowingServerIds() | |
28 | ||
29 | ActorFollowScoreCache.Instance.clearPendingFollowsScore() | |
30 | ActorFollowScoreCache.Instance.clearBadFollowingServerIds() | |
31 | ActorFollowScoreCache.Instance.clearGoodFollowingServerIds() | |
32 | ||
33 | for (const inbox of Object.keys(pendingScores)) { | |
34 | await ActorFollowModel.updateScore(inbox, pendingScores[inbox]) | |
35 | } | |
36 | ||
37 | await ActorFollowModel.updateScoreByFollowingServers(badServerIds, ACTOR_FOLLOW_SCORE.PENALTY) | |
38 | await ActorFollowModel.updateScoreByFollowingServers(goodServerIds, ACTOR_FOLLOW_SCORE.BONUS) | |
39 | } | |
40 | ||
41 | private async removeBadActorFollows () { | |
42 | if (!isTestInstance()) logger.info('Removing bad actor follows (scheduler).') | |
43 | ||
44 | try { | |
45 | await ActorFollowModel.removeBadActorFollows() | |
46 | } catch (err) { | |
47 | logger.error('Error in bad actor follows scheduler.', { err }) | |
48 | } | |
49 | } | |
50 | ||
51 | static get Instance () { | |
52 | return this.instance || (this.instance = new this()) | |
53 | } | |
54 | } |