the-algorithm/src/scala/com/twitter/simclusters_v2/hdfs_sources/injections/MultiTypeGraphInjections.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

32 lines
1.4 KiB
Scala

package com.twitter.simclusters_v2.hdfs_sources.injections
import com.twitter.scalding_internal.multiformat.format.keyval.KeyValInjection
import com.twitter.scalding_internal.multiformat.format.keyval.KeyValInjection.ScalaCompactThrift
import com.twitter.simclusters_v2.thriftscala.LeftNode
import com.twitter.simclusters_v2.thriftscala.NounWithFrequencyList
import com.twitter.simclusters_v2.thriftscala.RightNode
import com.twitter.simclusters_v2.thriftscala.RightNodeTypeStruct
import com.twitter.simclusters_v2.thriftscala.RightNodeWithEdgeWeightList
import com.twitter.simclusters_v2.thriftscala.SimilarRightNodes
import com.twitter.simclusters_v2.thriftscala.CandidateTweetsList
import com.twitter.scalding_internal.multiformat.format.keyval.KeyValInjection.Long2BigEndian
object MultiTypeGraphInjections {
final val truncatedMultiTypeGraphInjection =
KeyValInjection(ScalaCompactThrift(LeftNode), ScalaCompactThrift(RightNodeWithEdgeWeightList))
final val topKRightNounListInjection =
KeyValInjection(
ScalaCompactThrift(RightNodeTypeStruct),
ScalaCompactThrift(NounWithFrequencyList))
final val similarRightNodesInjection =
KeyValInjection[RightNode, SimilarRightNodes](
ScalaCompactThrift(RightNode),
ScalaCompactThrift(SimilarRightNodes)
)
final val tweetRecommendationsInjection =
KeyValInjection[Long, CandidateTweetsList](
Long2BigEndian,
ScalaCompactThrift(CandidateTweetsList)
)
}