mirror of
https://github.com/twitter/the-algorithm.git
synced 2024-06-01 00:38:46 +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.
44 lines
1.4 KiB
Scala
44 lines
1.4 KiB
Scala
package com.twitter.cr_mixer.logging
|
|
|
|
import com.twitter.cr_mixer.featureswitch.CrMixerImpressedBuckets
|
|
import com.twitter.cr_mixer.thriftscala.ImpressesedBucketInfo
|
|
import com.twitter.finagle.stats.StatsReceiver
|
|
import com.twitter.frigate.common.util.StatsUtil
|
|
import com.twitter.logging.Logger
|
|
import com.twitter.scrooge.BinaryThriftStructSerializer
|
|
import com.twitter.scrooge.ThriftStruct
|
|
import com.twitter.scrooge.ThriftStructCodec
|
|
|
|
object ScribeLoggerUtils {
|
|
|
|
/**
|
|
* Handles base64-encoding, serialization, and publish.
|
|
*/
|
|
private[logging] def publish[T <: ThriftStruct](
|
|
logger: Logger,
|
|
codec: ThriftStructCodec[T],
|
|
message: T
|
|
): Unit = {
|
|
logger.info(BinaryThriftStructSerializer(codec).toString(message))
|
|
}
|
|
|
|
private[logging] def getImpressedBuckets(
|
|
scopedStats: StatsReceiver
|
|
): Option[List[ImpressesedBucketInfo]] = {
|
|
StatsUtil.trackNonFutureBlockStats(scopedStats.scope("getImpressedBuckets")) {
|
|
CrMixerImpressedBuckets.getAllImpressedBuckets.map { listBuckets =>
|
|
val listBucketsSet = listBuckets.toSet
|
|
scopedStats.stat("impressed_buckets").add(listBucketsSet.size)
|
|
listBucketsSet.map { bucket =>
|
|
ImpressesedBucketInfo(
|
|
experimentId = bucket.experiment.settings.experimentId.getOrElse(-1L),
|
|
bucketName = bucket.name,
|
|
version = bucket.experiment.settings.version,
|
|
)
|
|
}.toList
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|