gerrit.onap Code Review - ccsdk/cds.git/blob

   1 /*
   2  *  Copyright © 2019 IBM.
   3  *
   4  *  Licensed under the Apache License, Version 2.0 (the "License");
   5  *  you may not use this file except in compliance with the License.
   6  *  You may obtain a copy of the License at
   7  *
   8  *      http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  *  Unless required by applicable law or agreed to in writing, software
  11  *  distributed under the License is distributed on an "AS IS" BASIS,
  12  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  *  See the License for the specific language governing permissions and
  14  *  limitations under the License.
  15  */
  16
  17 package org.onap.ccsdk.cds.blueprintsprocessor.selfservice.api
  18
  19 import kotlinx.coroutines.channels.consumeEach
  20 import kotlinx.coroutines.launch
  21 import kotlinx.coroutines.runBlocking
  22 import org.onap.ccsdk.cds.blueprintsprocessor.core.api.data.ExecutionServiceInput
  23 import org.onap.ccsdk.cds.blueprintsprocessor.message.service.BluePrintMessageLibPropertyService
  24 import org.onap.ccsdk.cds.blueprintsprocessor.message.service.BlueprintMessageConsumerService
  25 import org.onap.ccsdk.cds.controllerblueprints.core.BluePrintProcessorException
  26 import org.onap.ccsdk.cds.controllerblueprints.core.jsonAsType
  27 import org.onap.ccsdk.cds.controllerblueprints.core.logger
  28 import org.onap.ccsdk.cds.controllerblueprints.core.updateErrorMessage
  29 import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty
  30 import org.springframework.boot.context.event.ApplicationReadyEvent
  31 import org.springframework.context.event.EventListener
  32 import org.springframework.stereotype.Service
  33 import java.util.concurrent.Phaser
  34 import javax.annotation.PreDestroy
  35
  36 @ConditionalOnProperty(
  37     name = ["blueprintsprocessor.messageconsumer.self-service-api.kafkaEnable"],
  38     havingValue = "true"
  39 )
  40 @Service
  41 open class BluePrintProcessingKafkaConsumer(
  42     private val bluePrintMessageLibPropertyService: BluePrintMessageLibPropertyService,
  43     private val executionServiceHandler: ExecutionServiceHandler
  44 ) {
  45
  46     val log = logger(BluePrintProcessingKafkaConsumer::class)
  47
  48     private val ph = Phaser(1)
  49
  50     private lateinit var blueprintMessageConsumerService: BlueprintMessageConsumerService
  51
  52     companion object {
  53         const val CONSUMER_SELECTOR = "self-service-api"
  54     }
  55
  56     @EventListener(ApplicationReadyEvent::class)
  57     fun setupMessageListener() = runBlocking {
  58         try {
  59             log.info(
  60                 "Setting up message consumer($CONSUMER_SELECTOR)"
  61             )
  62
  63             /** Get the Message Consumer Service **/
  64             blueprintMessageConsumerService = try {
  65                 bluePrintMessageLibPropertyService
  66                     .blueprintMessageConsumerService(CONSUMER_SELECTOR)
  67             } catch (e: BluePrintProcessorException) {
  68                 val errorMsg = "Failed creating Kafka consumer message service."
  69                 throw e.updateErrorMessage(SelfServiceApiDomains.SELF_SERVICE_API, errorMsg,
  70                         "Wrong Kafka selector provided or internal error in Kafka service.")
  71             } catch (e: Exception) {
  72                 throw BluePrintProcessorException("failed to create consumer service ${e.message}")
  73             }
  74
  75             launch {
  76                 /** Subscribe to the consumer topics */
  77                 val additionalConfig: MutableMap<String, Any> = hashMapOf()
  78                 val channel = blueprintMessageConsumerService.subscribe(additionalConfig)
  79                 channel.consumeEach { message ->
  80                     launch {
  81                         try {
  82                             ph.register()
  83                             log.trace("Consumed Message : $message")
  84                             val executionServiceInput = message.jsonAsType<ExecutionServiceInput>()
  85                             executionServiceHandler.doProcess(executionServiceInput)
  86                         } catch (e: Exception) {
  87                             log.error("failed in processing the consumed message : $message", e)
  88                         } finally {
  89                             ph.arriveAndDeregister()
  90                         }
  91                     }
  92                 }
  93             }
  94         } catch (e: Exception) {
  95             log.error(
  96                 "failed to start message consumer($CONSUMER_SELECTOR) ", e
  97             )
  98         }
  99     }
 100
 101     @PreDestroy
 102     fun shutdownMessageListener() = runBlocking {
 103         try {
 104             log.info(
 105                 "Shutting down message consumer($CONSUMER_SELECTOR)"
 106             )
 107             blueprintMessageConsumerService.shutDown()
 108             ph.arriveAndAwaitAdvance()
 109         } catch (e: Exception) {
 110             log.error("failed to shutdown message listener($CONSUMER_SELECTOR)", e)
 111         }
 112     }
 113 }