mirror of
https://github.com/twitter/the-algorithm.git
synced 2024-06-14 07:08:53 +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.
42 lines
1.9 KiB
Scala
42 lines
1.9 KiB
Scala
package com.twitter.product_mixer.component_library.decorator.urt.builder.item.topic
|
|
|
|
import com.twitter.product_mixer.component_library.model.candidate.TopicCandidate
|
|
import com.twitter.product_mixer.core.feature.featuremap.FeatureMap
|
|
import com.twitter.product_mixer.core.pipeline.PipelineQuery
|
|
import com.twitter.timelines.configapi.FSEnumParam
|
|
import com.twitter.product_mixer.core.model.marshalling.response.urt.item.topic.BasicTopicDisplayType
|
|
import com.twitter.product_mixer.core.model.marshalling.response.urt.item.topic.PillTopicDisplayType
|
|
import com.twitter.product_mixer.core.model.marshalling.response.urt.item.topic.NoIconTopicDisplayType
|
|
import com.twitter.product_mixer.core.model.marshalling.response.urt.item.topic.PillWithoutActionIconDisplayType
|
|
import com.twitter.product_mixer.core.model.marshalling.response.urt.item.topic.TopicDisplayType
|
|
import com.twitter.product_mixer.core.functional_component.decorator.urt.builder.item.topic.BaseTopicDisplayTypeBuilder
|
|
|
|
object TopicCandidateDisplayType extends Enumeration {
|
|
type TopicDisplayType = Value
|
|
|
|
val Basic = Value
|
|
val Pill = Value
|
|
val NoIcon = Value
|
|
val PillWithoutActionIcon = Value
|
|
}
|
|
|
|
case class ParamTopicDisplayTypeBuilder(
|
|
displayTypeParam: FSEnumParam[TopicCandidateDisplayType.type])
|
|
extends BaseTopicDisplayTypeBuilder[PipelineQuery, TopicCandidate] {
|
|
|
|
override def apply(
|
|
query: PipelineQuery,
|
|
candidate: TopicCandidate,
|
|
candidateFeatures: FeatureMap
|
|
): Option[TopicDisplayType] = {
|
|
val displayType = query.params(displayTypeParam)
|
|
displayType match {
|
|
case TopicCandidateDisplayType.Basic => Some(BasicTopicDisplayType)
|
|
case TopicCandidateDisplayType.Pill => Some(PillTopicDisplayType)
|
|
case TopicCandidateDisplayType.NoIcon =>
|
|
Some(NoIconTopicDisplayType)
|
|
case TopicCandidateDisplayType.PillWithoutActionIcon => Some(PillWithoutActionIconDisplayType)
|
|
}
|
|
}
|
|
}
|