com.infobip.kafkistry.metric.config.PrometheusConfig.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.config

import io.micrometer.core.instrument.Clock
import io.micrometer.prometheusmetrics.PrometheusConfig
import io.micrometer.prometheusmetrics.PrometheusMeterRegistry
import io.prometheus.client.Collector
import io.prometheus.client.hotspot.DefaultExports
import io.prometheus.metrics.model.registry.PrometheusRegistry
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty
import org.springframework.context.annotation.Bean
import org.springframework.context.annotation.Configuration
import java.util.*
import java.util.concurrent.atomic.AtomicBoolean

@Configuration
@ConditionalOnProperty(APP_METRICS_ENDPOINT_ENABLED_PROPERTY, matchIfMissing = true)
class PrometheusConfigs(
    collectorBeans: Optional>,
    prometheusProperties: PrometheusMetricsProperties,
) {

    companion object {
        private var initialized = AtomicBoolean(false)

        //static instance which is re-used when separate ApplicationContext-s spin up within same JVM
        // to avoid duplicate meters registrations into same prometheus CollectorRegistry.defaultRegistry
        private val prometheusRegistry = PrometheusMeterRegistry(
            PrometheusConfig.DEFAULT, PrometheusRegistry.defaultRegistry, Clock.SYSTEM
        )
    }

    init {
        if (initialized.compareAndSet(false, true)) {
            if (prometheusProperties.defaultMetrics) {
                DefaultExports.initialize()
            }
            collectorBeans.ifPresent { collectors ->
                collectors.forEach { it.register() }
            }
        }
    }

    @Bean
    fun prometheusRegistry(): PrometheusMeterRegistry = prometheusRegistry

}