mirror of
https://github.com/twitter/the-algorithm.git
synced 2024-06-26 13:06:05 +02:00
39 lines
1.3 KiB
Scala
39 lines
1.3 KiB
Scala
|
package com.twitter.product_mixer.component_library.module
|
||
|
|
|
||
|
|
import com.twitter.conversions.DurationOps._
|
||
|
|
import com.twitter.explore_ranker.thriftscala.ExploreRanker
|
||
|
|
import com.twitter.finagle.thriftmux.MethodBuilder
|
||
|
|
import com.twitter.finatra.mtls.thriftmux.modules.MtlsClient
|
||
|
|
import com.twitter.inject.annotations.Flags
|
||
|
|
import com.twitter.inject.Injector
|
||
|
|
import com.twitter.inject.thrift.modules.ThriftMethodBuilderClientModule
|
||
|
|
import com.twitter.util.Duration
|
||
|
|
|
||
|
|
object ExploreRankerClientModule
|
||
|
|
extends ThriftMethodBuilderClientModule[
|
||
|
|
ExploreRanker.ServicePerEndpoint,
|
||
|
|
ExploreRanker.MethodPerEndpoint
|
||
|
|
]
|
||
|
|
with MtlsClient {
|
||
|
|
|
||
|
|
override val label: String = "explore-ranker"
|
||
|
|
override val dest: String = "/s/explore-ranker/explore-ranker"
|
||
|
|
|
||
|
|
private final val ExploreRankerTimeoutTotal = "explore_ranker.timeout_total"
|
||
|
|
|
||
|
|
flag[Duration](
|
||
|
|
name = ExploreRankerTimeoutTotal,
|
||
|
|
default = 800.milliseconds,
|
||
|
|
help = "Timeout total for ExploreRanker")
|
||
|
|
|
||
|
|
override protected def configureMethodBuilder(
|
||
|
|
injector: Injector,
|
||
|
|
methodBuilder: MethodBuilder
|
||
|
|
): MethodBuilder = {
|
||
|
|
val timeoutTotal: Duration = injector.instance[Duration](Flags.named(ExploreRankerTimeoutTotal))
|
||
|
|
methodBuilder
|
||
|
|
.withTimeoutTotal(timeoutTotal)
|
||
|
|
.nonIdempotent
|
||
|
|
}
|
||
|
|
}
|