65 lines
2.8 KiB
Scala
65 lines
2.8 KiB
Scala
package com.twitter.home_mixer.functional_component.feature_hydrator
|
|
|
|
import com.twitter.home_mixer.model.HomeFeatures.AuthorIdFeature
|
|
import com.twitter.ml.api.DataRecord
|
|
import com.twitter.product_mixer.component_library.model.candidate.TweetCandidate
|
|
import com.twitter.product_mixer.core.feature.datarecord.DataRecordInAFeature
|
|
import com.twitter.product_mixer.core.feature.Feature
|
|
import com.twitter.product_mixer.core.feature.FeatureWithDefaultOnFailure
|
|
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.feature_hydrator.BulkCandidateFeatureHydrator
|
|
import com.twitter.product_mixer.core.model.common.CandidateWithFeatures
|
|
import com.twitter.product_mixer.core.model.common.identifier.FeatureHydratorIdentifier
|
|
import com.twitter.product_mixer.core.pipeline.PipelineQuery
|
|
import com.twitter.stitch.Stitch
|
|
import com.twitter.timelines.prediction.adapters.realtime_interaction_graph.RealTimeInteractionGraphFeaturesAdapter
|
|
import com.twitter.timelines.prediction.features.realtime_interaction_graph.RealTimeInteractionGraphEdgeFeatures
|
|
import com.twitter.util.Time
|
|
|
|
import javax.inject.Inject
|
|
import javax.inject.Singleton
|
|
import scala.collection.JavaConverters._
|
|
|
|
object RealTimeInteractionGraphEdgeFeature
|
|
extends DataRecordInAFeature[TweetCandidate]
|
|
with FeatureWithDefaultOnFailure[TweetCandidate, DataRecord] {
|
|
override def defaultValue: DataRecord = new DataRecord()
|
|
}
|
|
|
|
@Singleton
|
|
class RealTimeInteractionGraphEdgeFeatureHydrator @Inject() ()
|
|
extends BulkCandidateFeatureHydrator[PipelineQuery, TweetCandidate] {
|
|
|
|
override val identifier: FeatureHydratorIdentifier = FeatureHydratorIdentifier(
|
|
"RealTimeInteractionGraphEdge")
|
|
|
|
override val features: Set[Feature[_, _]] = Set(RealTimeInteractionGraphEdgeFeature)
|
|
|
|
private val realTimeInteractionGraphFeaturesAdapter = new RealTimeInteractionGraphFeaturesAdapter
|
|
|
|
override def apply(
|
|
query: PipelineQuery,
|
|
candidates: Seq[CandidateWithFeatures[TweetCandidate]]
|
|
): Stitch[Seq[FeatureMap]] = {
|
|
val userVertex =
|
|
query.features.flatMap(_.getOrElse(RealTimeInteractionGraphUserVertexQueryFeature, None))
|
|
val realTimeInteractionGraphFeaturesMap =
|
|
userVertex.map(RealTimeInteractionGraphEdgeFeatures(_, Time.now))
|
|
|
|
Stitch.value {
|
|
candidates.map { candidate =>
|
|
val feature = candidate.features.getOrElse(AuthorIdFeature, None).flatMap { authorId =>
|
|
realTimeInteractionGraphFeaturesMap.flatMap(_.get(authorId))
|
|
}
|
|
|
|
FeatureMapBuilder()
|
|
.add(
|
|
RealTimeInteractionGraphEdgeFeature,
|
|
realTimeInteractionGraphFeaturesAdapter.adaptToDataRecords(feature).asScala.head)
|
|
.build()
|
|
}
|
|
}
|
|
}
|
|
}
|