123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172 |
- import { ACTOR_FOLLOW_SCORE } from '../../initializers/constants'
- import { logger } from '../../helpers/logger'
- // Cache follows scores, instead of writing them too often in database
- // Keep data in memory, we don't really need Redis here as we don't really care to loose some scores
- class ActorFollowScoreCache {
- private static instance: ActorFollowScoreCache
- private pendingFollowsScore: { [ url: string ]: number } = {}
- private pendingBadServer = new Set<number>()
- private pendingGoodServer = new Set<number>()
- private constructor () {}
- static get Instance () {
- return this.instance || (this.instance = new this())
- }
- updateActorFollowsScore (goodInboxes: string[], badInboxes: string[]) {
- if (goodInboxes.length === 0 && badInboxes.length === 0) return
- logger.info('Updating %d good actor follows and %d bad actor follows scores in cache.', goodInboxes.length, badInboxes.length)
- for (const goodInbox of goodInboxes) {
- if (this.pendingFollowsScore[goodInbox] === undefined) this.pendingFollowsScore[goodInbox] = 0
- this.pendingFollowsScore[goodInbox] += ACTOR_FOLLOW_SCORE.BONUS
- }
- for (const badInbox of badInboxes) {
- if (this.pendingFollowsScore[badInbox] === undefined) this.pendingFollowsScore[badInbox] = 0
- this.pendingFollowsScore[badInbox] += ACTOR_FOLLOW_SCORE.PENALTY
- }
- }
- addBadServerId (serverId: number) {
- this.pendingBadServer.add(serverId)
- }
- getBadFollowingServerIds () {
- return Array.from(this.pendingBadServer)
- }
- clearBadFollowingServerIds () {
- this.pendingBadServer = new Set<number>()
- }
- addGoodServerId (serverId: number) {
- this.pendingGoodServer.add(serverId)
- }
- getGoodFollowingServerIds () {
- return Array.from(this.pendingGoodServer)
- }
- clearGoodFollowingServerIds () {
- this.pendingGoodServer = new Set<number>()
- }
- getPendingFollowsScore () {
- return this.pendingFollowsScore
- }
- clearPendingFollowsScore () {
- this.pendingFollowsScore = {}
- }
- }
- export {
- ActorFollowScoreCache
- }
|