mirror of
https://github.com/twitter/the-algorithm.git
synced 2024-06-30 06:56:08 +02:00
39 lines
1.8 KiB
Scala
39 lines
1.8 KiB
Scala
package com.twitter.home_mixer.product.subscribed
|
|
|
|
import com.twitter.home_mixer.model.HomeFeatures.AuthorIdFeature
|
|
import com.twitter.home_mixer.model.HomeFeatures.InReplyToTweetIdFeature
|
|
import com.twitter.home_mixer.model.HomeFeatures.IsRetweetFeature
|
|
import com.twitter.home_mixer.model.HomeFeatures.SourceTweetIdFeature
|
|
import com.twitter.home_mixer.model.HomeFeatures.SourceUserIdFeature
|
|
import com.twitter.product_mixer.core.feature.Feature
|
|
import com.twitter.product_mixer.core.feature.featuremap.FeatureMap
|
|
import com.twitter.product_mixer.core.feature.featuremap.FeatureMapBuilder
|
|
import com.twitter.product_mixer.core.functional_component.transformer.CandidateFeatureTransformer
|
|
import com.twitter.product_mixer.core.model.common.identifier.TransformerIdentifier
|
|
import com.twitter.search.earlybird.{thriftscala => t}
|
|
|
|
object SubscribedEarlybirdResponseFeatureTransformer
|
|
extends CandidateFeatureTransformer[t.ThriftSearchResult] {
|
|
|
|
override val identifier: TransformerIdentifier =
|
|
TransformerIdentifier("SubscribedEarlybirdResponse")
|
|
|
|
override val features: Set[Feature[_, _]] = Set(
|
|
AuthorIdFeature,
|
|
InReplyToTweetIdFeature,
|
|
IsRetweetFeature,
|
|
SourceTweetIdFeature,
|
|
SourceUserIdFeature,
|
|
)
|
|
|
|
override def transform(candidate: t.ThriftSearchResult): FeatureMap = FeatureMapBuilder()
|
|
.add(AuthorIdFeature, candidate.tweetypieTweet.flatMap(_.coreData.map(_.userId)))
|
|
.add(
|
|
InReplyToTweetIdFeature,
|
|
candidate.tweetypieTweet.flatMap(_.coreData.flatMap(_.reply.flatMap(_.inReplyToStatusId))))
|
|
.add(IsRetweetFeature, candidate.metadata.exists(_.isRetweet.contains(true)))
|
|
.add(SourceTweetIdFeature, candidate.sourceTweetypieTweet.map(_.id))
|
|
.add(SourceUserIdFeature, candidate.sourceTweetypieTweet.flatMap(_.coreData.map(_.userId)))
|
|
.build()
|
|
}
|