Refactor nsh fields to new encoding
[openflowplugin.git] / extension / openflowplugin-extension-nicira / src / main / java / org / opendaylight / openflowplugin / extension / vendor / nicira / convertor / match / PktMarkConvertor.java
1 /*
2  * Copyright (c) 2018 Ericsson India Global Services Pvt Ltd. 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
9 package org.opendaylight.openflowplugin.extension.vendor.nicira.convertor.match;
10
11 import com.google.common.base.Optional;
12 import org.opendaylight.openflowplugin.extension.api.ConvertorFromOFJava;
13 import org.opendaylight.openflowplugin.extension.api.ConvertorToOFJava;
14 import org.opendaylight.openflowplugin.extension.api.ExtensionAugment;
15 import org.opendaylight.openflowplugin.extension.api.path.MatchPath;
16 import org.opendaylight.openflowplugin.extension.vendor.nicira.convertor.CodecPreconditionException;
17 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.oxm.rev150225.Nxm1Class;
18 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.oxm.rev150225.match.entries.grouping.MatchEntry;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.oxm.rev150225.match.entries.grouping.MatchEntryBuilder;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowjava.nx.match.rev140421.ofj.nxm.nx.match.pkt.mark.grouping.PktMarkValuesBuilder;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowjava.nx.match.rev140421.oxm.container.match.entry.value.PktMarkCaseValue;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowjava.nx.match.rev140421.oxm.container.match.entry.value.PktMarkCaseValueBuilder;
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.NxAugMatchNodesNodeTableFlow;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.nicira.match.rev140714.NxAugMatchNodesNodeTableFlowBuilder;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.nicira.match.rev140714.NxAugMatchNotifPacketIn;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.nicira.match.rev140714.NxAugMatchNotifPacketInBuilder;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.nicira.match.rev140714.NxAugMatchNotifSwitchFlowRemoved;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.nicira.match.rev140714.NxAugMatchNotifSwitchFlowRemovedBuilder;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.nicira.match.rev140714.NxAugMatchPacketInMessage;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.nicira.match.rev140714.NxAugMatchPacketInMessageBuilder;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.nicira.match.rev140714.NxAugMatchRpcGetFlowStats;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.nicira.match.rev140714.NxAugMatchRpcGetFlowStatsBuilder;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.nicira.match.rev140714.NxmNxPktMarkGrouping;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.nicira.match.rev140714.NxmNxPktMarkKey;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.nicira.match.rev140714.nxm.nx.pkt.mark.grouping.NxmNxPktMark;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.nicira.match.rev140714.nxm.nx.pkt.mark.grouping.NxmNxPktMarkBuilder;
39 import org.opendaylight.yangtools.yang.binding.Augmentation;
40
41 public class PktMarkConvertor implements ConvertorToOFJava<MatchEntry>, ConvertorFromOFJava<MatchEntry, MatchPath> {
42
43     @Override
44     public MatchEntry convert(Extension extension) {
45         Optional<NxmNxPktMarkGrouping> matchGrouping = MatchUtil.PKT_MARK_RESOLVER.getExtension(extension);
46         if (!matchGrouping.isPresent()) {
47             throw new CodecPreconditionException(extension);
48         }
49         PktMarkCaseValueBuilder pktMarkCaseValueBuilder = new PktMarkCaseValueBuilder();
50         PktMarkValuesBuilder pktMarkValuesBuilder = new PktMarkValuesBuilder();
51         pktMarkValuesBuilder.setPktMark(matchGrouping.get().getNxmNxPktMark().getPktMark());
52         pktMarkValuesBuilder.setMask(matchGrouping.get().getNxmNxPktMark().getMask());
53         pktMarkCaseValueBuilder.setPktMarkValues(pktMarkValuesBuilder.build());
54         MatchEntryBuilder ofMatch = MatchUtil
55                 .createDefaultMatchEntryBuilder(org.opendaylight.yang.gen.v1.urn
56                                 .opendaylight.openflowjava.nx.match.rev140421.NxmNxPktMark.class,
57                         Nxm1Class.class, pktMarkCaseValueBuilder.build());
58         return ofMatch.build();
59     }
60
61     @Override
62     public ExtensionAugment<? extends Augmentation<Extension>> convert(MatchEntry input, MatchPath path) {
63         PktMarkCaseValue pktMarkCaseValue = (PktMarkCaseValue) input.getMatchEntryValue();
64         NxmNxPktMarkBuilder pktMarkBuilder = new NxmNxPktMarkBuilder();
65         pktMarkBuilder.setPktMark(pktMarkCaseValue.getPktMarkValues().getPktMark());
66         pktMarkBuilder.setMask(pktMarkCaseValue.getPktMarkValues().getMask());
67         return resolveAugmentation(pktMarkBuilder.build(), path,
68                 NxmNxPktMarkKey.class);
69     }
70
71     private static ExtensionAugment<? extends Augmentation<Extension>> resolveAugmentation(NxmNxPktMark value,
72                                                                    MatchPath path, Class<? extends ExtensionKey> key) {
73         switch (path) {
74             case FLOWS_STATISTICS_UPDATE_MATCH:
75                 return new ExtensionAugment<>(NxAugMatchNodesNodeTableFlow.class,
76                         new NxAugMatchNodesNodeTableFlowBuilder().setNxmNxPktMark(value).build(), key);
77             case FLOWS_STATISTICS_RPC_MATCH:
78                 return new ExtensionAugment<>(NxAugMatchRpcGetFlowStats.class,
79                         new NxAugMatchRpcGetFlowStatsBuilder().setNxmNxPktMark(value).build(), key);
80             case PACKET_RECEIVED_MATCH:
81                 return new ExtensionAugment<>(NxAugMatchNotifPacketIn.class, new NxAugMatchNotifPacketInBuilder()
82                         .setNxmNxPktMark(value).build(), key);
83             case SWITCH_FLOW_REMOVED_MATCH:
84                 return new ExtensionAugment<>(NxAugMatchNotifSwitchFlowRemoved.class,
85                         new NxAugMatchNotifSwitchFlowRemovedBuilder().setNxmNxPktMark(value).build(), key);
86             case PACKET_IN_MESSAGE_MATCH:
87                 return new ExtensionAugment<>(NxAugMatchPacketInMessage.class,
88                         new NxAugMatchPacketInMessageBuilder().setNxmNxPktMark(value).build(), key);
89             default:
90                 throw new CodecPreconditionException(path);
91         }
92     }
93 }