01d720937f533687adc76315ad490110b0ca37fa
[cps.git] /
1 /*
2  * ============LICENSE_START=======================================================
3  *  Copyright (C) 2024 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
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  *  SPDX-License-Identifier: Apache-2.0
18  *  ============LICENSE_END=========================================================
19  */
20
21 package org.onap.cps.ncmp.impl.cmnotificationsubscription.ncmp;
22
23 import io.cloudevents.CloudEvent;
24 import io.cloudevents.core.builder.CloudEventBuilder;
25 import java.net.URI;
26 import java.util.Map;
27 import java.util.UUID;
28 import java.util.concurrent.ConcurrentHashMap;
29 import java.util.concurrent.Executors;
30 import java.util.concurrent.ScheduledExecutorService;
31 import java.util.concurrent.ScheduledFuture;
32 import java.util.concurrent.TimeUnit;
33 import lombok.RequiredArgsConstructor;
34 import lombok.extern.slf4j.Slf4j;
35 import org.onap.cps.events.EventsPublisher;
36 import org.onap.cps.ncmp.impl.cmnotificationsubscription.cache.DmiCacheHandler;
37 import org.onap.cps.ncmp.impl.cmnotificationsubscription_1_0_0.ncmp_to_client.NcmpOutEvent;
38 import org.onap.cps.utils.JsonObjectMapper;
39 import org.springframework.beans.factory.annotation.Value;
40 import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
41 import org.springframework.stereotype.Component;
42
43 @Component
44 @Slf4j
45 @RequiredArgsConstructor
46 @ConditionalOnProperty(name = "notification.enabled", havingValue = "true", matchIfMissing = true)
47 public class NcmpOutEventProducer {
48
49     @Value("${app.ncmp.avc.cm-subscription-ncmp-out}")
50     private String ncmpOutEventTopic;
51
52     @Value("${ncmp.timers.subscription-forwarding.dmi-response-timeout-ms}")
53     private Integer dmiOutEventTimeoutInMs;
54
55     private final EventsPublisher<CloudEvent> eventsPublisher;
56     private final JsonObjectMapper jsonObjectMapper;
57     private final NcmpOutEventMapper ncmpOutEventMapper;
58     private final DmiCacheHandler dmiCacheHandler;
59     private final ScheduledExecutorService scheduledExecutorService = Executors.newSingleThreadScheduledExecutor();
60     private static final Map<String, ScheduledFuture<?>> scheduledTasksPerSubscriptionId = new ConcurrentHashMap<>();
61
62     /**
63      * Publish the event to the client who requested the subscription with key as subscription id and event is Cloud
64      * Event compliant.
65      *
66      * @param subscriptionId   Cm Subscription Id
67      * @param eventType        Type of event
68      * @param ncmpOutEvent     Cm Notification Subscription Event for the
69      *                         client
70      * @param isScheduledEvent Determines if the event is to be scheduled
71      *                         or published now
72      */
73     public void publishNcmpOutEvent(final String subscriptionId, final String eventType,
74             final NcmpOutEvent ncmpOutEvent, final boolean isScheduledEvent) {
75
76         if (isScheduledEvent && !scheduledTasksPerSubscriptionId.containsKey(subscriptionId)) {
77             final ScheduledFuture<?> scheduledFuture = scheduleAndPublishNcmpOutEvent(subscriptionId, eventType);
78             scheduledTasksPerSubscriptionId.putIfAbsent(subscriptionId, scheduledFuture);
79             log.debug("Scheduled the CmNotificationSubscriptionEvent for subscriptionId : {}", subscriptionId);
80         } else {
81             cancelScheduledTaskForSubscriptionId(subscriptionId);
82             publishNcmpOutEventNow(subscriptionId, eventType, ncmpOutEvent);
83             log.info("Published CmNotificationSubscriptionEvent on demand for subscriptionId : {}", subscriptionId);
84         }
85     }
86
87     private ScheduledFuture<?> scheduleAndPublishNcmpOutEvent(final String subscriptionId, final String eventType) {
88         final NcmpOutEventPublishingTask ncmpOutEventPublishingTask =
89                 new NcmpOutEventPublishingTask(ncmpOutEventTopic, subscriptionId, eventType, eventsPublisher,
90                         jsonObjectMapper, ncmpOutEventMapper, dmiCacheHandler);
91         return scheduledExecutorService.schedule(ncmpOutEventPublishingTask, dmiOutEventTimeoutInMs,
92                 TimeUnit.MILLISECONDS);
93     }
94
95     private void cancelScheduledTaskForSubscriptionId(final String subscriptionId) {
96
97         final ScheduledFuture<?> scheduledFuture = scheduledTasksPerSubscriptionId.get(subscriptionId);
98         if (scheduledFuture != null) {
99             scheduledFuture.cancel(true);
100             scheduledTasksPerSubscriptionId.remove(subscriptionId);
101         }
102
103     }
104
105
106     private void publishNcmpOutEventNow(final String subscriptionId, final String eventType,
107             final NcmpOutEvent ncmpOutEvent) {
108         final CloudEvent ncmpOutEventAsCloudEvent =
109                 buildAndGetNcmpOutEventAsCloudEvent(jsonObjectMapper, subscriptionId, eventType,
110                         ncmpOutEvent);
111         eventsPublisher.publishCloudEvent(ncmpOutEventTopic, subscriptionId,
112                 ncmpOutEventAsCloudEvent);
113         dmiCacheHandler.removeAcceptedAndRejectedDmiSubscriptionEntries(subscriptionId);
114     }
115
116     /**
117      * Get an NCMP out event as cloud event.
118      *
119      * @param jsonObjectMapper JSON object mapper
120      * @param subscriptionId   subscription id
121      * @param eventType        event type
122      * @param ncmpOutEvent     cm notification subscription NCMP out event
123      * @return cm notification subscription NCMP out event as cloud event
124      */
125     public static CloudEvent buildAndGetNcmpOutEventAsCloudEvent(final JsonObjectMapper jsonObjectMapper,
126             final String subscriptionId, final String eventType, final NcmpOutEvent ncmpOutEvent) {
127
128         return CloudEventBuilder.v1().withId(UUID.randomUUID().toString()).withType(eventType)
129                        .withSource(URI.create("NCMP")).withDataSchema(URI.create("org.onap.ncmp.cm.subscription:1.0.0"))
130                        .withExtension("correlationid", subscriptionId)
131                        .withData(jsonObjectMapper.asJsonBytes(ncmpOutEvent)).build();
132     }
133
134 }