Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit af07f53c authored by Jernej Virag's avatar Jernej Virag Committed by Automerger Merge Worker
Browse files

Add ability to trace Flow values as counters am: 4f57fc60 

Original change: https://android-review.googlesource.com/c/platform/frameworks/libs/systemui/+/3323752



Change-Id: Id8fd5596babdea88eb5b76f236905b40e5c101ee
Signed-off-by: default avatarAutomerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com>
parents f6957cd3 4f57fc60

tracinglib/core/common/src/FlowTracing.kt

+14 −0
Original line number Diff line number Diff line
@@ -46,6 +46,20 @@ object FlowTracing { 
        return baseFlow.onEach { stateLogger.log(valueToString(it)) }

    }



    /** Records value of a given numeric flow as a counter track in traces. */

    fun <T : Number> Flow<T>.traceAsCounter(

        counterName: String,

        traceEmissionCount: Boolean = false,

        valueToInt: (T) -> Int = { it.toInt() }

    ): Flow<T> {

        val baseFlow = if (traceEmissionCount) traceEmissionCount(counterName) else this

        return baseFlow.onEach {

            if (isEnabled()) {

                traceCounter(counterName, valueToInt(it))

            }

        }

    }



    /** Adds a counter track to monitor emissions from a specific flow.] */

    fun <T> Flow<T>.traceEmissionCount(flowName: String, uniqueSuffix: Boolean = false): Flow<T> {

        val trackName by lazy {