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