the-algorithm/product-mixer/component-library/src/main/scala/com/twitter/product_mixer/component_library/gate/QualityFactorGate.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

26 lines
1.1 KiB
Scala

package com.twitter.product_mixer.component_library.gate
import com.twitter.product_mixer.core.functional_component.gate.Gate
import com.twitter.product_mixer.core.model.common.identifier.ComponentIdentifier
import com.twitter.product_mixer.core.model.common.identifier.GateIdentifier
import com.twitter.product_mixer.core.pipeline.PipelineQuery
import com.twitter.product_mixer.core.quality_factor.HasQualityFactorStatus
import com.twitter.stitch.Stitch
/**
* A Gate that only continues if the quality factor value of the pipeline is above the given
* threshold. This is useful for disabling an expensive function when the pipeline is under pressure
* (quality factor is low).
*/
case class QualityFactorGate(pipelineIdentifier: ComponentIdentifier, threshold: Double)
extends Gate[PipelineQuery with HasQualityFactorStatus] {
override val identifier: GateIdentifier = GateIdentifier(
s"${pipelineIdentifier.name}QualityFactor")
override def shouldContinue(
query: PipelineQuery with HasQualityFactorStatus
): Stitch[Boolean] =
Stitch.value(query.getQualityFactorCurrentValue(pipelineIdentifier) >= threshold)
}