Migrate to simplified addAugmentation() method

[genius.git] / mdsalutil / mdsalutil-api / src / main / java / org / opendaylight / genius / mdsalutil / ericmatches / EricMatchInfoHelper.java

diff --git a/mdsalutil/mdsalutil-api/src/main/java/org/opendaylight/genius/mdsalutil/ericmatches/EricMatchInfoHelper.java b/mdsalutil/mdsalutil-api/src/main/java/org/opendaylight/genius/mdsalutil/ericmatches/EricMatchInfoHelper.java
index 66bba53e74f7665e4a03f4d7c4f071fadb728e40..5d5a32196e91ede11a05e3701ff7fc219c9caf09 100644 (file)
--- a/mdsalutil/mdsalutil-api/src/main/java/org/opendaylight/genius/mdsalutil/ericmatches/EricMatchInfoHelper.java
+++ b/mdsalutil/mdsalutil-api/src/main/java/org/opendaylight/genius/mdsalutil/ericmatches/EricMatchInfoHelper.java
@@ -57,9 +57,7 @@ public abstract class EricMatchInfoHelper<T extends DataObject, B extends Builde
             EricAugMatchNodesNodeTableFlow ericAugMatch = ericAugMatchBuilder.build();
             GeneralAugMatchNodesNodeTableFlow existingAugmentations = matchBuilder
                     .augmentation(GeneralAugMatchNodesNodeTableFlow.class);
-            GeneralAugMatchNodesNodeTableFlow genAugMatch = generalAugMatchBuilder(existingAugmentations,
-                    ericAugMatch, keyClass);
-            matchBuilder.addAugmentation(GeneralAugMatchNodesNodeTableFlow.class, genAugMatch);
+            matchBuilder.addAugmentation(generalAugMatchBuilder(existingAugmentations, ericAugMatch, keyClass));
         }
     }
 
@@ -74,9 +72,7 @@ public abstract class EricMatchInfoHelper<T extends DataObject, B extends Builde
             extensions = new HashMap<>();
         }
         ExtensionList extensionList = new ExtensionListBuilder().setExtensionKey(extentionKey)
-                .setExtension(
-                        new ExtensionBuilder().addAugmentation(EricAugMatchNodesNodeTableFlow.class, ericAugMatch)
-                                .build())
+                .setExtension(new ExtensionBuilder().addAugmentation(ericAugMatch).build())
                 .build();
         extensions.put(extensionList.key(),extensionList);
         return new GeneralAugMatchNodesNodeTableFlowBuilder().setExtensionList(extensions).build();