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.
50 lines
1.9 KiB
Scala
50 lines
1.9 KiB
Scala
package com.twitter.product_mixer.component_library.candidate_source.audiospace
|
|
|
|
import com.twitter.periscope.audio_space.thriftscala.CreatedSpacesView
|
|
import com.twitter.periscope.audio_space.thriftscala.SpaceSlice
|
|
import com.twitter.product_mixer.component_library.model.cursor.NextCursorFeature
|
|
import com.twitter.product_mixer.component_library.model.cursor.PreviousCursorFeature
|
|
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.candidate_source.strato.StratoKeyViewFetcherWithSourceFeaturesSource
|
|
import com.twitter.product_mixer.core.model.common.identifier.CandidateSourceIdentifier
|
|
import com.twitter.strato.client.Fetcher
|
|
import com.twitter.strato.generated.client.periscope.CreatedSpacesSliceOnUserClientColumn
|
|
import javax.inject.Inject
|
|
import javax.inject.Singleton
|
|
|
|
@Singleton
|
|
class CreatedSpacesCandidateSource @Inject() (
|
|
column: CreatedSpacesSliceOnUserClientColumn)
|
|
extends StratoKeyViewFetcherWithSourceFeaturesSource[
|
|
Long,
|
|
CreatedSpacesView,
|
|
SpaceSlice,
|
|
String
|
|
] {
|
|
|
|
override val identifier: CandidateSourceIdentifier = CandidateSourceIdentifier("CreatedSpaces")
|
|
|
|
override val fetcher: Fetcher[Long, CreatedSpacesView, SpaceSlice] = column.fetcher
|
|
|
|
override def stratoResultTransformer(
|
|
stratoKey: Long,
|
|
stratoResult: SpaceSlice
|
|
): Seq[String] =
|
|
stratoResult.items
|
|
|
|
override protected def extractFeaturesFromStratoResult(
|
|
stratoKey: Long,
|
|
stratoResult: SpaceSlice
|
|
): FeatureMap = {
|
|
val featureMapBuilder = FeatureMapBuilder()
|
|
stratoResult.sliceInfo.previousCursor.foreach { cursor =>
|
|
featureMapBuilder.add(PreviousCursorFeature, cursor)
|
|
}
|
|
stratoResult.sliceInfo.nextCursor.foreach { cursor =>
|
|
featureMapBuilder.add(NextCursorFeature, cursor)
|
|
}
|
|
featureMapBuilder.build()
|
|
}
|
|
}
|