mirror of
https://github.com/twitter/the-algorithm.git
synced 2024-06-13 06:38:52 +02:00
ef4c5eb65e
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.
31 lines
1.1 KiB
Scala
31 lines
1.1 KiB
Scala
package com.twitter.cr_mixer.module
|
|
|
|
import com.google.inject.Provides
|
|
import com.twitter.cr_mixer.model.ModuleNames
|
|
import com.twitter.inject.TwitterModule
|
|
import com.twitter.recos.user_video_graph.thriftscala.ConsumersBasedRelatedTweetRequest
|
|
import com.twitter.recos.user_video_graph.thriftscala.RelatedTweetResponse
|
|
import com.twitter.recos.user_video_graph.thriftscala.UserVideoGraph
|
|
import com.twitter.storehaus.ReadableStore
|
|
import com.twitter.util.Future
|
|
import javax.inject.Named
|
|
import javax.inject.Singleton
|
|
|
|
object ConsumersBasedUserVideoGraphStoreModule extends TwitterModule {
|
|
|
|
@Provides
|
|
@Singleton
|
|
@Named(ModuleNames.ConsumerBasedUserVideoGraphStore)
|
|
def providesConsumerBasedUserVideoGraphStore(
|
|
userVideoGraphService: UserVideoGraph.MethodPerEndpoint
|
|
): ReadableStore[ConsumersBasedRelatedTweetRequest, RelatedTweetResponse] = {
|
|
new ReadableStore[ConsumersBasedRelatedTweetRequest, RelatedTweetResponse] {
|
|
override def get(
|
|
k: ConsumersBasedRelatedTweetRequest
|
|
): Future[Option[RelatedTweetResponse]] = {
|
|
userVideoGraphService.consumersBasedRelatedTweets(k).map(Some(_))
|
|
}
|
|
}
|
|
}
|
|
}
|