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
53 * @deprecated This method is not needed anymore since the use of headers will be in place.
56 public void publishEvent(final String topicName, final String eventKey, final T event) {
57 final ListenableFuture<SendResult<String, T>> eventFuture = eventKafkaTemplate.send(topicName, eventKey, event);
58 eventFuture.addCallback(handleCallback(topicName));
62 * Generic Event Publisher with headers.
64 * @param topicName valid topic name
65 * @param eventKey message key
66 * @param eventHeaders event headers
67 * @param event message payload
69 public void publishEvent(final String topicName, final String eventKey, final Headers eventHeaders, final T event) {
71 final ProducerRecord<String, T> producerRecord =
72 new ProducerRecord<>(topicName, null, eventKey, event, eventHeaders);
73 final ListenableFuture<SendResult<String, T>> eventFuture = eventKafkaTemplate.send(producerRecord);
74 eventFuture.addCallback(handleCallback(topicName));
78 * Generic Event Publisher with headers.
80 * @param topicName valid topic name
81 * @param eventKey message key
82 * @param eventHeaders map of event headers
83 * @param event message payload
85 public void publishEvent(final String topicName, final String eventKey, final Map<String, Object> eventHeaders,
88 publishEvent(topicName, eventKey, convertToKafkaHeaders(eventHeaders), event);
91 private ListenableFutureCallback<SendResult<String, T>> handleCallback(final String topicName) {
92 return new ListenableFutureCallback<>() {
94 public void onFailure(final Throwable throwable) {
95 log.error("Unable to publish event to topic : {} due to {}", topicName, throwable.getMessage());
99 public void onSuccess(final SendResult<String, T> sendResult) {
100 log.debug("Successfully published event to topic : {} , Event : {}",
101 sendResult.getRecordMetadata().topic(), sendResult.getProducerRecord().value());
106 private Headers convertToKafkaHeaders(final Map<String, Object> eventMessageHeaders) {
107 final Headers eventHeaders = new RecordHeaders();
108 eventMessageHeaders.forEach((key, value) -> eventHeaders.add(key, SerializationUtils.serialize(value)));