Rework AugmentRuntimeType and Choice/Case linkage

[mdsal.git] / binding / mdsal-binding-generator / src / main / java / org / opendaylight / mdsal / binding / generator / impl / reactor / ActionGenerator.java

diff --git a/binding/mdsal-binding-generator/src/main/java/org/opendaylight/mdsal/binding/generator/impl/reactor/ActionGenerator.java b/binding/mdsal-binding-generator/src/main/java/org/opendaylight/mdsal/binding/generator/impl/reactor/ActionGenerator.java
index 8b0c9b9b22c7ddd97084fee0cf5119e1eea0ddda..19cca901a5e8cce304992671d1f0d0fab1203e15 100644 (file)
--- a/binding/mdsal-binding-generator/src/main/java/org/opendaylight/mdsal/binding/generator/impl/reactor/ActionGenerator.java
+++ b/binding/mdsal-binding-generator/src/main/java/org/opendaylight/mdsal/binding/generator/impl/reactor/ActionGenerator.java
@@ -7,6 +7,8 @@
  */
 package org.opendaylight.mdsal.binding.generator.impl.reactor;
 
+import static com.google.common.base.Verify.verify;
+
 import java.util.List;
 import org.eclipse.jdt.annotation.NonNull;
 import org.opendaylight.mdsal.binding.generator.impl.rt.DefaultActionRuntimeType;
@@ -60,12 +62,6 @@ final class ActionGenerator extends CompositeSchemaTreeGenerator<ActionEffective
         return builder.build();
     }
 
-    @Override
-    ActionRuntimeType createRuntimeType(final GeneratedType type, final ActionEffectiveStatement statement,
-            final List<RuntimeType> children, final List<AugmentRuntimeType> augments) {
-        return new DefaultActionRuntimeType(type, statement, children, augments);
-    }
-
     private @NonNull Type implementedType(final TypeBuilderFactory builderFactory) {
         final GeneratedType input = getChild(this, InputEffectiveStatement.class).getOriginal()
             .getGeneratedType(builderFactory);
@@ -88,4 +84,17 @@ final class ActionGenerator extends CompositeSchemaTreeGenerator<ActionEffective
     void addAsGetterMethod(final GeneratedTypeBuilderBase<?> builder, final TypeBuilderFactory builderFactory) {
         // actions are a separate concept
     }
+
+    @Override
+    CompositeRuntimeTypeBuilder<ActionEffectiveStatement, ActionRuntimeType> createBuilder(
+            final ActionEffectiveStatement statement) {
+        return new CompositeRuntimeTypeBuilder<>(statement) {
+            @Override
+            ActionRuntimeType build(final GeneratedType generatedType, final ActionEffectiveStatement statement,
+                    final List<RuntimeType> childTypes, final List<AugmentRuntimeType> augmentTypes) {
+                verify(augmentTypes.isEmpty(), "Unexpected augments %s", augmentTypes);
+                return new DefaultActionRuntimeType(generatedType, statement, childTypes);
+            }
+        };
+    }
 }