4dfdb845f1933132149afcc92d666d59f51790fd
[dcaegen2/collectors/hv-ves.git] /
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.simulators.xnf.impl
21
22 import arrow.core.Either
23 import arrow.core.Some
24 import arrow.core.Try
25 import arrow.core.fix
26 import arrow.effects.IO
27 import arrow.instances.either.monad.monad
28 import arrow.typeclasses.binding
29 import org.onap.dcae.collectors.veshv.domain.WireFrameMessage
30 import org.onap.dcae.collectors.veshv.simulators.xnf.impl.adapters.VesHvClient
31 import org.onap.dcae.collectors.veshv.ves.message.generator.api.*
32 import org.onap.dcae.collectors.veshv.ves.message.generator.factory.MessageGeneratorFactory
33 import org.onap.ves.VesEventOuterClass.VesEvent
34 import reactor.core.publisher.Flux
35 import reactor.core.publisher.toFlux
36 import java.io.InputStream
37 import javax.json.Json
38 import javax.json.JsonArray
39
40 /**
41  * @author Piotr Jaszczyk <piotr.jaszczyk@nokia.com>
42  * @since August 2018
43  */
44 class XnfSimulator(
45         private val vesClient: VesHvClient,
46         private val generatorFactory: MessageGeneratorFactory,
47         private val messageParametersParser: MessageParametersParser = MessageParametersParser.INSTANCE) {
48
49     fun startSimulation(messageParameters: InputStream): Either<ParsingError, IO<Unit>> =
50             Either.monad<ParsingError>().binding {
51
52                 val json = parseJsonArray(messageParameters).bind()
53                 messageParametersParser.parse(json).bind()
54                         .toFlux()
55                         .flatMap(::generateMessages)
56                         .let { vesClient.sendIo(it) }
57             }.fix()
58
59     private fun parseJsonArray(jsonStream: InputStream): Either<ParsingError, JsonArray> =
60             Try { Json.createReader(jsonStream).readArray() }
61                     .toEither()
62                     .mapLeft { ParsingError("Failed to parse JSON", Some(it)) }
63
64     private fun generateMessages(parameters: MessageParameters): Flux<WireFrameMessage> =
65             when (parameters) {
66                 is VesEventParameters -> generatorFactory
67                         .createVesEventGenerator()
68                         .createMessageFlux(parameters)
69                         .map(::encodeToWireFrame)
70                 is WireFrameParameters -> generatorFactory
71                         .createWireFrameGenerator()
72                         .createMessageFlux(parameters)
73                 else -> throw IllegalStateException("Invalid parameters type")
74             }
75
76     private fun encodeToWireFrame(event: VesEvent): WireFrameMessage =
77             WireFrameMessage(event.toByteArray())
78 }