the-algorithm/follow-recommendations-service/common/src/main/scala/com/twitter/follow_recommendations/common/candidate_sources/sims/SimsStore.scala
twitter-team ef4c5eb65e Twitter Recommendation Algorithm
Please note we have force-pushed a new initial commit in order to remove some publicly-available Twitter user information. Note that this process may be required in the future.
2023-03-31 17:36:31 -05:00

37 lines
1.3 KiB
Scala

package com.twitter.follow_recommendations.common.candidate_sources.sims
import com.google.inject.Singleton
import com.google.inject.name.Named
import com.twitter.finagle.stats.StatsReceiver
import com.twitter.follow_recommendations.common.constants.GuiceNamedConstants
import com.twitter.hermit.candidate.thriftscala.Candidates
import com.twitter.hermit.model.Algorithm
import com.twitter.product_mixer.core.model.common.identifier.CandidateSourceIdentifier
import com.twitter.strato.client.Fetcher
import com.twitter.util.Duration
import javax.inject.Inject
@Singleton
class SimsStore @Inject() (
@Named(GuiceNamedConstants.SIMS_FETCHER) fetcher: Fetcher[Long, Unit, Candidates])
extends StratoBasedSimsCandidateSourceWithUnitView(fetcher, identifier = SimsStore.Identifier)
@Singleton
class CachedSimsStore @Inject() (
@Named(GuiceNamedConstants.SIMS_FETCHER) fetcher: Fetcher[Long, Unit, Candidates],
statsReceiver: StatsReceiver)
extends CacheBasedSimsStore(
id = SimsStore.Identifier,
fetcher = fetcher,
maxCacheSize = SimsStore.MaxCacheSize,
cacheTtl = SimsStore.CacheTTL,
statsReceiver = statsReceiver.scope("CachedSimsStore", "cache")
)
object SimsStore {
val Identifier = CandidateSourceIdentifier(Algorithm.Sims.toString)
val MaxCacheSize = 50000
val CacheTTL: Duration = Duration.fromHours(24)
}