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

   1 /*
   2  *  Copyright © 2019 IBM.
   3  *  Modifications Copyright © 2018-2019 AT&T Intellectual Property.
   4  *
   5  *  Licensed under the Apache License, Version 2.0 (the "License");
   6  *  you may not use this file except in compliance with the License.
   7  *  You may obtain a copy of the License at
   8  *
   9  *      http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  *  Unless required by applicable law or agreed to in writing, software
  12  *  distributed under the License is distributed on an "AS IS" BASIS,
  13  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  *  See the License for the specific language governing permissions and
  15  *  limitations under the License.
  16  */
  17
  18 package org.onap.ccsdk.cds.blueprintsprocessor.message.service
  19
  20 import org.apache.commons.lang.builder.ToStringBuilder
  21 import org.apache.kafka.clients.producer.Callback
  22 import org.apache.kafka.clients.producer.KafkaProducer
  23 import org.apache.kafka.clients.producer.ProducerConfig.ACKS_CONFIG
  24 import org.apache.kafka.clients.producer.ProducerConfig.BOOTSTRAP_SERVERS_CONFIG
  25 import org.apache.kafka.clients.producer.ProducerConfig.CLIENT_ID_CONFIG
  26 import org.apache.kafka.clients.producer.ProducerConfig.ENABLE_IDEMPOTENCE_CONFIG
  27 import org.apache.kafka.clients.producer.ProducerConfig.KEY_SERIALIZER_CLASS_CONFIG
  28 import org.apache.kafka.clients.producer.ProducerConfig.VALUE_SERIALIZER_CLASS_CONFIG
  29 import org.apache.kafka.clients.producer.ProducerRecord
  30 import org.apache.kafka.common.header.internals.RecordHeader
  31 import org.apache.kafka.common.serialization.ByteArraySerializer
  32 import org.apache.kafka.common.serialization.StringSerializer
  33 import org.onap.ccsdk.cds.blueprintsprocessor.message.KafkaBasicAuthMessageProducerProperties
  34 import org.onap.ccsdk.cds.controllerblueprints.core.asJsonString
  35 import org.onap.ccsdk.cds.controllerblueprints.core.defaultToUUID
  36 import org.slf4j.LoggerFactory
  37 import java.nio.charset.Charset
  38
  39 class KafkaBasicAuthMessageProducerService(
  40     private val messageProducerProperties: KafkaBasicAuthMessageProducerProperties
  41 ) :
  42     BlueprintMessageProducerService {
  43
  44     private val log = LoggerFactory.getLogger(KafkaBasicAuthMessageProducerService::class.java)!!
  45
  46     private var kafkaProducer: KafkaProducer<String, ByteArray>? = null
  47
  48     private val messageLoggerService = MessageLoggerService()
  49
  50     override suspend fun sendMessageNB(message: Any): Boolean {
  51         checkNotNull(messageProducerProperties.topic) { "default topic is not configured" }
  52         return sendMessageNB(messageProducerProperties.topic!!, message)
  53     }
  54
  55     override suspend fun sendMessageNB(message: Any, headers: MutableMap<String, String>?): Boolean {
  56         checkNotNull(messageProducerProperties.topic) { "default topic is not configured" }
  57         return sendMessageNB(messageProducerProperties.topic!!, message, headers)
  58     }
  59
  60     override suspend fun sendMessageNB(
  61         topic: String,
  62         message: Any,
  63         headers: MutableMap<String, String>?
  64     ): Boolean {
  65         val byteArrayMessage = when (message) {
  66             is String -> message.toByteArray(Charset.defaultCharset())
  67             else -> message.asJsonString().toByteArray(Charset.defaultCharset())
  68         }
  69
  70         val record = ProducerRecord<String, ByteArray>(topic, defaultToUUID(), byteArrayMessage)
  71         val recordHeaders = record.headers()
  72         messageLoggerService.messageProducing(recordHeaders)
  73         headers?.let {
  74             headers.forEach { (key, value) -> recordHeaders.add(RecordHeader(key, value.toByteArray())) }
  75         }
  76         val callback = Callback { metadata, exception ->
  77             log.trace("message published to(${metadata.topic()}), offset(${metadata.offset()}), headers :$headers")
  78         }
  79         messageTemplate().send(record, callback)
  80         return true
  81     }
  82
  83     fun messageTemplate(additionalConfig: Map<String, ByteArray>? = null): KafkaProducer<String, ByteArray> {
  84         log.trace("Client Properties : ${ToStringBuilder.reflectionToString(messageProducerProperties)}")
  85         val configProps = hashMapOf<String, Any>()
  86         configProps[BOOTSTRAP_SERVERS_CONFIG] = messageProducerProperties.bootstrapServers
  87         configProps[KEY_SERIALIZER_CLASS_CONFIG] = StringSerializer::class.java
  88         configProps[VALUE_SERIALIZER_CLASS_CONFIG] = ByteArraySerializer::class.java
  89         configProps[ACKS_CONFIG] = messageProducerProperties.acks
  90         configProps[ENABLE_IDEMPOTENCE_CONFIG] = messageProducerProperties.enableIdempotence
  91         if (messageProducerProperties.clientId != null) {
  92             configProps[CLIENT_ID_CONFIG] = messageProducerProperties.clientId!!
  93         }
  94         // TODO("Security Implementation based on type")
  95
  96         // Add additional Properties
  97         if (additionalConfig != null) {
  98             configProps.putAll(additionalConfig)
  99         }
 100
 101         if (kafkaProducer == null) {
 102             kafkaProducer = KafkaProducer(configProps)
 103         }
 104         return kafkaProducer!!
 105     }
 106 }