the-algorithm/visibilitylib/src/main/scala/com/twitter/visibility/builder/tweets/BlenderContextFeatures.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

46 lines
1.9 KiB
Scala

package com.twitter.visibility.builder.tweets
import com.twitter.finagle.stats.StatsReceiver
import com.twitter.search.common.constants.thriftscala.ThriftQuerySource
import com.twitter.visibility.builder.FeatureMapBuilder
import com.twitter.visibility.features.SearchCandidateCount
import com.twitter.visibility.features.SearchQueryHasUser
import com.twitter.visibility.features.SearchQuerySource
import com.twitter.visibility.features.SearchResultsPageNumber
import com.twitter.visibility.interfaces.common.blender.BlenderVFRequestContext
@Deprecated
class BlenderContextFeatures(
statsReceiver: StatsReceiver) {
private[this] val scopedStatsReceiver = statsReceiver.scope("blender_context_features")
private[this] val requests = scopedStatsReceiver.counter("requests")
private[this] val searchResultsPageNumber =
scopedStatsReceiver.scope(SearchResultsPageNumber.name).counter("requests")
private[this] val searchCandidateCount =
scopedStatsReceiver.scope(SearchCandidateCount.name).counter("requests")
private[this] val searchQuerySource =
scopedStatsReceiver.scope(SearchQuerySource.name).counter("requests")
private[this] val searchQueryHasUser =
scopedStatsReceiver.scope(SearchQueryHasUser.name).counter("requests")
def forBlenderContext(
blenderContext: BlenderVFRequestContext
): FeatureMapBuilder => FeatureMapBuilder = {
requests.incr()
searchResultsPageNumber.incr()
searchCandidateCount.incr()
searchQuerySource.incr()
searchQueryHasUser.incr()
_.withConstantFeature(SearchResultsPageNumber, blenderContext.resultsPageNumber)
.withConstantFeature(SearchCandidateCount, blenderContext.candidateCount)
.withConstantFeature(
SearchQuerySource,
blenderContext.querySourceOption match {
case Some(querySource) => querySource
case _ => ThriftQuerySource.Unknown
})
.withConstantFeature(SearchQueryHasUser, blenderContext.queryHasUser)
}
}