mirror of
https://github.com/twitter/the-algorithm.git
synced 2024-06-15 07:38:45 +02:00
ef4c5eb65e
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.
40 lines
1.8 KiB
Scala
40 lines
1.8 KiB
Scala
package com.twitter.product_mixer.component_library.feature_hydrator.query.param_gated
|
|
|
|
import com.twitter.product_mixer.core.feature.Feature
|
|
import com.twitter.product_mixer.core.feature.featuremap.FeatureMap
|
|
import com.twitter.product_mixer.core.functional_component.common.alert.Alert
|
|
import com.twitter.product_mixer.core.functional_component.feature_hydrator.QueryFeatureHydrator
|
|
import com.twitter.product_mixer.core.model.common.Conditionally
|
|
import com.twitter.product_mixer.core.model.common.UniversalNoun
|
|
import com.twitter.product_mixer.core.model.common.identifier.FeatureHydratorIdentifier
|
|
import com.twitter.product_mixer.core.pipeline.PipelineQuery
|
|
import com.twitter.stitch.Stitch
|
|
import com.twitter.timelines.configapi.Param
|
|
|
|
/**
|
|
* A [[QueryFeatureHydrator]] with [[Conditionally]] based on a [[Param]]
|
|
*
|
|
* @param enabledParam the param to turn this [[QueryFeatureHydrator]] on and off
|
|
* @param queryFeatureHydrator the underlying [[QueryFeatureHydrator]] to run when `enabledParam` is true
|
|
* @tparam Query The domain model for the query or request
|
|
* @tparam Result The type of the candidates
|
|
*/
|
|
case class ParamGatedQueryFeatureHydrator[-Query <: PipelineQuery, Result <: UniversalNoun[Any]](
|
|
enabledParam: Param[Boolean],
|
|
queryFeatureHydrator: QueryFeatureHydrator[Query])
|
|
extends QueryFeatureHydrator[Query]
|
|
with Conditionally[Query] {
|
|
|
|
override val identifier: FeatureHydratorIdentifier = FeatureHydratorIdentifier(
|
|
"ParamGated" + queryFeatureHydrator.identifier.name)
|
|
|
|
override val alerts: Seq[Alert] = queryFeatureHydrator.alerts
|
|
|
|
override val features: Set[Feature[_, _]] = queryFeatureHydrator.features
|
|
|
|
override def onlyIf(query: Query): Boolean =
|
|
Conditionally.and(query, queryFeatureHydrator, query.params(enabledParam))
|
|
|
|
override def hydrate(query: Query): Stitch[FeatureMap] = queryFeatureHydrator.hydrate(query)
|
|
}
|