mirror of
https://github.com/twitter/the-algorithm.git
synced 2024-06-13 14:48:54 +02:00
b389c3d302
Pushservice is the main recommendation service we use to surface recommendations to our users via notifications. It fetches candidates from various sources, ranks them in order of relevance, and applies filters to determine the best one to send.
59 lines
1.9 KiB
Scala
59 lines
1.9 KiB
Scala
package com.twitter.frigate.pushservice.store
|
|
|
|
import com.twitter.cr_mixer.thriftscala.CrMixer
|
|
import com.twitter.cr_mixer.thriftscala.CrMixerTweetRequest
|
|
import com.twitter.cr_mixer.thriftscala.CrMixerTweetResponse
|
|
import com.twitter.cr_mixer.thriftscala.FrsTweetRequest
|
|
import com.twitter.cr_mixer.thriftscala.FrsTweetResponse
|
|
import com.twitter.finagle.stats.NullStatsReceiver
|
|
import com.twitter.finagle.stats.Stat
|
|
import com.twitter.finagle.stats.StatsReceiver
|
|
import com.twitter.util.Future
|
|
|
|
/**
|
|
* Store to get content recs from content recommender.
|
|
*/
|
|
case class CrMixerTweetStore(
|
|
crMixer: CrMixer.MethodPerEndpoint
|
|
)(
|
|
implicit statsReceiver: StatsReceiver = NullStatsReceiver) {
|
|
|
|
private val requestsCounter = statsReceiver.counter("requests")
|
|
private val successCounter = statsReceiver.counter("success")
|
|
private val failuresCounter = statsReceiver.counter("failures")
|
|
private val nonEmptyCounter = statsReceiver.counter("non_empty")
|
|
private val emptyCounter = statsReceiver.counter("empty")
|
|
private val failuresScope = statsReceiver.scope("failures")
|
|
private val latencyStat = statsReceiver.stat("latency")
|
|
|
|
private def updateStats[T](f: => Future[Option[T]]): Future[Option[T]] = {
|
|
requestsCounter.incr()
|
|
Stat
|
|
.timeFuture(latencyStat)(f)
|
|
.onSuccess { r =>
|
|
if (r.isDefined) nonEmptyCounter.incr() else emptyCounter.incr()
|
|
successCounter.incr()
|
|
}
|
|
.onFailure { e =>
|
|
{
|
|
failuresCounter.incr()
|
|
failuresScope.counter(e.getClass.getName).incr()
|
|
}
|
|
}
|
|
}
|
|
|
|
def getTweetRecommendations(
|
|
request: CrMixerTweetRequest
|
|
): Future[Option[CrMixerTweetResponse]] = {
|
|
updateStats(crMixer.getTweetRecommendations(request).map { response =>
|
|
Some(response)
|
|
})
|
|
}
|
|
|
|
def getFRSTweetCandidates(request: FrsTweetRequest): Future[Option[FrsTweetResponse]] = {
|
|
updateStats(crMixer.getFrsBasedTweetRecommendations(request).map { response =>
|
|
Some(response)
|
|
})
|
|
}
|
|
}
|