catalog-be/src/main/java/org/openecomp/sdc/be/components/kafka/SdcKafkaProducer.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * SDC
   4  * ================================================================================
   5  * Copyright (C) 2022 Nordix Foundation. All rights reserved.
   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.openecomp.sdc.be.components.kafka;
  21
  22 import com.google.common.annotations.VisibleForTesting;
  23 import java.util.Properties;
  24 import java.util.UUID;
  25 import org.apache.kafka.clients.CommonClientConfigs;
  26 import org.apache.kafka.clients.producer.KafkaProducer;
  27 import org.apache.kafka.clients.producer.ProducerConfig;
  28 import org.apache.kafka.clients.producer.ProducerRecord;
  29 import org.apache.kafka.common.KafkaException;
  30 import org.apache.kafka.common.config.SaslConfigs;
  31 import org.openecomp.sdc.be.config.DistributionEngineConfiguration;
  32 import org.slf4j.Logger;
  33 import org.slf4j.LoggerFactory;
  34
  35 /**
  36  * Utility class that provides a KafkaProducer to communicate with a kafka cluster
  37  */
  38 public class SdcKafkaProducer {
  39     private static final Logger log = LoggerFactory.getLogger(SdcKafkaProducer.class.getName());
  40
  41     private KafkaProducer<String, String> kafkaProducer;
  42
  43     /**
  44      * Constructor setting up the KafkaProducer from a predefined set of configurations
  45      */
  46     public SdcKafkaProducer(DistributionEngineConfiguration deConfiguration) {
  47         log.info("Create SdcKafkaProducer via constructor");
  48         Properties properties = new Properties();
  49
  50         properties.put(ProducerConfig.CLIENT_ID_CONFIG, deConfiguration.getDistributionStatusTopic().getConsumerId() + "-producer-" + UUID.randomUUID());
  51         properties.put(ProducerConfig.VALUE_SERIALIZER_CLASS_CONFIG, "org.apache.kafka.common.serialization.StringSerializer");
  52         properties.put(ProducerConfig.KEY_SERIALIZER_CLASS_CONFIG,  "org.apache.kafka.common.serialization.StringSerializer");
  53         properties.put(CommonClientConfigs.BOOTSTRAP_SERVERS_CONFIG, deConfiguration.getKafkaBootStrapServers());
  54         properties.put(CommonClientConfigs.SECURITY_PROTOCOL_CONFIG, "SASL_PLAINTEXT");
  55         properties.put(SaslConfigs.SASL_JAAS_CONFIG, getKafkaSaslJaasConfig());
  56         properties.put(SaslConfigs.SASL_MECHANISM, "SCRAM-SHA-512");
  57         kafkaProducer = new KafkaProducer<>(properties);
  58     }
  59
  60     /**
  61      *
  62      * @param kafkaProducer Setting a KafkaProducer to use within the sdcKafkaProducer class
  63      */
  64     @VisibleForTesting
  65     SdcKafkaProducer(KafkaProducer kafkaProducer) {
  66         this.kafkaProducer = kafkaProducer;
  67     }
  68
  69     /**
  70      * @return The Sasl Jaas Configuration
  71      */
  72     private static String getKafkaSaslJaasConfig() throws KafkaException {
  73         String saslJaasConfFromEnv = System.getenv("SASL_JAAS_CONFIG");
  74         if(saslJaasConfFromEnv != null) {
  75             return saslJaasConfFromEnv;
  76         } else {
  77             throw new KafkaException("sasl.jaas.config not set for Kafka Consumer");
  78         }
  79     }
  80
  81     /**
  82      * @param message A message to Send
  83      * @param topicName The name of the topic to publish to
  84      * @return The status of the send request
  85      */
  86     public void send(String message, String topicName) throws KafkaException {
  87         ProducerRecord<String, String> kafkaMessagePayload = new ProducerRecord<>(topicName, "PartitionKey", message);
  88         kafkaProducer.send(kafkaMessagePayload);
  89     }
  90
  91     /**
  92      * Kafka FLush operation
  93      */
  94     public void flush() throws KafkaException {
  95         log.info("SdcKafkaProducer - flush");
  96         kafkaProducer.flush();
  97     }
  98 }