Extension support - easy lookup for toOFJava
[openflowplugin.git] / extension / openflowplugin-extension-nicira / src / main / java / org / opendaylight / openflowplugin / extension / vendor / nicira / convertor / match / TunIpv4SrcConvertor.java
1 /**
2  * Copyright (c) 2014 Cisco Systems, Inc. and others.  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.openflowplugin.extension.vendor.nicira.convertor.match;
9
10 import org.opendaylight.openflowplugin.extension.api.ConvertorFromOFJava;
11 import org.opendaylight.openflowplugin.extension.api.ConvertorToOFJava;
12 import org.opendaylight.openflowplugin.extension.api.ExtensionAugment;
13 import org.opendaylight.openflowplugin.extension.api.path.MatchPath;
14 import org.opendaylight.openflowplugin.extension.vendor.nicira.convertor.CodecPreconditionException;
15 import org.opendaylight.openflowplugin.extension.vendor.nicira.convertor.IpConverter;
16 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.Ipv4Address;
17 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.oxm.rev130731.Nxm1Class;
18 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.oxm.rev130731.oxm.fields.grouping.MatchEntries;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowjava.nx.match.rev140421.OfjAugNxMatch;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowjava.nx.match.rev140421.OfjAugNxMatchBuilder;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowjava.nx.match.rev140421.ofj.nxm.nx.match.tun.ipv4.src.grouping.TunIpv4SrcValues;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowjava.nx.match.rev140421.ofj.nxm.nx.match.tun.ipv4.src.grouping.TunIpv4SrcValuesBuilder;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.general.rev140714.ExtensionKey;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.general.rev140714.general.extension.grouping.Extension;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.nicira.match.rev140714.NxAugMatchNotifPacketIn;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.nicira.match.rev140714.NxAugMatchNotifPacketInBuilder;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.nicira.match.rev140714.NxAugMatchNotifSwitchFlowRemoved;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.nicira.match.rev140714.NxAugMatchNotifSwitchFlowRemovedBuilder;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.nicira.match.rev140714.NxAugMatchNotifUpdateFlowStats;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.nicira.match.rev140714.NxAugMatchNotifUpdateFlowStatsBuilder;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.nicira.match.rev140714.NxmNxTunIpv4SrcGrouping;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.nicira.match.rev140714.NxmNxTunIpv4SrcKey;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.nicira.match.rev140714.nxm.nx.tun.ipv4.src.grouping.NxmNxTunIpv4Src;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.nicira.match.rev140714.nxm.nx.tun.ipv4.src.grouping.NxmNxTunIpv4SrcBuilder;
35 import org.opendaylight.yangtools.yang.binding.Augmentation;
36
37 import com.google.common.base.Optional;
38
39 /**
40  * @author msunal
41  *
42  */
43 public class TunIpv4SrcConvertor implements ConvertorToOFJava<MatchEntries>,
44         ConvertorFromOFJava<MatchEntries, MatchPath> {
45
46     /*
47      * (non-Javadoc)
48      * 
49      * @see
50      * org.opendaylight.openflowplugin.extension.api.ConvertorFromOFJava#convert
51      * (org.opendaylight.yangtools.yang.binding.DataContainer,
52      * org.opendaylight.openflowplugin.extension.api.path.AugmentationPath)
53      */
54     @Override
55     public ExtensionAugment<? extends Augmentation<Extension>> convert(MatchEntries input, MatchPath path) {
56         TunIpv4SrcValues values = input.getAugmentation(OfjAugNxMatch.class).getTunIpv4SrcValues();
57         Ipv4Address ipv4Address = IpConverter.longToIpv4Address(values.getValue());
58         return resolveAugmentation(new NxmNxTunIpv4SrcBuilder().setIpv4Address(ipv4Address).build(), path,
59                 NxmNxTunIpv4SrcKey.class);
60     }
61
62     private static ExtensionAugment<? extends Augmentation<Extension>> resolveAugmentation(NxmNxTunIpv4Src value,
63             MatchPath path, Class<? extends ExtensionKey> key) {
64         switch (path) {
65         case FLOWSSTATISTICSUPDATE_FLOWANDSTATISTICSMAPLIST_MATCH:
66             return new ExtensionAugment<>(NxAugMatchNotifUpdateFlowStats.class,
67                     new NxAugMatchNotifUpdateFlowStatsBuilder().setNxmNxTunIpv4Src(value).build(), key);
68         case PACKETRECEIVED_MATCH:
69             return new ExtensionAugment<>(NxAugMatchNotifPacketIn.class, new NxAugMatchNotifPacketInBuilder()
70                     .setNxmNxTunIpv4Src(value).build(), key);
71         case SWITCHFLOWREMOVED_MATCH:
72             return new ExtensionAugment<>(NxAugMatchNotifSwitchFlowRemoved.class,
73                     new NxAugMatchNotifSwitchFlowRemovedBuilder().setNxmNxTunIpv4Src(value).build(), key);
74         default:
75             throw new CodecPreconditionException(path);
76         }
77     }
78
79     /*
80      * (non-Javadoc)
81      * 
82      * @see
83      * org.opendaylight.openflowplugin.extension.api.ConvertorToOFJava#convert
84      * (org
85      * .opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.general
86      * .rev140714.general.extension.grouping.Extension)
87      */
88     @Override
89     public MatchEntries convert(Extension extension) {
90         Optional<NxmNxTunIpv4SrcGrouping> matchGrouping = MatchUtil.tunIpv4SrcResolver.getExtension(extension);
91         if (!matchGrouping.isPresent()) {
92             throw new CodecPreconditionException(extension);
93         }
94         long ipv4AddressAsLong = IpConverter.Ipv4AddressToLong(matchGrouping.get().getNxmNxTunIpv4Src()
95                 .getIpv4Address());
96         OfjAugNxMatchBuilder augNxMatchBuilder = new OfjAugNxMatchBuilder()
97                 .setTunIpv4SrcValues(new TunIpv4SrcValuesBuilder().setValue(ipv4AddressAsLong).build());
98         return MatchUtil
99                 .createNiciraMatchEntries(
100                         Nxm1Class.class,
101                         org.opendaylight.yang.gen.v1.urn.opendaylight.openflowjava.nx.match.rev140421.NxmNxTunIpv4Src.class,
102                         false, augNxMatchBuilder.build());
103     }
104
105 }