the-algorithm/home-mixer/server/src/main/scala/com/twitter/home_mixer/util/TweetImpressionsHelper.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

16 lines
630 B
Scala

package com.twitter.home_mixer.util
import com.twitter.home_mixer.model.HomeFeatures.TweetImpressionsFeature
import com.twitter.product_mixer.component_library.feature_hydrator.query.impressed_tweets.ImpressedTweets
import com.twitter.product_mixer.core.feature.featuremap.FeatureMap
object TweetImpressionsHelper {
def tweetImpressions(features: FeatureMap): Set[Long] = {
val manhattanImpressions =
features.getOrElse(TweetImpressionsFeature, Seq.empty).flatMap(_.tweetIds)
val memcacheImpressions = features.getOrElse(ImpressedTweets, Seq.empty)
(manhattanImpressions ++ memcacheImpressions).toSet
}
}