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     fun healthCheckPort(args: Array<String>): Int = cmd.getHealthcheckPort(args)
  59
  60     fun hvVesConfigurationUpdates(args: Array<String>,
  61                                   mdc: MappedDiagnosticContext): Flux<HvVesConfiguration> =
  62             Mono.just(cmd.getConfigurationFile(args))
  63                     .throwOnLeft(::MissingArgumentException)
  64                     .doOnNext { logger.info { "Using base configuration file: ${it.absolutePath}" } }
  65                     .map { it.reader().use(configParser::parse) }
  66                     .doOnNext { logger.info { "Successfully parsed configuration file to: $it" } }
  67                     .cache()
  68                     .flatMapMany { basePartialConfig ->
  69                         cbsClientAdapter(basePartialConfig, mdc).let { cbsClientAdapter ->
  70                             cbsConfigurationProvider(cbsClientAdapter, mdc)
  71                                     .invoke()
  72                                     .map { configMerger.merge(basePartialConfig, it) }
  73                                     .map(configValidator::validate)
  74                                     .throwOnLeft()
  75                                     .map(configTransformer::toFinalConfiguration)
  76                                     .doOnNext {
  77                                         cbsClientAdapter.updateCbsInterval(it.cbs.requestInterval)
  78                                     }
  79                         }
  80                     }
  81
  82     private fun cbsClientAdapter(basePartialConfig: PartialConfiguration,
  83                                  mdc: MappedDiagnosticContext) = CbsClientAdapter(
  84             cbsClient,
  85             cbsConfigurationFrom(basePartialConfig).firstRequestDelay,
  86             configStateListener,
  87             mdc,
  88             infiniteRetry
  89     )
  90
  91     private fun cbsConfigurationProvider(cbsClientAdapter: CbsClientAdapter,
  92                                          mdc: MappedDiagnosticContext) = CbsConfigurationProvider(
  93             cbsClientAdapter,
  94             configParser,
  95             configStateListener,
  96             mdc,
  97             infiniteRetry
  98     )
  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 val FIRST_BACKOFF_DURATION = Duration.ofSeconds(5)
 108         private val MAX_BACKOFF_DURATION = Duration.ofMinutes(5)
 109         private val infiniteRetry: Retry<Any> = Retry.any<Any>()
 110                 .retryMax(Long.MAX_VALUE)
 111                 .exponentialBackoff(FIRST_BACKOFF_DURATION, MAX_BACKOFF_DURATION)
 112                 .jitter(Jitter.random())
 113     }
 114
 115 }