the-algorithm/follow-recommendations-service/common/src/main/scala/com/twitter/follow_recommendations/common/transforms/recommendation_flow_identifier/AddRecommendationFlowIdentifierTransform.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

21 lines
800 B
Scala

package com.twitter.follow_recommendations.common.transforms.recommendation_flow_identifier
import com.google.inject.Inject
import com.twitter.follow_recommendations.common.base.Transform
import com.twitter.follow_recommendations.common.models.CandidateUser
import com.twitter.follow_recommendations.common.models.HasRecommendationFlowIdentifier
import com.twitter.stitch.Stitch
class AddRecommendationFlowIdentifierTransform @Inject()
extends Transform[HasRecommendationFlowIdentifier, CandidateUser] {
override def transform(
target: HasRecommendationFlowIdentifier,
items: Seq[CandidateUser]
): Stitch[Seq[CandidateUser]] = {
Stitch.value(items.map { candidateUser =>
candidateUser.copy(recommendationFlowIdentifier = target.recommendationFlowIdentifier)
})
}
}