com.infobip.kafkistry.metric.LastRefreshLagCollector.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kafkistry-prometheus-metrics-export Show documentation

The newest version!

package com.infobip.kafkistry.metric

import io.prometheus.client.Collector.MetricFamilySamples
import io.prometheus.client.Collector.Type
import com.infobip.kafkistry.kafkastate.*
import com.infobip.kafkistry.metric.config.PrometheusMetricsProperties
import org.springframework.stereotype.Component

@Component
class LastRefreshLagCollector(
    promProperties: PrometheusMetricsProperties,
    private val providers: List>
) : KafkistryMetricsCollector {

    //default: kafkistry_pool_refresh_lag
    private val metricName = promProperties.prefix + "pool_refresh_lag"

    val labelNames = listOf("cluster", "type")

    override fun expose(context: MetricsDataContext): List {
        val now = System.currentTimeMillis()
        val allSamples = providers.flatMap { provider ->
            val latestStates = provider.getAllLatestStates()
            latestStates.map { (clusterIdentifier, state) ->
                MetricFamilySamples.Sample(
                    metricName, labelNames,
                    listOf(clusterIdentifier, provider.stateTypeName),
                    (now - state.lastRefreshTime).toDouble()
                )
            }
        }
        return mutableListOf(
                MetricFamilySamples(
                    metricName, Type.GAUGE,
                    "How much time (ms) has passed since last pool from cluster",
                    allSamples.toMutableList()
                )
        )
    }

}