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

   1 /*
   2  * ============LICENSE_START=======================================================
   3  * dcaegen2-collectors-veshv
   4  * ================================================================================
   5  * Copyright (C) 2018 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.dcaeapp.impl
  21
  22 import arrow.effects.IO
  23 import arrow.effects.fix
  24 import arrow.effects.instances.io.monadError.monadError
  25 import arrow.typeclasses.bindingCatch
  26 import org.onap.dcae.collectors.veshv.utils.arrow.asIo
  27 import org.onap.dcae.collectors.veshv.utils.logging.Logger
  28 import org.onap.dcae.collectors.veshv.ves.message.generator.api.MessageParameters
  29 import org.onap.dcae.collectors.veshv.ves.message.generator.api.MessageParametersParser
  30 import org.onap.dcae.collectors.veshv.ves.message.generator.api.VesEventParameters
  31 import org.onap.dcae.collectors.veshv.ves.message.generator.api.VesEventType.FIXED_PAYLOAD
  32 import org.onap.dcae.collectors.veshv.ves.message.generator.generators.VesEventGenerator
  33 import org.onap.ves.VesEventOuterClass.VesEvent
  34 import reactor.core.publisher.Flux
  35 import java.io.InputStream
  36 import javax.json.Json
  37
  38 class MessageStreamValidation(
  39         private val messageGenerator: VesEventGenerator,
  40         private val messageParametersParser: MessageParametersParser = MessageParametersParser.INSTANCE) {
  41
  42     fun validate(jsonDescription: InputStream, consumedMessages: List<ByteArray>): IO<Boolean> =
  43             IO.monadError().bindingCatch {
  44                 val messageParams = parseMessageParams(jsonDescription)
  45                 logger.debug { "Parsed message parameters: $messageParams" }
  46
  47                 val expectedEvents = generateEvents(messageParams).bind()
  48                 val actualEvents = decodeConsumedEvents(consumedMessages)
  49
  50                 if (shouldValidatePayloads(messageParams))
  51                     expectedEvents == actualEvents
  52                 else
  53                     validateHeaders(actualEvents, expectedEvents)
  54
  55             }.fix()
  56
  57     private fun parseMessageParams(input: InputStream): List<VesEventParameters> {
  58         val paramsArray = Json.createReader(input).readArray()
  59         val messageParams = messageParametersParser.parse(paramsArray)
  60
  61         return messageParams.fold(
  62                 {
  63                     logger.warn { "Error while parsing message parameters: ${it::class.qualifiedName} : ${it.message}" }
  64                     logger.debug { "Detailed stack trace: $it" }
  65                     throw IllegalArgumentException("Parsing error: " + it.message)
  66                 },
  67                 {
  68                     toVesEventParams(it)
  69                 }
  70         )
  71     }
  72
  73     private fun toVesEventParams(params: List<MessageParameters>): List<VesEventParameters> =
  74             if (params.isEmpty()) {
  75                 val message = "Message param list cannot be empty"
  76                 logger.warn { message }
  77                 throw IllegalArgumentException(message)
  78             } else params.map(::validateMessageParams)
  79
  80
  81     private fun validateMessageParams(params: MessageParameters): VesEventParameters =
  82             if (params !is VesEventParameters) {
  83                 val message = "Only VesEvent-related message types can be validated. " +
  84                         "Correct values are: VALID, TOO_BIG_PAYLOAD, FIXED_PAYLOAD"
  85                 logger.warn { message }
  86                 throw IllegalArgumentException(message)
  87             } else params
  88
  89
  90     private fun shouldValidatePayloads(parameters: List<VesEventParameters>) =
  91             parameters.all { it.messageType == FIXED_PAYLOAD }
  92
  93     private fun validateHeaders(actual: List<VesEvent>,
  94                                 expected: List<VesEvent>): Boolean {
  95         val consumedHeaders = actual.map { it.commonEventHeader }
  96         val generatedHeaders = expected.map { it.commonEventHeader }
  97         return generatedHeaders == consumedHeaders
  98     }
  99
 100     private fun generateEvents(parameters: List<VesEventParameters>): IO<List<VesEvent>> = Flux
 101             .fromIterable(parameters)
 102             .flatMap { messageGenerator.createMessageFlux(it) }
 103             .collectList()
 104             .asIo()
 105
 106     private fun decodeConsumedEvents(consumedMessages: List<ByteArray>) =
 107             consumedMessages.map(VesEvent::parseFrom)
 108
 109     companion object {
 110         private val logger = Logger(MessageStreamValidation::class)
 111     }
 112 }