the-algorithm/product-mixer/component-library/src/main/scala/com/twitter/product_mixer/component_library/premarshaller/urt/builder/IncludeInstruction.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

32 lines
1.2 KiB
Scala

package com.twitter.product_mixer.component_library.premarshaller.urt.builder
import com.twitter.product_mixer.core.model.marshalling.response.urt.TimelineEntry
import com.twitter.product_mixer.core.pipeline.HasPipelineCursor
import com.twitter.product_mixer.core.pipeline.PipelineQuery
trait IncludeInstruction[-Query <: PipelineQuery] { self =>
def apply(query: Query, entries: Seq[TimelineEntry]): Boolean
def inverse(): IncludeInstruction[Query] = new IncludeInstruction[Query] {
def apply(query: Query, entries: Seq[TimelineEntry]): Boolean = !self.apply(query, entries)
}
}
object AlwaysInclude extends IncludeInstruction[PipelineQuery] {
override def apply(query: PipelineQuery, entries: Seq[TimelineEntry]): Boolean = true
}
object IncludeOnFirstPage extends IncludeInstruction[PipelineQuery with HasPipelineCursor[_]] {
override def apply(
query: PipelineQuery with HasPipelineCursor[_],
entries: Seq[TimelineEntry]
): Boolean = query.isFirstPage
}
object IncludeAfterFirstPage extends IncludeInstruction[PipelineQuery with HasPipelineCursor[_]] {
override def apply(
query: PipelineQuery with HasPipelineCursor[_],
entries: Seq[TimelineEntry]
): Boolean = !query.isFirstPage
}