the-algorithm/cr-mixer/server/src/main/scala/com/twitter/cr_mixer/module/RealGraphOonStoreModule.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

40 lines
1.4 KiB
Scala

package com.twitter.cr_mixer.module
import com.google.inject.Provides
import com.twitter.app.Flag
import com.twitter.cr_mixer.model.ModuleNames
import com.twitter.finagle.stats.StatsReceiver
import com.twitter.frigate.common.store.strato.StratoFetchableStore
import com.twitter.hermit.store.common.ObservedReadableStore
import com.twitter.inject.TwitterModule
import com.twitter.simclusters_v2.common.UserId
import com.twitter.storehaus.ReadableStore
import javax.inject.Named
import javax.inject.Singleton
import com.twitter.strato.client.{Client => StratoClient}
import com.twitter.wtf.candidate.thriftscala.CandidateSeq
object RealGraphOonStoreModule extends TwitterModule {
private val userRealGraphOonColumnPath: Flag[String] = flag[String](
name = "crMixer.userRealGraphOonColumnPath",
default = "recommendations/twistly/userRealgraphOon",
help = "Strato column path for user real graph OON Store"
)
@Provides
@Singleton
@Named(ModuleNames.RealGraphOonStore)
def providesRealGraphOonStore(
stratoClient: StratoClient,
statsReceiver: StatsReceiver
): ReadableStore[UserId, CandidateSeq] = {
val realGraphOonStratoFetchableStore = StratoFetchableStore
.withUnitView[UserId, CandidateSeq](stratoClient, userRealGraphOonColumnPath())
ObservedReadableStore(
realGraphOonStratoFetchableStore
)(statsReceiver.scope("user_real_graph_oon_store"))
}
}