mirror of
https://github.com/twitter/the-algorithm.git
synced 2024-06-13 06:38:52 +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.
56 lines
2.1 KiB
Scala
56 lines
2.1 KiB
Scala
package com.twitter.frigate.pushservice.send_handler.generator
|
|
|
|
import com.twitter.frigate.common.base.ScheduledSpaceSubscriberCandidate
|
|
import com.twitter.frigate.pushservice.model.PushTypes.RawCandidate
|
|
import com.twitter.frigate.pushservice.model.PushTypes.Target
|
|
import com.twitter.frigate.thriftscala.CommonRecommendationType
|
|
import com.twitter.frigate.thriftscala.FrigateNotification
|
|
import com.twitter.util.Future
|
|
|
|
object ScheduledSpaceSubscriberCandidateGenerator extends CandidateGenerator {
|
|
|
|
override def getCandidate(
|
|
targetUser: Target,
|
|
notification: FrigateNotification
|
|
): Future[RawCandidate] = {
|
|
|
|
/**
|
|
* frigateNotification recommendation type should be [[CommonRecommendationType.ScheduledSpaceSubscriber]]
|
|
*
|
|
**/
|
|
require(
|
|
notification.commonRecommendationType == CommonRecommendationType.ScheduledSpaceSubscriber,
|
|
"ScheduledSpaceSubscriber: unexpected CRT " + notification.commonRecommendationType
|
|
)
|
|
|
|
val spaceNotification = notification.spaceNotification.getOrElse(
|
|
throw new IllegalStateException("ScheduledSpaceSubscriber notification object not defined"))
|
|
|
|
require(
|
|
spaceNotification.hostUserId.isDefined,
|
|
"ScheduledSpaceSubscriber notification - hostUserId not defined"
|
|
)
|
|
|
|
val spaceHostId = spaceNotification.hostUserId
|
|
|
|
require(
|
|
spaceNotification.scheduledStartTime.isDefined,
|
|
"ScheduledSpaceSubscriber notification - scheduledStartTime not defined"
|
|
)
|
|
|
|
val scheduledStartTime = spaceNotification.scheduledStartTime.get
|
|
|
|
val candidate = new RawCandidate with ScheduledSpaceSubscriberCandidate {
|
|
override val target: Target = targetUser
|
|
override val frigateNotification: FrigateNotification = notification
|
|
override val spaceId: String = spaceNotification.broadcastId
|
|
override val hostId: Option[Long] = spaceHostId
|
|
override val startTime: Long = scheduledStartTime
|
|
override val speakerIds: Option[Seq[Long]] = spaceNotification.speakers
|
|
override val listenerIds: Option[Seq[Long]] = spaceNotification.listeners
|
|
}
|
|
|
|
Future.value(candidate)
|
|
}
|
|
}
|