mirror of
https://github.com/twitter/the-algorithm.git
synced 2024-06-27 21:46:04 +02:00
50 lines
1.9 KiB
Scala
50 lines
1.9 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.timelinemixer.clients.persistence.TimelinePersistenceManhattanClientBuilder
|
|
import com.twitter.timelinemixer.clients.persistence.TimelinePersistenceManhattanClientConfig
|
|
import com.twitter.timelinemixer.clients.persistence.TimelineResponseBatchesClient
|
|
import com.twitter.timelinemixer.clients.persistence.TimelineResponseV3
|
|
import com.twitter.util.Duration
|
|
import javax.inject.Singleton
|
|
|
|
object TimelinesPersistenceStoreClientModule extends TwitterModule {
|
|
private val StagingDataset = "timeline_response_batches_v5_nonprod"
|
|
private val ProdDataset = "timeline_response_batches_v5"
|
|
private final val Timeout = "mh_persistence_store.timeout"
|
|
|
|
flag[Duration](Timeout, 300.millis, "Timeout per request")
|
|
|
|
@Provides
|
|
@Singleton
|
|
def providesTimelinesPersistenceStoreClient(
|
|
@Flag(Timeout) timeout: Duration,
|
|
injectedServiceIdentifier: ServiceIdentifier,
|
|
statsReceiver: StatsReceiver
|
|
): TimelineResponseBatchesClient[TimelineResponseV3] = {
|
|
val timelineResponseBatchesDataset =
|
|
injectedServiceIdentifier.environment.toLowerCase match {
|
|
case "prod" => ProdDataset
|
|
case _ => StagingDataset
|
|
}
|
|
|
|
val timelineResponseBatchesConfig = new TimelinePersistenceManhattanClientConfig {
|
|
val dataset = timelineResponseBatchesDataset
|
|
val isReadOnly = false
|
|
val serviceIdentifier = injectedServiceIdentifier
|
|
override val defaultMaxTimeout = timeout
|
|
override val maxRetryCount = 2
|
|
}
|
|
|
|
TimelinePersistenceManhattanClientBuilder.buildTimelineResponseV3BatchesClient(
|
|
timelineResponseBatchesConfig,
|
|
statsReceiver
|
|
)
|
|
}
|
|
}
|