Bug 9116: SMR children of a prefix too
[lispflowmapping.git] / mappingservice / implementation / src / main / java / org / opendaylight / lispflowmapping / implementation / util / MSNotificationInputUtil.java
1 /*
2  * Copyright (c) 2015, 2017 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
9 package org.opendaylight.lispflowmapping.implementation.util;
10
11 import com.google.common.collect.Sets;
12 import java.util.ArrayList;
13 import java.util.List;
14 import java.util.Set;
15 import org.opendaylight.lispflowmapping.interfaces.dao.Subscriber;
16 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.eid.container.Eid;
17 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.container.MappingRecord;
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.MappingChanged;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.MappingChangedBuilder;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.mapping.changed.DstSubscriberItem;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.mapping.changed.DstSubscriberItemBuilder;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.mapping.changed.SubscriberItem;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.mapping.changed.SubscriberItemBuilder;
25
26 /**
27  * Utility class to convert a mapping change into a notification.
28  *
29  * @author Florin Coras
30  * @author Lorand Jakab
31  *
32  */
33 public final class MSNotificationInputUtil {
34     // Utility class, should not be instantiated
35     private MSNotificationInputUtil() {
36     }
37
38     public static MappingChanged toMappingChanged(MappingRecord mapping, Eid eid, Set<Subscriber> subscribers,
39                                                   Set<Subscriber> dstSubscribers, MappingChange change) {
40         return new MappingChangedBuilder()
41                 .setMappingRecord(mapping)
42                 .setEid(eid)
43                 .setSubscriberItem(toSubscriberList(subscribers))
44                 .setDstSubscriberItem(toDstSubscriberList(dstSubscribers))
45                 .setChangeType(change).build();
46     }
47
48     public static List<SubscriberItem> toSubscriberList(Set<Subscriber> subscribers) {
49         if (subscribers == null) {
50             return null;
51         }
52         List<SubscriberItem> subscriberList = new ArrayList<SubscriberItem>();
53         for (Subscriber subscriber : subscribers) {
54             subscriberList.add(new SubscriberItemBuilder().setSubscriberData(
55                     subscriber.getSubscriberData()).build());
56         }
57         return subscriberList;
58     }
59
60     public static List<DstSubscriberItem> toDstSubscriberList(Set<Subscriber> subscribers) {
61         if (subscribers == null) {
62             return null;
63         }
64         List<DstSubscriberItem> subscriberList = new ArrayList<DstSubscriberItem>();
65         for (Subscriber subscriber : subscribers) {
66             subscriberList.add(new DstSubscriberItemBuilder().setSubscriberData(
67                     subscriber.getSubscriberData()).build());
68         }
69         return subscriberList;
70     }
71
72     public static Set<Subscriber> toSubscriberSet(List<SubscriberItem> subscribers) {
73         if (subscribers == null) {
74             return null;
75         }
76         Set<Subscriber> subscriberSet = Sets.newConcurrentHashSet();
77         for (SubscriberItem subscriber : subscribers) {
78             subscriberSet.add(new Subscriber(subscriber.getSubscriberData()));
79         }
80         return subscriberSet;
81     }
82
83     public static Set<Subscriber> toSubscriberSetFromDst(List<DstSubscriberItem> subscribers) {
84         if (subscribers == null) {
85             return null;
86         }
87         Set<Subscriber> subscriberSet = Sets.newConcurrentHashSet();
88         for (DstSubscriberItem subscriber : subscribers) {
89             subscriberSet.add(new Subscriber(subscriber.getSubscriberData()));
90         }
91         return subscriberSet;
92     }
93 }