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
11 * http://www.apache.org/licenses/LICENSE-2.0
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=========================================================
20 package org.onap.dcae.collectors.veshv.simulators.xnf.impl
22 import arrow.core.Either
23 import arrow.core.Some
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
41 * @author Piotr Jaszczyk <piotr.jaszczyk@nokia.com>
45 private val vesClient: VesHvClient,
46 private val generatorFactory: MessageGeneratorFactory,
47 private val messageParametersParser: MessageParametersParser = MessageParametersParser.INSTANCE) {
49 fun startSimulation(messageParameters: InputStream): Either<ParsingError, IO<Unit>> =
50 Either.monad<ParsingError>().binding {
52 val json = parseJsonArray(messageParameters).bind()
53 messageParametersParser.parse(json).bind()
55 .flatMap(::generateMessages)
56 .let { vesClient.sendIo(it) }
59 private fun parseJsonArray(jsonStream: InputStream): Either<ParsingError, JsonArray> =
60 Try { Json.createReader(jsonStream).readArray() }
62 .mapLeft { ParsingError("Failed to parse JSON", Some(it)) }
64 private fun generateMessages(parameters: MessageParameters): Flux<WireFrameMessage> =
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")
76 private fun encodeToWireFrame(event: VesEvent): WireFrameMessage =
77 WireFrameMessage(event.toByteArray())