net.nemerosa.ontrack.extension.influxdb.metrics.InfluxDBMetricsExportExtension.kt Maven / Gradle / Ivy

Go to download
package net.nemerosa.ontrack.extension.influxdb.metrics

import net.nemerosa.ontrack.common.Time
import net.nemerosa.ontrack.extension.api.MetricsExportExtension
import net.nemerosa.ontrack.extension.influxdb.InfluxDBConnection
import net.nemerosa.ontrack.extension.influxdb.InfluxDBExtensionFeature
import net.nemerosa.ontrack.extension.support.AbstractExtension
import org.influxdb.dto.Point
import java.time.LocalDateTime
import java.util.concurrent.TimeUnit

class InfluxDBMetricsExportExtension(
        extensionFeature: InfluxDBExtensionFeature,
        private val influxDBConnection: InfluxDBConnection
) : AbstractExtension(extensionFeature), MetricsExportExtension {

    override fun exportMetrics(metric: String, tags: Map, fields: Map, timestamp: LocalDateTime?) {
        if (fields.isNotEmpty()) {
            val time = timestamp?.run { Time.toEpochMillis(this) } ?: System.currentTimeMillis()
            influxDBConnection.safe {
                write(
                    Point.measurement(metric)
                        // Tags
                        .tag(tags)
                        // Fields
                        .fields(fields)
                        // OK
                        .time(time, TimeUnit.MILLISECONDS)
                        .build()
                )
                flush()
            }
        }
    }

}