af6a1ce4ddce749de37b6cd7f29d0b3440bc44fd
[transportpce.git] / nbinotifications / src / main / java / org / opendaylight / transportpce / nbinotifications / listener / NbiNotificationsHandler.java
1 /*
2  * Copyright © 2020 Orange, Inc. and others.  All rights reserved.
3  *
4  * This program and the accompanying materials are made available under the
5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6  * and is available at http://www.eclipse.org/legal/epl-v10.html
7  */
8 package org.opendaylight.transportpce.nbinotifications.listener;
9
10 import java.util.Map;
11 import java.util.Set;
12 import org.opendaylight.mdsal.binding.api.NotificationService.CompositeListener;
13 import org.opendaylight.transportpce.nbinotifications.producer.Publisher;
14 import org.opendaylight.transportpce.nbinotifications.utils.NbiNotificationsUtils;
15 import org.opendaylight.yang.gen.v1.nbi.notifications.rev230726.NotificationAlarmService;
16 import org.opendaylight.yang.gen.v1.nbi.notifications.rev230726.NotificationAlarmServiceBuilder;
17 import org.opendaylight.yang.gen.v1.nbi.notifications.rev230726.NotificationProcessService;
18 import org.opendaylight.yang.gen.v1.nbi.notifications.rev230726.NotificationProcessServiceBuilder;
19 import org.opendaylight.yang.gen.v1.nbi.notifications.rev230726.NotificationTapiService;
20 import org.opendaylight.yang.gen.v1.nbi.notifications.rev230726.NotificationTapiServiceBuilder;
21 import org.opendaylight.yang.gen.v1.nbi.notifications.rev230726.PublishNotificationAlarmService;
22 import org.opendaylight.yang.gen.v1.nbi.notifications.rev230726.PublishNotificationProcessService;
23 import org.opendaylight.yang.gen.v1.nbi.notifications.rev230726.PublishTapiNotificationService;
24 import org.opendaylight.yangtools.concepts.Registration;
25 import org.slf4j.Logger;
26 import org.slf4j.LoggerFactory;
27
28 public class NbiNotificationsHandler {
29     private static final Logger LOG = LoggerFactory.getLogger(NbiNotificationsHandler.class);
30     private Map<String, Publisher<NotificationProcessService>> publishersServiceMap;
31     private Map<String, Publisher<NotificationAlarmService>> publishersAlarmMap;
32     private Map<String, Publisher<NotificationTapiService>> tapiPublisherMap;
33     private Registration reg;
34
35     public NbiNotificationsHandler(Map<String, Publisher<NotificationProcessService>> publishersServiceMap,
36                                         Map<String, Publisher<NotificationAlarmService>> publishersAlarmMap,
37                                         Map<String, Publisher<NotificationTapiService>> tapiPublisherMap) {
38         this.publishersServiceMap = publishersServiceMap;
39         this.publishersAlarmMap = publishersAlarmMap;
40         this.tapiPublisherMap = tapiPublisherMap;
41     }
42
43     public CompositeListener getCompositeListener() {
44         return new CompositeListener(Set.of(
45             new CompositeListener.Component<>(
46                 PublishNotificationProcessService.class, this::onPublishNotificationProcessService),
47             new CompositeListener.Component<>(
48                 PublishNotificationAlarmService.class, this::onPublishNotificationAlarmService),
49             new CompositeListener.Component<>(
50                 PublishTapiNotificationService.class, this::onPublishTapiNotificationService)
51         ));
52     }
53
54     void onPublishNotificationProcessService(PublishNotificationProcessService notification) {
55         LOG.info("Receiving request for publishing notification service");
56         String publisherName = notification.getPublisherName();
57         if (!publishersServiceMap.containsKey(publisherName)) {
58             LOG.error("Unknown publisher {}", publisherName);
59             return;
60         }
61         Publisher<NotificationProcessService> publisher = publishersServiceMap.get(publisherName);
62         NotificationProcessServiceBuilder notificationProcessServiceBuilder = new NotificationProcessServiceBuilder()
63                 .setCommonId(notification.getCommonId())
64                 .setConnectionType(notification.getConnectionType())
65                 .setMessage(notification.getMessage())
66                 .setOperationalState(notification.getOperationalState())
67                 .setResponseFailed(notification.getResponseFailed())
68                 .setServiceAEnd(notification.getServiceAEnd())
69                 .setServiceName(notification.getServiceName())
70                 .setServiceZEnd(notification.getServiceZEnd());
71
72         if (notification.getIsTempService()) {
73             // A-to-Z and Z-to-A containers are only needed for temp-service-create notification
74             publisher.sendEvent(notificationProcessServiceBuilder
75                     .setAToZ(notification.getAToZ())
76                     .setZToA(notification.getZToA())
77                     .build(), notification.getConnectionType().getName());
78         } else {
79             publisher.sendEvent(notificationProcessServiceBuilder
80                     .build(), notification.getConnectionType().getName());
81         }
82     }
83
84     void onPublishNotificationAlarmService(PublishNotificationAlarmService notification) {
85         LOG.info("Receiving request for publishing notification alarm service");
86         String publisherName = notification.getPublisherName();
87         if (!publishersAlarmMap.containsKey(publisherName)) {
88             LOG.error("Unknown topic {}", publisherName);
89             return;
90         }
91         Publisher<NotificationAlarmService> publisherAlarm = publishersAlarmMap.get(publisherName);
92         publisherAlarm.sendEvent(new NotificationAlarmServiceBuilder()
93                 .setConnectionType(notification.getConnectionType())
94                 .setMessage(notification.getMessage())
95                 .setOperationalState(notification.getOperationalState())
96                 .setServiceName(notification.getServiceName())
97                         .build(), "alarm" + notification.getConnectionType().getName());
98     }
99
100     void onPublishTapiNotificationService(PublishTapiNotificationService notification) {
101         LOG.info("Receiving request for publishing TAPI notification");
102         String topic = notification.getTopic();
103         if (!tapiPublisherMap.containsKey(topic)) {
104             LOG.error("Unknown topic {}", topic);
105             return;
106         }
107         Publisher<NotificationTapiService> publisher = tapiPublisherMap.get(topic);
108         publisher.sendEvent(new NotificationTapiServiceBuilder(
109                 NbiNotificationsUtils.transformTapiNotification(notification)).build(), topic);
110     }
111
112     public void setPublishersServiceMap(Map<String, Publisher<NotificationProcessService>> publishersServiceMap) {
113         this.publishersServiceMap = publishersServiceMap;
114     }
115
116     public void setPublishersAlarmMap(Map<String, Publisher<NotificationAlarmService>> publishersAlarmMap) {
117         this.publishersAlarmMap = publishersAlarmMap;
118     }
119
120     public void setTapiPublishersMap(Map<String, Publisher<NotificationTapiService>> tapiPublishersMap) {
121         this.tapiPublisherMap = tapiPublishersMap;
122     }
123
124     public Publisher<NotificationTapiService> getTapiPublisherFromTopic(String topic) {
125         return this.tapiPublisherMap.get(topic);
126     }
127 }