7aa8206301602b2e1b48eda88321a52cad8f21d5
[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.lispflowmapping.lisp.type.MappingData;
17 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.MappingChange;
18 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.MappingChanged;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.MappingChangedBuilder;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.db.instance.Mapping;
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(Mapping input, Set<Subscriber> subscribers,
39             Set<Subscriber> dstSubscribers, MappingChange change) {
40         return new MappingChangedBuilder()
41                 .setMappingRecord(input.getMappingRecord())
42                 .setSubscriberItem(toSubscriberList(subscribers))
43                 .setDstSubscriberItem(toDstSubscriberList(dstSubscribers))
44                 .setChangeType(change).build();
45     }
46
47     public static MappingChanged toMappingChanged(MappingData mapping, Set<Subscriber> subscribers,
48             Set<Subscriber> dstSubscribers, MappingChange change) {
49         return new MappingChangedBuilder()
50                 .setMappingRecord(mapping.getRecord())
51                 .setSubscriberItem(toSubscriberList(subscribers))
52                 .setDstSubscriberItem(toDstSubscriberList(dstSubscribers))
53                 .setChangeType(change).build();
54     }
55
56     public static List<SubscriberItem> toSubscriberList(Set<Subscriber> subscribers) {
57         if (subscribers == null) {
58             return null;
59         }
60         List<SubscriberItem> subscriberList = new ArrayList<SubscriberItem>();
61         for (Subscriber subscriber : subscribers) {
62             subscriberList.add(new SubscriberItemBuilder().setSubscriberData(
63                     subscriber.getSubscriberData()).build());
64         }
65         return subscriberList;
66     }
67
68     public static List<DstSubscriberItem> toDstSubscriberList(Set<Subscriber> subscribers) {
69         if (subscribers == null) {
70             return null;
71         }
72         List<DstSubscriberItem> subscriberList = new ArrayList<DstSubscriberItem>();
73         for (Subscriber subscriber : subscribers) {
74             subscriberList.add(new DstSubscriberItemBuilder().setSubscriberData(
75                     subscriber.getSubscriberData()).build());
76         }
77         return subscriberList;
78     }
79
80     public static Set<Subscriber> toSubscriberSet(List<SubscriberItem> subscribers) {
81         if (subscribers == null) {
82             return null;
83         }
84         Set<Subscriber> subscriberSet = Sets.newConcurrentHashSet();
85         for (SubscriberItem subscriber : subscribers) {
86             subscriberSet.add(new Subscriber(subscriber.getSubscriberData()));
87         }
88         return subscriberSet;
89     }
90
91     public static Set<Subscriber> toSubscriberSetFromDst(List<DstSubscriberItem> subscribers) {
92         if (subscribers == null) {
93             return null;
94         }
95         Set<Subscriber> subscriberSet = Sets.newConcurrentHashSet();
96         for (DstSubscriberItem subscriber : subscribers) {
97             subscriberSet.add(new Subscriber(subscriber.getSubscriberData()));
98         }
99         return subscriberSet;
100     }
101 }