Rework AugmentRuntimeType and Choice/Case linkage

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

diff --git a/binding/mdsal-binding-generator/src/main/java/org/opendaylight/mdsal/binding/generator/impl/reactor/OutputGenerator.java b/binding/mdsal-binding-generator/src/main/java/org/opendaylight/mdsal/binding/generator/impl/reactor/OutputGenerator.java
index 320593d7e6d61e8ce29431969c4d38107c0ac25e..83c48d26348fe1c9d184b2f8cf4fb6609a1eddc6 100644 (file)
--- a/binding/mdsal-binding-generator/src/main/java/org/opendaylight/mdsal/binding/generator/impl/reactor/OutputGenerator.java
+++ b/binding/mdsal-binding-generator/src/main/java/org/opendaylight/mdsal/binding/generator/impl/reactor/OutputGenerator.java
@@ -25,8 +25,14 @@ class OutputGenerator extends OperationContainerGenerator<OutputEffectiveStateme
     }
 
     @Override
-    final OutputRuntimeType createRuntimeType(final GeneratedType type, final OutputEffectiveStatement statement,
-            final List<RuntimeType> children, final List<AugmentRuntimeType> augments) {
-        return new DefaultOutputRuntimeType(type, statement, children, augments);
+    final CompositeRuntimeTypeBuilder<OutputEffectiveStatement, OutputRuntimeType> createBuilder(
+            final OutputEffectiveStatement statement) {
+        return new CompositeRuntimeTypeBuilder<>(statement) {
+            @Override
+            OutputRuntimeType build(final GeneratedType type, final OutputEffectiveStatement statement,
+                    final List<RuntimeType> children, final List<AugmentRuntimeType> augments) {
+                return new DefaultOutputRuntimeType(type, statement, children, augments);
+            }
+        };
     }
 }