sources/hv-collector-main/src/main/kotlin/org/onap/dcae/collectors/veshv/main/main.kt

   1 /*
   2  * ============LICENSE_START=======================================================
   3  * dcaegen2-collectors-veshv
   4  * ================================================================================
   5  * Copyright (C) 2018-2019 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.main
  21
  22 import arrow.effects.IO
  23 import arrow.effects.fix
  24 import arrow.effects.instances.io.monad.monad
  25 import arrow.typeclasses.binding
  26 import org.onap.dcae.collectors.veshv.commandline.handleWrongArgumentErrorCurried
  27 import org.onap.dcae.collectors.veshv.healthcheck.api.HealthDescription
  28 import org.onap.dcae.collectors.veshv.healthcheck.api.HealthState
  29 import org.onap.dcae.collectors.veshv.main.servers.HealthCheckServer
  30 import org.onap.dcae.collectors.veshv.main.servers.VesServer
  31 import org.onap.dcae.collectors.veshv.model.ServerConfiguration
  32 import org.onap.dcae.collectors.veshv.model.ServiceContext
  33 import org.onap.dcae.collectors.veshv.utils.Closeable
  34 import org.onap.dcae.collectors.veshv.utils.ServerHandle
  35 import org.onap.dcae.collectors.veshv.utils.arrow.ExitFailure
  36 import org.onap.dcae.collectors.veshv.utils.arrow.unsafeRunEitherSync
  37 import org.onap.dcae.collectors.veshv.utils.logging.Logger
  38 import org.onap.dcae.collectors.veshv.utils.registerShutdownHook
  39
  40 private const val VESHV_PACKAGE = "org.onap.dcae.collectors.veshv"
  41 private val logger = Logger("$VESHV_PACKAGE.main")
  42 private const val PROGRAM_NAME = "java $VESHV_PACKAGE.main.MainKt"
  43
  44 fun main(args: Array<String>) =
  45         ArgVesHvConfiguration().parse(args)
  46                 .mapLeft(handleWrongArgumentErrorCurried(PROGRAM_NAME))
  47                 .map(::startAndAwaitServers)
  48                 .unsafeRunEitherSync(
  49                         { ex ->
  50                             logger.withError(ServiceContext::mdc) { log("Failed to start a server", ex) }
  51                             ExitFailure(1)
  52                         },
  53                         { logger.debug(ServiceContext::mdc) { "High Volume VES Collector execution finished" } }
  54                 )
  55
  56 private fun startAndAwaitServers(config: ServerConfiguration) =
  57         IO.monad().binding {
  58             Logger.setLogLevel(VESHV_PACKAGE, config.logLevel)
  59             logger.info { "Using configuration: $config" }
  60
  61             val healthCheckServerHandle = HealthCheckServer.start(config).bind()
  62             val hvVesHandle = VesServer.start(config).bind()
  63
  64             registerShutdownHook(closeServers(hvVesHandle, healthCheckServerHandle))
  65             hvVesHandle.await().bind()
  66         }.fix()
  67
  68 internal fun closeServers(vararg handles: ServerHandle,
  69                           healthState: HealthState = HealthState.INSTANCE) = {
  70     logger.debug(ServiceContext::mdc) { "Graceful shutdown started" }
  71     healthState.changeState(HealthDescription.SHUTTING_DOWN)
  72     Closeable.closeAll(handles.asIterable()).unsafeRunSync()
  73     logger.info(ServiceContext::mdc) { "Graceful shutdown completed" }
  74 }