2 * ============LICENSE_START=======================================================
3 * Copyright (C) 2022-2023 Nordix Foundation
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
9 * http://www.apache.org/licenses/LICENSE-2.0
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.
17 * SPDX-License-Identifier: Apache-2.0
18 * ============LICENSE_END=========================================================
21 package org.onap.cps.ncmp.api.impl.events;
24 import lombok.RequiredArgsConstructor;
25 import lombok.extern.slf4j.Slf4j;
26 import org.apache.kafka.clients.producer.ProducerRecord;
27 import org.apache.kafka.common.header.Headers;
28 import org.apache.kafka.common.header.internals.RecordHeaders;
29 import org.springframework.kafka.core.KafkaTemplate;
30 import org.springframework.kafka.support.SendResult;
31 import org.springframework.stereotype.Service;
32 import org.springframework.util.SerializationUtils;
33 import org.springframework.util.concurrent.ListenableFuture;
34 import org.springframework.util.concurrent.ListenableFutureCallback;
37 * EventsPublisher to publish events.
42 @RequiredArgsConstructor
43 public class EventsPublisher<T> {
45 private final KafkaTemplate<String, T> eventKafkaTemplate;
48 * Generic Event publisher.
50 * @param topicName valid topic name
51 * @param eventKey message key
52 * @param event message payload
55 public void publishEvent(final String topicName, final String eventKey, final T event) {
56 final ListenableFuture<SendResult<String, T>> eventFuture = eventKafkaTemplate.send(topicName, eventKey, event);
57 eventFuture.addCallback(handleCallback(topicName));
61 * Generic Event Publisher with headers.
63 * @param topicName valid topic name
64 * @param eventKey message key
65 * @param eventHeaders event headers
66 * @param event message payload
68 public void publishEvent(final String topicName, final String eventKey, final Headers eventHeaders, final T event) {
70 final ProducerRecord<String, T> producerRecord =
71 new ProducerRecord<>(topicName, null, eventKey, event, eventHeaders);
72 final ListenableFuture<SendResult<String, T>> eventFuture = eventKafkaTemplate.send(producerRecord);
73 eventFuture.addCallback(handleCallback(topicName));
77 * Generic Event Publisher with headers.
79 * @param topicName valid topic name
80 * @param eventKey message key
81 * @param eventHeaders map of event headers
82 * @param event message payload
84 public void publishEvent(final String topicName, final String eventKey, final Map<String, Object> eventHeaders,
87 publishEvent(topicName, eventKey, convertToKafkaHeaders(eventHeaders), event);
90 private ListenableFutureCallback<SendResult<String, T>> handleCallback(final String topicName) {
91 return new ListenableFutureCallback<>() {
93 public void onFailure(final Throwable throwable) {
94 log.error("Unable to publish event to topic : {} due to {}", topicName, throwable.getMessage());
98 public void onSuccess(final SendResult<String, T> sendResult) {
99 log.debug("Successfully published event to topic : {} , Event : {}",
100 sendResult.getRecordMetadata().topic(), sendResult.getProducerRecord().value());
105 private Headers convertToKafkaHeaders(final Map<String, Object> eventMessageHeaders) {
106 final Headers eventHeaders = new RecordHeaders();
107 eventMessageHeaders.forEach((key, value) -> eventHeaders.add(key, SerializationUtils.serialize(value)));