Init commit

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

@@ -0,0 +1,60 @@
+import { isProdInstance, isTestOrDevInstance } from '@peertube/peertube-node-utils'
+import { logger, loggerTagsFactory } from '../../helpers/logger.js'
+import { ACTOR_FOLLOW_SCORE, SCHEDULER_INTERVALS_MS } from '../../initializers/constants.js'
+import { ActorFollowModel } from '../../models/actor/actor-follow.js'
+import { ActorFollowHealthCache } from '../actor-follow-health-cache.js'
+import { AbstractScheduler } from './abstract-scheduler.js'
+
+const lTags = loggerTagsFactory('schedulers')
+
+export class ActorFollowScheduler extends AbstractScheduler {
+  private static instance: AbstractScheduler
+
+  protected schedulerIntervalMs = SCHEDULER_INTERVALS_MS.ACTOR_FOLLOW_SCORES
+
+  private constructor () {
+    super({ randomRunOnEnable: false })
+  }
+
+  protected async internalExecute () {
+    // Run too often in test/dev instances
+    if (isProdInstance()) {
+      logger.info('Processing pending actor follow scores.', lTags())
+    }
+
+    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)) {
+      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 (!isTestOrDevInstance()) logger.info('Removing bad actor follows (scheduler).', lTags())
+
+    try {
+      await ActorFollowModel.removeBadActorFollows()
+    } catch (err) {
+      logger.error('Error in bad actor follows scheduler.', { err, ...lTags() })
+    }
+  }
+
+  static get Instance () {
+    return this.instance || (this.instance = new this())
+  }
+}