mirror of
https://github.com/twitter/the-algorithm.git
synced 2024-06-28 14:06:05 +02:00
59 lines
2.2 KiB
Scala
59 lines
2.2 KiB
Scala
package com.twitter.home_mixer.module
|
|
|
|
import com.google.inject.Provides
|
|
import com.twitter.conversions.DurationOps._
|
|
import com.twitter.finagle.mtls.authentication.ServiceIdentifier
|
|
import com.twitter.finagle.stats.StatsReceiver
|
|
import com.twitter.inject.TwitterModule
|
|
import com.twitter.inject.annotations.Flag
|
|
import com.twitter.storage.client.manhattan.kv.Guarantee
|
|
import com.twitter.storehaus_internal.manhattan.ManhattanClusters
|
|
import com.twitter.timelines.clients.manhattan.mhv3.ManhattanClientBuilder
|
|
import com.twitter.timelines.impressionstore.store.ManhattanTweetImpressionStoreClientConfig
|
|
import com.twitter.timelines.impressionstore.store.ManhattanTweetImpressionStoreClient
|
|
import com.twitter.util.Duration
|
|
import javax.inject.Singleton
|
|
|
|
object ManhattanTweetImpressionStoreModule extends TwitterModule {
|
|
|
|
private val ProdAppId = "timelines_tweet_impression_store_v2"
|
|
private val ProdDataset = "timelines_tweet_impressions_v2"
|
|
private val StagingAppId = "timelines_tweet_impression_store_staging"
|
|
private val StagingDataset = "timelines_tweet_impressions_staging"
|
|
private val StatsScope = "manhattanTweetImpressionStoreClient"
|
|
private val DefaultTTL = 2.days
|
|
private final val Timeout = "mh_impression_store.timeout"
|
|
|
|
flag[Duration](Timeout, 150.millis, "Timeout per request")
|
|
|
|
@Provides
|
|
@Singleton
|
|
def providesManhattanTweetImpressionStoreClient(
|
|
@Flag(Timeout) timeout: Duration,
|
|
serviceIdentifier: ServiceIdentifier,
|
|
statsReceiver: StatsReceiver
|
|
): ManhattanTweetImpressionStoreClient = {
|
|
|
|
val (appId, dataset) = serviceIdentifier.environment.toLowerCase match {
|
|
case "prod" => (ProdAppId, ProdDataset)
|
|
case _ => (StagingAppId, StagingDataset)
|
|
}
|
|
|
|
val config = ManhattanTweetImpressionStoreClientConfig(
|
|
cluster = ManhattanClusters.nash,
|
|
appId = appId,
|
|
dataset = dataset,
|
|
statsScope = StatsScope,
|
|
defaultGuarantee = Guarantee.SoftDcReadMyWrites,
|
|
defaultMaxTimeout = timeout,
|
|
maxRetryCount = 2,
|
|
isReadOnly = false,
|
|
serviceIdentifier = serviceIdentifier,
|
|
ttl = DefaultTTL
|
|
)
|
|
|
|
val manhattanEndpoint = ManhattanClientBuilder.buildManhattanEndpoint(config, statsReceiver)
|
|
ManhattanTweetImpressionStoreClient(config, manhattanEndpoint, statsReceiver)
|
|
}
|
|
}
|