Handle sigterm signal
[dcaegen2/collectors/hv-ves.git] / 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.typeclasses.binding
26 import org.onap.dcae.collectors.veshv.main.servers.HealthCheckServer
27 import org.onap.dcae.collectors.veshv.main.servers.VesServer
28 import org.onap.dcae.collectors.veshv.model.ServerConfiguration
29 import org.onap.dcae.collectors.veshv.utils.arrow.ExitFailure
30 import org.onap.dcae.collectors.veshv.utils.arrow.unsafeRunEitherSync
31 import org.onap.dcae.collectors.veshv.utils.commandline.handleWrongArgumentErrorCurried
32 import org.onap.dcae.collectors.veshv.utils.logging.Logger
33 import org.onap.dcae.collectors.veshv.utils.registerShutdownHook
34
35 private const val VESHV_PACKAGE = "org.onap.dcae.collectors.veshv"
36 private val logger = Logger("$VESHV_PACKAGE.main")
37 private const val PROGRAM_NAME = "java $VESHV_PACKAGE.main.MainKt"
38
39 fun main(args: Array<String>) =
40     ArgVesHvConfiguration().parse(args)
41         .mapLeft(handleWrongArgumentErrorCurried(PROGRAM_NAME))
42         .map(::startAndAwaitServers)
43         .unsafeRunEitherSync(
44             { ex ->
45                 logger.withError { log("Failed to start a server", ex) }
46                 ExitFailure(1)
47             },
48             { logger.info { "Finished" } }
49         )
50
51 private fun startAndAwaitServers(config: ServerConfiguration) =
52     IO.monad().binding {
53         Logger.setLogLevel(VESHV_PACKAGE, config.logLevel)
54         logger.info { "Using configuration: $config" }
55         HealthCheckServer.start(config).bind()
56         VesServer.start(config).bind().run {
57             registerShutdownHook(shutdown()).bind()
58             await().bind()
59         }
60     }.fix()