mirror of
https://github.com/twitter/the-algorithm.git
synced 2025-01-02 23:51:53 +01:00
Compare commits
2 Commits
a6bfec92bb
...
6c87fb6ffc
Author | SHA1 | Date | |
---|---|---|---|
|
6c87fb6ffc | ||
|
f51d763c5e |
@ -4,41 +4,28 @@ import com.twitter.discovery.common.stats.DiscoveryStatsFilter
|
||||
import com.twitter.finagle.Service
|
||||
import com.twitter.finagle.stats.StatsReceiver
|
||||
import com.twitter.finatra.thrift.Controller
|
||||
import com.twitter.graph_feature_service.server.handlers.ServerGetIntersectionHandler.GetIntersectionRequest
|
||||
import com.twitter.graph_feature_service.server.handlers.ServerGetIntersectionHandler
|
||||
import com.twitter.graph_feature_service.thriftscala
|
||||
import com.twitter.graph_feature_service.thriftscala.Server.GetIntersection
|
||||
import com.twitter.graph_feature_service.thriftscala.Server.GetPresetIntersection
|
||||
import com.twitter.graph_feature_service.server.handlers.ServerGetIntersectionHandler.{fromGfsIntersectionRequest, fromGfsPresetIntersectionRequest}
|
||||
import com.twitter.graph_feature_service.thriftscala._
|
||||
import javax.inject.Inject
|
||||
import javax.inject.Singleton
|
||||
import javax.inject.{Inject, Singleton}
|
||||
|
||||
@Singleton
|
||||
class ServerController @Inject() (
|
||||
serverGetIntersectionHandler: ServerGetIntersectionHandler
|
||||
)(
|
||||
implicit statsReceiver: StatsReceiver)
|
||||
extends Controller(thriftscala.Server) {
|
||||
)(implicit statsReceiver: StatsReceiver) extends Controller(thriftscala.Server) {
|
||||
|
||||
private val getIntersectionService: Service[GetIntersectionRequest, GfsIntersectionResponse] =
|
||||
new DiscoveryStatsFilter(statsReceiver.scope("srv").scope("get_intersection"))
|
||||
.andThen(Service.mk(serverGetIntersectionHandler))
|
||||
|
||||
val getIntersection: Service[GetIntersection.Args, GfsIntersectionResponse] = { args =>
|
||||
// TODO: Disable updateCache after HTL switch to use PresetIntersection endpoint.
|
||||
getIntersectionService(
|
||||
GetIntersectionRequest.fromGfsIntersectionRequest(args.request, cacheable = true))
|
||||
getIntersectionService(fromGfsIntersectionRequest(args.request, cacheable = true))
|
||||
}
|
||||
|
||||
handle(GetIntersection) { getIntersection }
|
||||
|
||||
def getPresetIntersection: Service[
|
||||
GetPresetIntersection.Args,
|
||||
GfsIntersectionResponse
|
||||
] = { args =>
|
||||
// TODO: Refactor after HTL switch to PresetIntersection
|
||||
val cacheable = args.request.presetFeatureTypes == PresetFeatureTypes.HtlTwoHop
|
||||
getIntersectionService(
|
||||
GetIntersectionRequest.fromGfsPresetIntersectionRequest(args.request, cacheable))
|
||||
def getPresetIntersection: Service[GetPresetIntersection.Args, GfsIntersectionResponse] = { args =>
|
||||
getIntersectionService(fromGfsPresetIntersectionRequest(args.request, cacheable = args.request.presetFeatureTypes == PresetFeatureTypes.HtlTwoHop))
|
||||
}
|
||||
|
||||
handle(GetPresetIntersection) { getPresetIntersection }
|
||||
|
Loading…
Reference in New Issue
Block a user