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