Bug 7947: Store MappingOrigin in MappingData
[lispflowmapping.git] / mappingservice / implementation / src / main / java / org / opendaylight / lispflowmapping / implementation / util / DSBEInputUtil.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
9 package org.opendaylight.lispflowmapping.implementation.util;
10
11 import com.google.common.base.Preconditions;
12 import java.util.Arrays;
13 import java.util.List;
14 import javax.annotation.Nullable;
15 import org.opendaylight.lispflowmapping.lisp.util.LispAddressStringifier;
16 import org.opendaylight.lispflowmapping.type.MappingData;
17 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.SiteId;
18 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.eid.container.Eid;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.authkey.container.MappingAuthkey;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.container.MappingRecord;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.container.MappingRecordBuilder;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.EidUri;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.MappingOrigin;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.XtrIdUri;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.db.instance.AuthenticationKey;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.db.instance.AuthenticationKeyBuilder;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.db.instance.Mapping;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.db.instance.MappingBuilder;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.db.instance.mapping.XtrIdMapping;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.db.instance.mapping.XtrIdMappingBuilder;
31
32 /**
33  * DataStoreBackEnd utilities.
34  *
35  * @author Florin Coras
36  *
37  */
38 public final class DSBEInputUtil {
39     // Utility class, should not be instantiated
40     private DSBEInputUtil() {
41     }
42
43     public static Mapping toMapping(MappingOrigin origin, Eid key, SiteId siteId, MappingData mappingData) {
44         MappingRecord record = (mappingData != null) ? mappingData.getRecord() : null;
45         List<SiteId> siteIds = (siteId != null) ? Arrays.asList(siteId) : null;
46         return new MappingBuilder()
47                     .setEidUri(new EidUri(LispAddressStringifier.getURIString(key)))
48                     .setOrigin(origin)
49                     .setSiteId(siteIds)
50                     .setMappingRecord(record).build();
51     }
52
53     public static Mapping toMapping(MappingOrigin origin, Eid key, @Nullable MappingData mappingData) {
54         MappingRecord record = (mappingData != null) ? mappingData.getRecord() :
55                                                         new MappingRecordBuilder().setEid(key).build();
56         SiteId siteId = (record != null) ? record.getSiteId() : null;
57         List<SiteId> siteIds = (siteId != null) ? Arrays.asList(siteId) : null;
58         return new MappingBuilder()
59                 .setEidUri(new EidUri(LispAddressStringifier.getURIString(key)))
60                 .setOrigin(origin)
61                 .setSiteId(siteIds)
62                 .setMappingRecord(record).build();
63     }
64
65     public static Mapping toMapping(MappingOrigin origin, Eid key) {
66         MappingBuilder mb = new MappingBuilder();
67         mb.setEidUri(new EidUri(LispAddressStringifier.getURIString(key)));
68         mb.setOrigin(origin);
69         mb.setMappingRecord(new MappingRecordBuilder().setEid(key).build());
70         return mb.build();
71     }
72
73     public static XtrIdMapping toXtrIdMapping(MappingData mappingData) {
74         Preconditions.checkNotNull(mappingData);
75         MappingRecord record = mappingData.getRecord();
76         Preconditions.checkNotNull(mappingData.getRecord());
77         return new XtrIdMappingBuilder()
78                     .setXtrIdUri(new XtrIdUri(LispAddressStringifier.getURIString(record.getXtrId())))
79                     .setMappingRecord(record).build();
80     }
81
82     public static AuthenticationKey toAuthenticationKey(Eid key, MappingAuthkey authKey) {
83         AuthenticationKeyBuilder akb = new AuthenticationKeyBuilder();
84         akb.setEidUri(new EidUri(LispAddressStringifier.getURIString(key)));
85         akb.setEid(key);
86         akb.setMappingAuthkey(authKey);
87         return akb.build();
88     }
89 }