Migrate to simplified addAugmentation() method

[genius.git] / mdsalutil / mdsalutil-api / src / main / java / org / opendaylight / genius / mdsalutil / actions / ActionNdReserved.java

diff --git a/mdsalutil/mdsalutil-api/src/main/java/org/opendaylight/genius/mdsalutil/actions/ActionNdReserved.java b/mdsalutil/mdsalutil-api/src/main/java/org/opendaylight/genius/mdsalutil/actions/ActionNdReserved.java
index 1c8400909406a21329026438c982b3c3ff3c2c2e..26c2b4761ab427b1f7c3c21d3468ea91694b4e63 100644 (file)
--- a/mdsalutil/mdsalutil-api/src/main/java/org/opendaylight/genius/mdsalutil/actions/ActionNdReserved.java
+++ b/mdsalutil/mdsalutil-api/src/main/java/org/opendaylight/genius/mdsalutil/actions/ActionNdReserved.java
@@ -9,24 +9,20 @@ package org.opendaylight.genius.mdsalutil.actions;
 
 import java.util.Collections;
 import org.opendaylight.genius.mdsalutil.ActionInfo;
-import org.opendaylight.openflowplugin.extension.api.ExtensionAugment;
 import org.opendaylight.openflowplugin.extension.api.GroupingLooseResolver;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.action.types.rev131112.action.action.SetFieldCaseBuilder;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.action.types.rev131112.action.action.set.field._case.SetFieldBuilder;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.action.types.rev131112.action.list.Action;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.action.types.rev131112.action.list.ActionBuilder;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.action.types.rev131112.action.list.ActionKey;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.eric.match.rev180730.EricAugMatchNodesNodeTableFlow;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.eric.match.rev180730.EricAugMatchNodesNodeTableFlowBuilder;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.eric.match.rev180730.Icmpv6NdReservedKey;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.eric.match.rev180730.eric.of.icmpv6.nd.reserved.grouping.EricOfIcmpv6NdReservedBuilder;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.general.rev140714.GeneralAugMatchNodesNodeTableFlowWriteActionsSetField;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.general.rev140714.GeneralAugMatchNodesNodeTableFlowWriteActionsSetFieldBuilder;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.general.rev140714.GeneralExtensionListGrouping;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.general.rev140714.general.extension.grouping.Extension;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.general.rev140714.general.extension.grouping.ExtensionBuilder;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.general.rev140714.general.extension.list.grouping.ExtensionListBuilder;
-import org.opendaylight.yangtools.yang.binding.Augmentation;
 
 /**
  * Set source IPv6 action.
@@ -59,33 +55,25 @@ public class ActionNdReserved extends ActionInfo {
                 new GroupingLooseResolver<>(GeneralExtensionListGrouping.class);
         eqGroup.add(GeneralAugMatchNodesNodeTableFlowWriteActionsSetField.class);
 
-        ExtensionAugment<? extends Augmentation<Extension>> extensionMatch
-                =  new ExtensionAugment<>(EricAugMatchNodesNodeTableFlow.class,
-                new EricAugMatchNodesNodeTableFlowBuilder().setEricOfIcmpv6NdReserved(
-                        new EricOfIcmpv6NdReservedBuilder().setIcmpv6NdReserved(ndReserved).build()).build(),
-                Icmpv6NdReservedKey.class);
-
-        ExtensionListBuilder extListBld = null;
-        ExtensionBuilder extBld = new ExtensionBuilder();
-        extBld.addAugmentation(extensionMatch.getAugmentationClass(), extensionMatch.getAugmentationObject());
-
-        extListBld = new ExtensionListBuilder();
-        extListBld.setExtension(extBld.build());
-        extListBld.setExtensionKey(extensionMatch.getKey());
-
-        GeneralAugMatchNodesNodeTableFlowWriteActionsSetField ndReservedSetField =
-                new GeneralAugMatchNodesNodeTableFlowWriteActionsSetFieldBuilder()
-                        .setExtensionList(Collections.singletonList(extListBld.build())).build();
-
         return new ActionBuilder()
-                .setAction(new SetFieldCaseBuilder()
-                        .setSetField(new SetFieldBuilder()
-                                .addAugmentation(GeneralAugMatchNodesNodeTableFlowWriteActionsSetField.class,
-                                        ndReservedSetField)
+            .setAction(new SetFieldCaseBuilder()
+                .setSetField(new SetFieldBuilder()
+                    .addAugmentation(new GeneralAugMatchNodesNodeTableFlowWriteActionsSetFieldBuilder()
+                        .setExtensionList(Collections.singletonList(new ExtensionListBuilder()
+                            .setExtension(new ExtensionBuilder()
+                                .addAugmentation(new EricAugMatchNodesNodeTableFlowBuilder()
+                                    .setEricOfIcmpv6NdReserved(new EricOfIcmpv6NdReservedBuilder()
+                                        .setIcmpv6NdReserved(ndReserved)
+                                        .build())
+                                    .build())
                                 .build())
+                            .setExtensionKey(Icmpv6NdReservedKey.class)
+                            .build()))
                         .build())
-                .withKey(new ActionKey(newActionKey))
-                .build();
+                    .build())
+                .build())
+            .withKey(new ActionKey(newActionKey))
+            .build();
     }
 
     @Override