Use CBS by means of SDK in place of Consul
[dcaegen2/collectors/hv-ves.git] / sources / hv-collector-core / src / main / kotlin / org / onap / dcae / collectors / veshv / factory / CollectorFactory.kt
1 /*
2  * ============LICENSE_START=======================================================
3  * dcaegen2-collectors-veshv
4  * ================================================================================
5  * Copyright (C) 2018-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.factory
21
22 import arrow.core.Option
23 import arrow.effects.IO
24 import org.onap.dcae.collectors.veshv.boundary.Collector
25 import org.onap.dcae.collectors.veshv.boundary.CollectorProvider
26 import org.onap.dcae.collectors.veshv.boundary.ConfigurationProvider
27 import org.onap.dcae.collectors.veshv.boundary.Metrics
28 import org.onap.dcae.collectors.veshv.boundary.SinkProvider
29 import org.onap.dcae.collectors.veshv.domain.WireFrameDecoder
30 import org.onap.dcae.collectors.veshv.healthcheck.api.HealthDescription
31 import org.onap.dcae.collectors.veshv.healthcheck.api.HealthState
32 import org.onap.dcae.collectors.veshv.impl.Router
33 import org.onap.dcae.collectors.veshv.impl.VesDecoder
34 import org.onap.dcae.collectors.veshv.impl.VesHvCollector
35 import org.onap.dcae.collectors.veshv.impl.wire.WireChunkDecoder
36 import org.onap.dcae.collectors.veshv.model.ClientContext
37 import org.onap.dcae.collectors.veshv.model.CollectorConfiguration
38 import org.onap.dcae.collectors.veshv.model.ServiceContext
39 import org.onap.dcae.collectors.veshv.utils.arrow.getOption
40 import org.onap.dcae.collectors.veshv.utils.logging.Logger
41 import java.util.concurrent.atomic.AtomicReference
42
43 /**
44  * @author Piotr Jaszczyk <piotr.jaszczyk@nokia.com>
45  * @since May 2018
46  */
47 class CollectorFactory(val configuration: ConfigurationProvider,
48                        private val sinkProvider: SinkProvider,
49                        private val metrics: Metrics,
50                        private val maximumPayloadSizeBytes: Int,
51                        private val healthState: HealthState = HealthState.INSTANCE) {
52
53     fun createVesHvCollectorProvider(): CollectorProvider {
54         val config: AtomicReference<CollectorConfiguration> = AtomicReference()
55         configuration()
56                 .doOnNext {
57                     logger.info(ServiceContext::mdc) { "Using updated configuration for new connections" }
58                     healthState.changeState(HealthDescription.HEALTHY)
59                 }
60                 .doOnError {
61                     logger.error(ServiceContext::mdc) { "Failed to acquire configuration ${it.message}" }
62                     logger.debug(ServiceContext::mdc) { "Detailed stack trace: $it" }
63                     healthState.changeState(HealthDescription.DYNAMIC_CONFIGURATION_NOT_FOUND)
64                 }
65                 .subscribe(config::set)
66
67         return object : CollectorProvider {
68             override fun invoke(ctx: ClientContext): Option<Collector> =
69                     config.getOption().map { createVesHvCollector(it, ctx) }
70
71             override fun close() = sinkProvider.close()
72         }
73     }
74
75     private fun createVesHvCollector(config: CollectorConfiguration, ctx: ClientContext): Collector =
76             VesHvCollector(
77                     clientContext = ctx,
78                     wireChunkDecoder = WireChunkDecoder(WireFrameDecoder(maximumPayloadSizeBytes), ctx),
79                     protobufDecoder = VesDecoder(),
80                     router = Router(config.routing, ctx),
81                     sink = sinkProvider(ctx),
82                     metrics = metrics)
83
84     companion object {
85         private val logger = Logger(CollectorFactory::class)
86     }
87 }
88