f243989e4292654b09cf1f259cc109b62db17100
[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.rev211013.NotificationAlarmService;
16 import org.opendaylight.yang.gen.v1.nbi.notifications.rev211013.NotificationAlarmServiceBuilder;
17 import org.opendaylight.yang.gen.v1.nbi.notifications.rev211013.NotificationProcessService;
18 import org.opendaylight.yang.gen.v1.nbi.notifications.rev211013.NotificationProcessServiceBuilder;
19 import org.opendaylight.yang.gen.v1.nbi.notifications.rev211013.NotificationTapiService;
20 import org.opendaylight.yang.gen.v1.nbi.notifications.rev211013.NotificationTapiServiceBuilder;
21 import org.opendaylight.yang.gen.v1.nbi.notifications.rev211013.PublishNotificationAlarmService;
22 import org.opendaylight.yang.gen.v1.nbi.notifications.rev211013.PublishNotificationProcessService;
23 import org.opendaylight.yang.gen.v1.nbi.notifications.rev211013.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         publisher.sendEvent(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                         .build(), notification.getConnectionType().getName());
72     }
73
74     void onPublishNotificationAlarmService(PublishNotificationAlarmService notification) {
75         LOG.info("Receiving request for publishing notification alarm service");
76         String publisherName = notification.getPublisherName();
77         if (!publishersAlarmMap.containsKey(publisherName)) {
78             LOG.error("Unknown topic {}", publisherName);
79             return;
80         }
81         Publisher<NotificationAlarmService> publisherAlarm = publishersAlarmMap.get(publisherName);
82         publisherAlarm.sendEvent(new NotificationAlarmServiceBuilder()
83                 .setConnectionType(notification.getConnectionType())
84                 .setMessage(notification.getMessage())
85                 .setOperationalState(notification.getOperationalState())
86                 .setServiceName(notification.getServiceName())
87                         .build(), "alarm" + notification.getConnectionType().getName());
88     }
89
90     void onPublishTapiNotificationService(PublishTapiNotificationService notification) {
91         LOG.info("Receiving request for publishing TAPI notification");
92         String topic = notification.getTopic();
93         if (!tapiPublisherMap.containsKey(topic)) {
94             LOG.error("Unknown topic {}", topic);
95             return;
96         }
97         Publisher<NotificationTapiService> publisher = tapiPublisherMap.get(topic);
98         publisher.sendEvent(new NotificationTapiServiceBuilder(
99                 NbiNotificationsUtils.transformTapiNotification(notification)).build(), topic);
100     }
101
102     public void setPublishersServiceMap(Map<String, Publisher<NotificationProcessService>> publishersServiceMap) {
103         this.publishersServiceMap = publishersServiceMap;
104     }
105
106     public void setPublishersAlarmMap(Map<String, Publisher<NotificationAlarmService>> publishersAlarmMap) {
107         this.publishersAlarmMap = publishersAlarmMap;
108     }
109
110     public void setTapiPublishersMap(Map<String, Publisher<NotificationTapiService>> tapiPublishersMap) {
111         this.tapiPublisherMap = tapiPublishersMap;
112     }
113
114     public Publisher<NotificationTapiService> getTapiPublisherFromTopic(String topic) {
115         return this.tapiPublisherMap.get(topic);
116     }
117 }