mirror of
https://github.com/twitter/the-algorithm.git
synced 2024-06-15 07:38:45 +02:00
ef4c5eb65e
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.
56 lines
2.3 KiB
Scala
56 lines
2.3 KiB
Scala
package com.twitter.follow_recommendations.common.candidate_sources.sims
|
|
|
|
import com.twitter.finagle.stats.NullStatsReceiver
|
|
import com.twitter.finagle.stats.StatsReceiver
|
|
import com.twitter.follow_recommendations.common.models.CandidateUser
|
|
import com.twitter.follow_recommendations.common.models.HasSimilarToContext
|
|
import com.twitter.hermit.model.Algorithm
|
|
import com.twitter.product_mixer.core.functional_component.candidate_source.CandidateSource
|
|
import com.twitter.product_mixer.core.model.common.identifier.CandidateSourceIdentifier
|
|
import com.twitter.stitch.Stitch
|
|
import com.twitter.timelines.configapi.HasParams
|
|
|
|
import javax.inject.Inject
|
|
import javax.inject.Singleton
|
|
|
|
@Singleton
|
|
class SwitchingSimsSource @Inject() (
|
|
cachedDBV2SimsStore: CachedDBV2SimsStore,
|
|
cachedDBV2SimsRefreshStore: CachedDBV2SimsRefreshStore,
|
|
cachedSimsExperimentalStore: CachedSimsExperimentalStore,
|
|
cachedSimsStore: CachedSimsStore,
|
|
statsReceiver: StatsReceiver = NullStatsReceiver)
|
|
extends CandidateSource[HasParams with HasSimilarToContext, CandidateUser] {
|
|
|
|
override val identifier: CandidateSourceIdentifier = SwitchingSimsSource.Identifier
|
|
|
|
private val stats = statsReceiver.scope("SwitchingSimsSource")
|
|
private val dbV2SimsStoreCounter = stats.counter("DBV2SimsStore")
|
|
private val dbV2SimsRefreshStoreCounter = stats.counter("DBV2SimsRefreshStore")
|
|
private val simsExperimentalStoreCounter = stats.counter("SimsExperimentalStore")
|
|
private val simsStoreCounter = stats.counter("SimsStore")
|
|
|
|
override def apply(request: HasParams with HasSimilarToContext): Stitch[Seq[CandidateUser]] = {
|
|
val selectedSimsStore =
|
|
if (request.params(SimsSourceParams.EnableDBV2SimsStore)) {
|
|
dbV2SimsStoreCounter.incr()
|
|
cachedDBV2SimsStore
|
|
} else if (request.params(SimsSourceParams.EnableDBV2SimsRefreshStore)) {
|
|
dbV2SimsRefreshStoreCounter.incr()
|
|
cachedDBV2SimsRefreshStore
|
|
} else if (request.params(SimsSourceParams.EnableExperimentalSimsStore)) {
|
|
simsExperimentalStoreCounter.incr()
|
|
cachedSimsExperimentalStore
|
|
} else {
|
|
simsStoreCounter.incr()
|
|
cachedSimsStore
|
|
}
|
|
stats.counter("total").incr()
|
|
selectedSimsStore(request)
|
|
}
|
|
}
|
|
|
|
object SwitchingSimsSource {
|
|
val Identifier: CandidateSourceIdentifier = CandidateSourceIdentifier(Algorithm.Sims.toString)
|
|
}
|