sources/hv-collector-kafka-consumer/src/main/kotlin/org/onap/dcae/collectors/veshv/kafkaconsumer/metrics/http/PrometheusApiServer.kt

   1 /*
   2  * ============LICENSE_START=======================================================
   3  * dcaegen2-collectors-veshv
   4  * ================================================================================
   5  * Copyright (C) 2019-2020 NOKIA
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20 package org.onap.dcae.collectors.veshv.kafkaconsumer.metrics.http
  21
  22 import org.onap.dcae.collectors.veshv.kafkaconsumer.metrics.MicrometerMetrics
  23 import org.onap.dcae.collectors.veshv.utils.NettyServerHandle
  24 import org.onap.dcae.collectors.veshv.utils.ServerHandle
  25 import org.onap.dcae.collectors.veshv.utils.logging.Logger
  26 import reactor.core.publisher.Mono
  27 import reactor.netty.http.server.HttpServer
  28 import reactor.netty.http.server.HttpServerRequest
  29 import reactor.netty.http.server.HttpServerResponse
  30 import java.net.InetSocketAddress
  31
  32 internal class PrometheusApiServer(private val listenAddress: InetSocketAddress,
  33                                    private val metrics: MicrometerMetrics) {
  34
  35     private val logger = Logger(PrometheusApiServer::class)
  36
  37     fun start(): Mono<NettyServerHandle> =
  38             HttpServer.create()
  39                     .tcpConfiguration { it.bindAddress { listenAddress } }
  40                     .route { it.get("/monitoring/prometheus", ::metricsHandler) }
  41                     .bind()
  42                     .map { NettyServerHandle(it) }
  43                     .doOnSuccess(::logServerStarted)
  44
  45
  46     private fun metricsHandler(_req: HttpServerRequest, resp: HttpServerResponse) =
  47             resp.sendString(metrics.lastStatus())
  48
  49
  50     private fun logServerStarted(handle: ServerHandle) =
  51             logger.info {
  52                 "Kafka Consumer API server is up and listening on ${handle.host}:${handle.port}"
  53             }
  54 }