mirror of
https://github.com/twitter/the-algorithm.git
synced 2024-06-28 14:06:05 +02:00
23 lines
996 B
Scala
23 lines
996 B
Scala
|
package com.twitter.home_mixer.functional_component.gate
|
||
|
|
|
||
|
|
import com.twitter.home_mixer.model.request.DeviceContext.RequestContext
|
||
|
|
import com.twitter.home_mixer.model.request.HasDeviceContext
|
||
|
|
import com.twitter.product_mixer.core.functional_component.gate.Gate
|
||
|
|
import com.twitter.product_mixer.core.model.common.identifier.GateIdentifier
|
||
|
|
import com.twitter.product_mixer.core.pipeline.PipelineQuery
|
||
|
|
import com.twitter.stitch.Stitch
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Gate that fetches the request context from the device context and
|
||
|
|
* continues if the request context matches *any* of the specified ones.
|
||
|
|
*/
|
||
|
|
case class RequestContextGate(requestContexts: Seq[RequestContext.Value])
|
||
|
|
extends Gate[PipelineQuery with HasDeviceContext] {
|
||
|
|
|
||
|
|
override val identifier: GateIdentifier = GateIdentifier("RequestContext")
|
||
|
|
|
||
|
|
override def shouldContinue(query: PipelineQuery with HasDeviceContext): Stitch[Boolean] =
|
||
|
|
Stitch.value(
|
||
|
|
requestContexts.exists(query.deviceContext.flatMap(_.requestContextValue).contains))
|
||
|
|
}
|