Bump upstreams for Silicon

[lispflowmapping.git] / mappingservice / implementation / src / main / java / org / opendaylight / lispflowmapping / implementation / util / DSBEInputUtil.java

diff --git a/mappingservice/implementation/src/main/java/org/opendaylight/lispflowmapping/implementation/util/DSBEInputUtil.java b/mappingservice/implementation/src/main/java/org/opendaylight/lispflowmapping/implementation/util/DSBEInputUtil.java
index 3870c07acf98d25d2985233bbc95820098831686..64b5778e98d74f38a9ac35cefd398c8c4fc2d7d2 100644 (file)
--- a/mappingservice/implementation/src/main/java/org/opendaylight/lispflowmapping/implementation/util/DSBEInputUtil.java
+++ b/mappingservice/implementation/src/main/java/org/opendaylight/lispflowmapping/implementation/util/DSBEInputUtil.java
@@ -5,51 +5,84 @@
  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
  * and is available at http://www.eclipse.org/legal/epl-v10.html
  */
-
 package org.opendaylight.lispflowmapping.implementation.util;
 
+import static java.util.Objects.requireNonNull;
+
+import edu.umd.cs.findbugs.annotations.Nullable;
 import java.util.Arrays;
 import java.util.List;
-
+import org.opendaylight.lispflowmapping.lisp.type.MappingData;
 import org.opendaylight.lispflowmapping.lisp.util.LispAddressStringifier;
-import org.opendaylight.lispflowmapping.lisp.util.MaskUtil;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.eidtolocatorrecords.EidToLocatorRecord;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.lispaddress.LispAddressContainer;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.SiteId;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.eid.container.Eid;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.authkey.container.MappingAuthkey;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.container.MappingRecord;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.container.MappingRecordBuilder;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.EidUri;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.MappingOrigin;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.SiteId;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.XtrIdUri;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.db.instance.AuthenticationKey;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.db.instance.AuthenticationKeyBuilder;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.db.instance.Mapping;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.db.instance.MappingBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.db.instance.mapping.XtrIdMapping;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.db.instance.mapping.XtrIdMappingBuilder;
 
 /**
- * @author Florin Coras
+ * DataStoreBackEnd utilities.
  *
+ * @author Florin Coras
  */
-public class DSBEInputUtil {
-    public static Mapping toMapping(MappingOrigin origin, LispAddressContainer key, SiteId siteId,
-            EidToLocatorRecord record) {
-        List<SiteId> siteIds = (siteId != null) ? Arrays.asList(siteId) : null;
-        return new MappingBuilder(record).setEid(new EidUri(LispAddressStringifier.getURIString(key)))
-                .setOrigin(origin).setSiteId(siteIds).build();
+public final class DSBEInputUtil {
+    // Utility class, should not be instantiated
+    private DSBEInputUtil() {
     }
 
-    public static Mapping toMapping(MappingOrigin origin, LispAddressContainer key) {
+    public static Mapping toMapping(MappingOrigin origin, Eid key, SiteId siteId, MappingData mappingData) {
+        MappingRecord record = mappingData != null ? mappingData.getRecord() : null;
+        List<SiteId> siteIds = siteId != null ? Arrays.asList(siteId) : null;
+        return new MappingBuilder()
+                    .setEidUri(new EidUri(LispAddressStringifier.getURIString(key)))
+                    .setOrigin(origin)
+                    .setSiteId(siteIds)
+                    .setMappingRecord(record).build();
+    }
+
+    public static Mapping toMapping(MappingOrigin origin, Eid key, @Nullable MappingData mappingData) {
+        MappingRecord record = mappingData != null ? mappingData.getRecord() :
+                                                        new MappingRecordBuilder().setEid(key).build();
+        SiteId siteId = record != null ? record.getSiteId() : null;
+        List<SiteId> siteIds = siteId != null ? Arrays.asList(siteId) : null;
+        return new MappingBuilder()
+                .setEidUri(new EidUri(LispAddressStringifier.getURIString(key)))
+                .setOrigin(origin)
+                .setSiteId(siteIds)
+                .setMappingRecord(record).build();
+    }
+
+    public static Mapping toMapping(MappingOrigin origin, Eid key) {
         MappingBuilder mb = new MappingBuilder();
-        mb.setEid(new EidUri(LispAddressStringifier.getURIString(key)));
+        mb.setEidUri(new EidUri(LispAddressStringifier.getURIString(key)));
         mb.setOrigin(origin);
-        mb.setMaskLength(MaskUtil.getMaskForAddress(key));
-        mb.setLispAddressContainer(key);
+        mb.setMappingRecord(new MappingRecordBuilder().setEid(key).build());
         return mb.build();
     }
 
-    public static AuthenticationKey toAuthenticationKey(LispAddressContainer key, String authKey) {
+    public static XtrIdMapping toXtrIdMapping(MappingData mappingData) {
+        requireNonNull(mappingData);
+        MappingRecord record = mappingData.getRecord();
+        requireNonNull(mappingData.getRecord());
+        return new XtrIdMappingBuilder()
+                    .setXtrIdUri(new XtrIdUri(LispAddressStringifier.getURIString(record.getXtrId())))
+                    .setMappingRecord(record).build();
+    }
+
+    public static AuthenticationKey toAuthenticationKey(Eid key, MappingAuthkey authKey) {
         AuthenticationKeyBuilder akb = new AuthenticationKeyBuilder();
-        akb.setEid(new EidUri(LispAddressStringifier.getURIString(key)));
-        akb.setLispAddressContainer(key);
-        akb.setMaskLength(MaskUtil.getMaskForAddress(key));
-        akb.setAuthkey(authKey);
+        akb.setEidUri(new EidUri(LispAddressStringifier.getURIString(key)));
+        akb.setEid(key);
+        akb.setMappingAuthkey(authKey);
         return akb.build();
     }
 }