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

   1 /*
   2  * ============LICENSE_START=======================================================
   3  * dcaegen2-collectors-veshv
   4  * ================================================================================
   5  * Copyright (C) 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.config.api
  21
  22 import org.onap.dcae.collectors.veshv.config.api.model.HvVesConfiguration
  23 import org.onap.dcae.collectors.veshv.config.api.model.MissingArgumentException
  24 import org.onap.dcae.collectors.veshv.config.impl.CbsConfigurationProvider
  25 import org.onap.dcae.collectors.veshv.config.impl.ConfigurationMerger
  26 import org.onap.dcae.collectors.veshv.config.impl.ConfigurationTransformer
  27 import org.onap.dcae.collectors.veshv.config.impl.ConfigurationValidator
  28 import org.onap.dcae.collectors.veshv.config.impl.CbsClientAdapter
  29 import org.onap.dcae.collectors.veshv.config.impl.HvVesCommandLineParser
  30 import org.onap.dcae.collectors.veshv.config.impl.JsonConfigurationParser
  31 import org.onap.dcae.collectors.veshv.config.impl.PartialConfiguration
  32 import org.onap.dcae.collectors.veshv.utils.arrow.throwOnLeft
  33 import org.onap.dcae.collectors.veshv.utils.logging.Logger
  34 import org.onap.dcae.collectors.veshv.utils.logging.MappedDiagnosticContext
  35 import org.onap.dcaegen2.services.sdk.rest.services.cbs.client.api.CbsClient
  36 import org.onap.dcaegen2.services.sdk.rest.services.cbs.client.api.CbsClientFactory
  37 import org.onap.dcaegen2.services.sdk.rest.services.cbs.client.model.EnvProperties
  38 import reactor.core.publisher.Flux
  39 import reactor.core.publisher.Mono
  40 import reactor.retry.Jitter
  41 import reactor.retry.Retry
  42 import java.time.Duration
  43
  44 class ConfigurationModule internal constructor(private val configStateListener: ConfigurationStateListener,
  45                                                private val cbsClient: Mono<CbsClient>) {
  46
  47     private val cmd = HvVesCommandLineParser()
  48     private val configParser = JsonConfigurationParser()
  49     private val configMerger = ConfigurationMerger()
  50     private val configValidator = ConfigurationValidator()
  51     private val configTransformer = ConfigurationTransformer()
  52
  53     constructor(configStateListener: ConfigurationStateListener) : this(
  54             configStateListener,
  55             CbsClientFactory.createCbsClient(EnvProperties.fromEnvironment())
  56     )
  57
  58
  59     fun healthCheckPort(args: Array<String>): Int = cmd.getHealthcheckPort(args)
  60
  61     fun hvVesConfigurationUpdates(args: Array<String>,
  62                                   mdc: MappedDiagnosticContext): Flux<HvVesConfiguration> =
  63             Mono.just(cmd.getConfigurationFile(args))
  64                     .throwOnLeft(::MissingArgumentException)
  65                     .doOnNext { logger.info { "Using base configuration file: ${it.absolutePath}" } }
  66                     .map { it.reader().use(configParser::parse) }
  67                     .doOnNext { logger.info { "Successfully parsed configuration file to: $it" } }
  68                     .cache()
  69                     .flatMapMany { basePartialConfig ->
  70                         cbsClientAdapter(basePartialConfig).let { cbsClientAdapter ->
  71                             cbsConfigurationProvider(cbsClientAdapter, mdc)
  72                                     .invoke()
  73                                     .map { configMerger.merge(basePartialConfig, it) }
  74                                     .map(configValidator::validate)
  75                                     .throwOnLeft()
  76                                     .map(configTransformer::toFinalConfiguration)
  77                                     .doOnNext {
  78                                         cbsClientAdapter.updateCbsInterval(it.cbs.requestInterval, mdc)
  79                                     }
  80                         }
  81                     }
  82
  83     private fun cbsClientAdapter(basePartialConfig: PartialConfiguration) =
  84             CbsClientAdapter(
  85                     cbsClient,
  86                     configStateListener,
  87                     cbsConfigurationFrom(basePartialConfig).firstRequestDelay,
  88                     retrySpec
  89             )
  90
  91     private fun cbsConfigurationProvider(cbsClientAdapter: CbsClientAdapter,
  92                                          mdc: MappedDiagnosticContext) =
  93             CbsConfigurationProvider(
  94                     cbsClientAdapter,
  95                     configParser,
  96                     configStateListener,
  97                     mdc,
  98                     retrySpec)
  99
 100     private fun cbsConfigurationFrom(basePartialConfig: PartialConfiguration) =
 101             configValidator.validatedCbsConfiguration(basePartialConfig)
 102                     .let { configTransformer.toCbsConfiguration(it) }
 103
 104     companion object {
 105         private val logger = Logger(ConfigurationModule::class)
 106
 107         private const val MAX_RETRIES = 5L
 108         private const val INITIAL_BACKOFF = 10L
 109         private val retrySpec: Retry<Any> = Retry.any<Any>()
 110                 .retryMax(MAX_RETRIES)
 111                 .fixedBackoff(Duration.ofSeconds(INITIAL_BACKOFF))
 112                 .jitter(Jitter.random())
 113     }
 114
 115 }