Bug 5228: More SMR fixes
[lispflowmapping.git] / mappingservice / implementation / src / main / java / org / opendaylight / lispflowmapping / implementation / mdsal / MappingDataListener.java
1 /*
2  * Copyright (c) 2015 Cisco Systems, Inc.  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.lispflowmapping.implementation.mdsal;
9
10 import java.util.Map;
11 import java.util.Set;
12
13 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
14 import org.opendaylight.controller.md.sal.binding.api.NotificationPublishService;
15 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeEvent;
16 import org.opendaylight.lispflowmapping.implementation.util.MSNotificationInputUtil;
17 import org.opendaylight.lispflowmapping.interfaces.mapcache.IMappingSystem;
18 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.MappingChange;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.MappingDatabase;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.MappingOrigin;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.db.instance.Mapping;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.mapping.database.VirtualNetworkIdentifier;
23 import org.opendaylight.yangtools.yang.binding.DataObject;
24 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
25 import org.slf4j.Logger;
26 import org.slf4j.LoggerFactory;
27
28 /**
29  * DataListener for all Mapping modification events.
30  *
31  * @author Lorand Jakab
32  * @author Florin Coras
33  *
34  */
35 public class MappingDataListener extends AbstractDataListener {
36     private static final Logger LOG = LoggerFactory.getLogger(MappingDataListener.class);
37     private IMappingSystem mapSystem;
38     private NotificationPublishService notificationPublishService;
39
40     public MappingDataListener(DataBroker broker, IMappingSystem msmr, NotificationPublishService nps) {
41         setBroker(broker);
42         setMappingSystem(msmr);
43         setNotificationProviderService(nps);
44         setPath(InstanceIdentifier.create(MappingDatabase.class).child(VirtualNetworkIdentifier.class)
45                 .child(Mapping.class));
46         LOG.trace("Registering Mapping listener.");
47         registerDataChangeListener();
48     }
49
50     public void setNotificationProviderService(NotificationPublishService nps) {
51         this.notificationPublishService = nps;
52     }
53
54     @Override
55     public void onDataChanged(
56             AsyncDataChangeEvent<InstanceIdentifier<?>, DataObject> change) {
57
58      // Process newly created mappings
59         Map<InstanceIdentifier<?>, DataObject> createdData = change.getCreatedData();
60         for (Map.Entry<InstanceIdentifier<?>, DataObject> entry : createdData.entrySet()) {
61             if (entry.getValue() instanceof Mapping) {
62                 Mapping mapping = (Mapping)entry.getValue();
63
64                 LOG.trace("Received created data");
65                 LOG.trace("Key: {}", entry.getKey());
66                 LOG.trace("Value: {}", mapping);
67
68                 mapSystem.addMapping(mapping.getOrigin(), mapping.getMappingRecord().getEid(),
69                         mapping.getMappingRecord());
70             }
71         }
72
73         // Process updated mappings
74         Map<InstanceIdentifier<?>, DataObject> updatedData = change.getUpdatedData();
75         for (Map.Entry<InstanceIdentifier<?>, DataObject> entry : updatedData.entrySet()) {
76             if (entry.getValue() instanceof Mapping) {
77                 Mapping mapping = (Mapping)entry.getValue();
78
79                 LOG.trace("Received changed data");
80                 LOG.trace("Key: {}", entry.getKey());
81                 LOG.trace("Value: {}", entry.getValue());
82
83                 mapSystem.addMapping(mapping.getOrigin(), mapping.getMappingRecord().getEid(),
84                         mapping.getMappingRecord());
85
86                 if (mapping.getOrigin() == MappingOrigin.Northbound) {
87                     // Only publish notifications for mapping changes caused by Northbound, since Southbound has a
88                     // dedicated code path for detecting mapping updates. The notifications are used for sending SMR.
89                     try {
90                         notificationPublishService.putNotification(MSNotificationInputUtil.toMappingChanged(mapping,
91                                 MappingChange.Updated));
92                     } catch (InterruptedException e) {
93                         LOG.warn("Notification publication interrupted!");
94                     }
95                 }
96             }
97         }
98
99         // Process deleted mappings
100         Set<InstanceIdentifier<?>> removedData = change.getRemovedPaths();
101         for (InstanceIdentifier<?> entry : removedData) {
102             DataObject dataObject = change.getOriginalData().get(entry);
103             if (dataObject instanceof Mapping) {
104                 Mapping mapping = (Mapping)dataObject;
105
106                 LOG.trace("Received deleted data");
107                 LOG.trace("Key: {}", entry);
108                 LOG.trace("Value: {}", dataObject);
109
110                 mapSystem.removeMapping(mapping.getOrigin(), mapping.getMappingRecord().getEid());
111                 try {
112                     notificationPublishService.putNotification(MSNotificationInputUtil.toMappingChanged(mapping, MappingChange.Removed));
113                 } catch (InterruptedException e) {
114                     LOG.warn("Notification publication interrupted!");
115                 }
116             }
117         }
118     }
119
120     void setMappingSystem(IMappingSystem msmr) {
121         this.mapSystem = msmr;
122     }
123 }