Migrate to simplified addAugmentation() method

[genius.git] / mdsalutil / mdsalutil-api / src / main / java / org / opendaylight / genius / mdsalutil / nxmatches / NxMatchInfoHelper.java

diff --git a/mdsalutil/mdsalutil-api/src/main/java/org/opendaylight/genius/mdsalutil/nxmatches/NxMatchInfoHelper.java b/mdsalutil/mdsalutil-api/src/main/java/org/opendaylight/genius/mdsalutil/nxmatches/NxMatchInfoHelper.java
index 09b87ee54d86d03c535c4a16f18971acddcee40a..cd809ce895f0b5a37b9e9fe3253d0c500c57bcb6 100644 (file)
--- a/mdsalutil/mdsalutil-api/src/main/java/org/opendaylight/genius/mdsalutil/nxmatches/NxMatchInfoHelper.java
+++ b/mdsalutil/mdsalutil-api/src/main/java/org/opendaylight/genius/mdsalutil/nxmatches/NxMatchInfoHelper.java
@@ -59,9 +59,7 @@ public abstract class NxMatchInfoHelper<T extends DataObject, B extends Builder<
             NxAugMatchNodesNodeTableFlow nxAugMatch = nxAugMatchBuilder.build();
             GeneralAugMatchNodesNodeTableFlow existingAugmentations = matchBuilder
                     .augmentation(GeneralAugMatchNodesNodeTableFlow.class);
-            GeneralAugMatchNodesNodeTableFlow genAugMatch = generalAugMatchBuilder(existingAugmentations,
-                    nxAugMatch, keyClass);
-            matchBuilder.addAugmentation(GeneralAugMatchNodesNodeTableFlow.class, genAugMatch);
+            matchBuilder.addAugmentation(generalAugMatchBuilder(existingAugmentations, nxAugMatch, keyClass));
         }
     }
 
@@ -79,8 +77,7 @@ public abstract class NxMatchInfoHelper<T extends DataObject, B extends Builder<
         }
 
         ExtensionList extensionList = new ExtensionListBuilder().setExtensionKey(extentionKey)
-                .setExtension(new ExtensionBuilder().addAugmentation(NxAugMatchNodesNodeTableFlow.class,
-                        nxAugMatch).build()).build();
+                .setExtension(new ExtensionBuilder().addAugmentation(nxAugMatch).build()).build();
         extensions.put(extensionList.key(),extensionList);
         return new GeneralAugMatchNodesNodeTableFlowBuilder().setExtensionList(extensions).build();
     }