aboutsummaryrefslogtreecommitdiffstats
path: root/cps-ncmp-service/src/main/java/org/onap/cps/ncmp/api/impl/events/EventsPublisher.java
blob: 4c9064852169cb4229e4be7766e91fe737f362ed (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
/*
 * ============LICENSE_START=======================================================
 * Copyright (C) 2022-2023 Nordix Foundation
 * ================================================================================
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 * SPDX-License-Identifier: Apache-2.0
 * ============LICENSE_END=========================================================
 */

package org.onap.cps.ncmp.api.impl.events;

import io.cloudevents.CloudEvent;
import java.util.Map;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.apache.kafka.clients.producer.ProducerRecord;
import org.apache.kafka.common.header.Headers;
import org.apache.kafka.common.header.internals.RecordHeaders;
import org.springframework.kafka.core.KafkaTemplate;
import org.springframework.kafka.support.SendResult;
import org.springframework.stereotype.Service;
import org.springframework.util.SerializationUtils;
import org.springframework.util.concurrent.ListenableFuture;
import org.springframework.util.concurrent.ListenableFutureCallback;

/**
 * EventsPublisher to publish events.
 */

@Slf4j
@Service
@RequiredArgsConstructor
public class EventsPublisher<T> {

    /**
     * KafaTemplate for legacy (non-cloud) events.
     *
     * @deprecated Cloud events should be used. Will address soon as part of  https://jira.onap.org/browse/CPS-1717
     */
    @Deprecated(forRemoval = true)
    private final KafkaTemplate<String, T> legacyKafkaEventTemplate;

    private final KafkaTemplate<String, CloudEvent> cloudEventKafkaTemplate;

    /**
     * Generic CloudEvent publisher.
     *
     * @param topicName valid topic name
     * @param eventKey  message key
     * @param event     message payload
     */
    public void publishCloudEvent(final String topicName, final String eventKey, final CloudEvent event) {
        final ListenableFuture<SendResult<String, CloudEvent>> eventFuture
                = cloudEventKafkaTemplate.send(topicName, eventKey, event);
        eventFuture.addCallback(handleCallback(topicName));
    }

    /**
     * Generic Event publisher.
     *
     * @param topicName valid topic name
     * @param eventKey  message key
     * @param event     message payload
     * @deprecated Cloud events should be used. Will address soon as part of  https://jira.onap.org/browse/CPS-1717
     */
    @Deprecated(forRemoval = true)
    public void publishEvent(final String topicName, final String eventKey, final T event) {
        final ListenableFuture<SendResult<String, T>> eventFuture
                = legacyKafkaEventTemplate.send(topicName, eventKey, event);
        eventFuture.addCallback(handleCallback(topicName));
    }

    /**
     * Generic Event Publisher with headers.
     *
     * @param topicName    valid topic name
     * @param eventKey     message key
     * @param eventHeaders event headers
     * @param event        message payload
     */
    public void publishEvent(final String topicName, final String eventKey, final Headers eventHeaders, final T event) {

        final ProducerRecord<String, T> producerRecord =
                new ProducerRecord<>(topicName, null, eventKey, event, eventHeaders);
        final ListenableFuture<SendResult<String, T>> eventFuture = legacyKafkaEventTemplate.send(producerRecord);
        eventFuture.addCallback(handleCallback(topicName));
    }

    /**
     * Generic Event Publisher with headers.
     *
     * @param topicName    valid topic name
     * @param eventKey     message key
     * @param eventHeaders map of event headers
     * @param event        message payload
     */
    public void publishEvent(final String topicName, final String eventKey, final Map<String, Object> eventHeaders,
            final T event) {

        publishEvent(topicName, eventKey, convertToKafkaHeaders(eventHeaders), event);
    }

    private ListenableFutureCallback<SendResult<String, ?>> handleCallback(final String topicName) {
        return new ListenableFutureCallback<>() {
            @Override
            public void onFailure(final Throwable throwable) {
                log.error("Unable to publish event to topic : {} due to {}", topicName, throwable.getMessage());
            }

            @Override
            public void onSuccess(final SendResult<String, ?> sendResult) {
                log.debug("Successfully published event to topic : {} , Event : {}",
                        sendResult.getRecordMetadata().topic(), sendResult.getProducerRecord().value());
            }
        };
    }

    private Headers convertToKafkaHeaders(final Map<String, Object> eventMessageHeaders) {
        final Headers eventHeaders = new RecordHeaders();
        eventMessageHeaders.forEach((key, value) -> eventHeaders.add(key, SerializationUtils.serialize(value)));
        return eventHeaders;
    }

}