gerrit.onap Code Review - dcaegen2/collectors/hv-ves.git/blob

   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.core.Option
  23 import arrow.core.fix
  24 import arrow.core.getOrElse
  25 import arrow.instances.option.monad.monad
  26 import arrow.typeclasses.binding
  27 import org.apache.commons.cli.CommandLine
  28 import org.apache.commons.cli.DefaultParser
  29 import org.onap.dcae.collectors.veshv.domain.WireFrameMessage
  30 import org.onap.dcae.collectors.veshv.model.ConfigurationProviderParams
  31 import org.onap.dcae.collectors.veshv.model.KafkaConfiguration
  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.ssl.boundary.createSecurityConfiguration
  35 import org.onap.dcae.collectors.veshv.utils.arrow.doOnFailure
  36 import org.onap.dcae.collectors.veshv.utils.commandline.*
  37 import org.onap.dcae.collectors.veshv.utils.commandline.CommandLineOption.CONSUL_CONFIG_URL
  38 import org.onap.dcae.collectors.veshv.utils.commandline.CommandLineOption.KAFKA_SERVERS
  39 import org.onap.dcae.collectors.veshv.utils.commandline.CommandLineOption.CONSUL_FIRST_REQUEST_DELAY
  40 import org.onap.dcae.collectors.veshv.utils.commandline.CommandLineOption.CONSUL_REQUEST_INTERVAL
  41 import org.onap.dcae.collectors.veshv.utils.commandline.CommandLineOption.DUMMY_MODE
  42 import org.onap.dcae.collectors.veshv.utils.commandline.CommandLineOption.HEALTH_CHECK_API_PORT
  43 import org.onap.dcae.collectors.veshv.utils.commandline.CommandLineOption.IDLE_TIMEOUT_SEC
  44 import org.onap.dcae.collectors.veshv.utils.commandline.CommandLineOption.KEY_STORE_FILE
  45 import org.onap.dcae.collectors.veshv.utils.commandline.CommandLineOption.KEY_STORE_PASSWORD
  46 import org.onap.dcae.collectors.veshv.utils.commandline.CommandLineOption.LISTEN_PORT
  47 import org.onap.dcae.collectors.veshv.utils.commandline.CommandLineOption.MAXIMUM_PAYLOAD_SIZE_BYTES
  48 import org.onap.dcae.collectors.veshv.utils.commandline.CommandLineOption.SSL_DISABLE
  49 import org.onap.dcae.collectors.veshv.utils.commandline.CommandLineOption.TRUST_STORE_FILE
  50 import org.onap.dcae.collectors.veshv.utils.commandline.CommandLineOption.TRUST_STORE_PASSWORD
  51 import org.onap.dcae.collectors.veshv.utils.commandline.CommandLineOption.LOG_LEVEL
  52 import org.onap.dcae.collectors.veshv.utils.logging.LogLevel
  53 import org.onap.dcae.collectors.veshv.utils.logging.Logger
  54 import java.net.InetSocketAddress
  55 import java.time.Duration
  56
  57
  58 internal class ArgVesHvConfiguration : ArgBasedConfiguration<ServerConfiguration>(DefaultParser()) {
  59     override val cmdLineOptionsList = listOf(
  60             KAFKA_SERVERS,
  61             HEALTH_CHECK_API_PORT,
  62             LISTEN_PORT,
  63             CONSUL_CONFIG_URL,
  64             CONSUL_FIRST_REQUEST_DELAY,
  65             CONSUL_REQUEST_INTERVAL,
  66             SSL_DISABLE,
  67             KEY_STORE_FILE,
  68             KEY_STORE_PASSWORD,
  69             TRUST_STORE_FILE,
  70             TRUST_STORE_PASSWORD,
  71             IDLE_TIMEOUT_SEC,
  72             MAXIMUM_PAYLOAD_SIZE_BYTES,
  73             DUMMY_MODE,
  74             LOG_LEVEL
  75     )
  76
  77     override fun getConfiguration(cmdLine: CommandLine): Option<ServerConfiguration> =
  78             Option.monad().binding {
  79                 val healthCheckApiPort = cmdLine.intValue(
  80                         HEALTH_CHECK_API_PORT,
  81                         DefaultValues.HEALTH_CHECK_API_PORT
  82                 )
  83                 val kafkaServers = cmdLine.stringValue(KAFKA_SERVERS).bind()
  84                 val listenPort = cmdLine.intValue(LISTEN_PORT).bind()
  85                 val idleTimeoutSec = cmdLine.longValue(IDLE_TIMEOUT_SEC, DefaultValues.IDLE_TIMEOUT_SEC)
  86                 val maxPayloadSizeBytes = cmdLine.intValue(
  87                         MAXIMUM_PAYLOAD_SIZE_BYTES,
  88                         DefaultValues.MAX_PAYLOAD_SIZE_BYTES
  89                 )
  90                 val dummyMode = cmdLine.hasOption(DUMMY_MODE)
  91                 val security = createSecurityConfiguration(cmdLine)
  92                         .doOnFailure { ex ->
  93                             logger.withError(ServiceContext::mdc) {
  94                                 log("Could not read security keys", ex)
  95                             }
  96                         }
  97                         .toOption()
  98                         .bind()
  99                 val logLevel = cmdLine.stringValue(LOG_LEVEL, DefaultValues.LOG_LEVEL)
 100                 val configurationProviderParams = createConfigurationProviderParams(cmdLine).bind()
 101                 ServerConfiguration(
 102                         serverListenAddress = InetSocketAddress(listenPort),
 103                         kafkaConfiguration = KafkaConfiguration(kafkaServers, maxPayloadSizeBytes),
 104                         healthCheckApiListenAddress = InetSocketAddress(healthCheckApiPort),
 105                         configurationProviderParams = configurationProviderParams,
 106                         securityConfiguration = security,
 107                         idleTimeout = Duration.ofSeconds(idleTimeoutSec),
 108                         maximumPayloadSizeBytes = maxPayloadSizeBytes,
 109                         dummyMode = dummyMode,
 110                         logLevel = determineLogLevel(logLevel)
 111                 )
 112             }.fix()
 113
 114
 115     private fun createConfigurationProviderParams(cmdLine: CommandLine): Option<ConfigurationProviderParams> =
 116             Option.monad().binding {
 117                 val configUrl = cmdLine.stringValue(CONSUL_CONFIG_URL).bind()
 118                 val firstRequestDelay = cmdLine.longValue(
 119                         CONSUL_FIRST_REQUEST_DELAY,
 120                         DefaultValues.CONSUL_FIRST_REQUEST_DELAY
 121                 )
 122                 val requestInterval = cmdLine.longValue(
 123                         CONSUL_REQUEST_INTERVAL,
 124                         DefaultValues.CONSUL_REQUEST_INTERVAL
 125                 )
 126                 ConfigurationProviderParams(
 127                         configUrl,
 128                         Duration.ofSeconds(firstRequestDelay),
 129                         Duration.ofSeconds(requestInterval)
 130                 )
 131             }.fix()
 132
 133     private fun determineLogLevel(logLevel: String) = LogLevel.optionFromString(logLevel)
 134             .getOrElse {
 135                 logger.warn {
 136                     "Failed to parse $logLevel as $LOG_LEVEL command line. " +
 137                             "Using default log level (${DefaultValues.LOG_LEVEL})"
 138                 }
 139                 LogLevel.valueOf(DefaultValues.LOG_LEVEL)
 140             }
 141
 142
 143     internal object DefaultValues {
 144         const val HEALTH_CHECK_API_PORT = 6060
 145         const val CONSUL_FIRST_REQUEST_DELAY = 10L
 146         const val CONSUL_REQUEST_INTERVAL = 5L
 147         const val IDLE_TIMEOUT_SEC = 60L
 148         const val MAX_PAYLOAD_SIZE_BYTES = WireFrameMessage.DEFAULT_MAX_PAYLOAD_SIZE_BYTES
 149         val LOG_LEVEL = LogLevel.INFO.name
 150     }
 151
 152     companion object {
 153         private val logger = Logger(ArgVesHvConfiguration::class)
 154     }
 155 }