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

   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.simulators.xnf.impl.factory.ClientFactory
  30 import org.onap.dcae.collectors.veshv.utils.arrow.asIo
  31 import org.onap.dcae.collectors.veshv.ves.message.generator.api.MessageParameters
  32 import org.onap.dcae.collectors.veshv.ves.message.generator.api.MessageParametersParser
  33 import org.onap.dcae.collectors.veshv.ves.message.generator.api.ParsingError
  34 import org.onap.dcae.collectors.veshv.ves.message.generator.api.VesEventParameters
  35 import org.onap.dcae.collectors.veshv.ves.message.generator.api.WireFrameParameters
  36 import org.onap.dcae.collectors.veshv.ves.message.generator.factory.MessageGeneratorFactory
  37 import reactor.core.publisher.Mono
  38 import reactor.core.publisher.toFlux
  39 import java.io.InputStream
  40 import javax.json.Json
  41 import javax.json.JsonArray
  42
  43 /**
  44  * @author Piotr Jaszczyk <piotr.jaszczyk@nokia.com>
  45  * @since August 2018
  46  */
  47 class XnfSimulator(
  48         private val clientFactory: ClientFactory,
  49         private val generatorFactory: MessageGeneratorFactory,
  50         private val messageParametersParser: MessageParametersParser = MessageParametersParser.INSTANCE) {
  51
  52     private val wireFrameGenerator by lazy { generatorFactory.createWireFrameGenerator() }
  53     private val vesEventGenerator by lazy { generatorFactory.createVesEventGenerator() }
  54
  55     fun startSimulation(messageParameters: InputStream): Either<ParsingError, IO<Unit>> =
  56             Either.monad<ParsingError>().binding {
  57                 val json = parseJsonArray(messageParameters).bind()
  58                 val parameters = messageParametersParser.parse(json).bind()
  59                 simulationFrom(parameters)
  60             }.fix()
  61
  62     private fun parseJsonArray(jsonStream: InputStream): Either<ParsingError, JsonArray> =
  63             Try { Json.createReader(jsonStream).readArray() }
  64                     .toEither()
  65                     .mapLeft { ParsingError("Failed to parse JSON", Some(it)) }
  66
  67     private fun simulationFrom(parameters: List<MessageParameters>): IO<Unit> = parameters
  68             .toFlux()
  69             .map(::simulate)
  70             .then(Mono.just(Unit))
  71             .asIo()
  72
  73     private fun simulate(parameters: MessageParameters): Mono<Unit> =
  74             when (parameters) {
  75                 is VesEventParameters -> {
  76                     val messages = vesEventGenerator.createMessageFlux(parameters)
  77                     val client = clientFactory.create()
  78                     client.sendVesEvents(messages)
  79                 }
  80                 is WireFrameParameters -> {
  81                     val messages = wireFrameGenerator.createMessageFlux(parameters)
  82                     val client = clientFactory.create(parameters.wireFrameVersion)
  83                     client.sendRawPayload(messages)
  84                 }
  85             }
  86 }