the-algorithm/home-mixer/server/src/main/scala/com/twitter/home_mixer/module/ConversationServiceModule.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

38 lines
1.3 KiB
Scala

package com.twitter.home_mixer.module
import com.twitter.conversions.DurationOps._
import com.twitter.finagle.ThriftMux
import com.twitter.finagle.thriftmux.MethodBuilder
import com.twitter.finatra.mtls.thriftmux.modules.MtlsClient
import com.twitter.inject.Injector
import com.twitter.inject.thrift.modules.ThriftMethodBuilderClientModule
import com.twitter.tweetconvosvc.thriftscala.ConversationService
import com.twitter.util.Duration
import org.apache.thrift.protocol.TCompactProtocol
object ConversationServiceModule
extends ThriftMethodBuilderClientModule[
ConversationService.ServicePerEndpoint,
ConversationService.MethodPerEndpoint
]
with MtlsClient {
override val label: String = "tweetconvosvc"
override val dest: String = "/s/tweetconvosvc/tweetconvosvc"
override protected def configureMethodBuilder(
injector: Injector,
methodBuilder: MethodBuilder
): MethodBuilder = methodBuilder.withTimeoutPerRequest(100.milliseconds)
override def configureThriftMuxClient(
injector: Injector,
client: ThriftMux.Client
): ThriftMux.Client =
super
.configureThriftMuxClient(injector, client)
.withProtocolFactory(new TCompactProtocol.Factory())
override protected def sessionAcquisitionTimeout: Duration = 500.milliseconds
}