Rework AugmentRuntimeType and Choice/Case linkage

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

diff --git a/binding/mdsal-binding-generator/src/main/java/org/opendaylight/mdsal/binding/generator/impl/reactor/InputGenerator.java b/binding/mdsal-binding-generator/src/main/java/org/opendaylight/mdsal/binding/generator/impl/reactor/InputGenerator.java
index 2400b4a9ee51eeaccae6e70c118c2bd4e42bd6ce..08e4da9b3ad8b6cbeeeb63c6cbe7f90ecafa2beb 100644 (file)
--- a/binding/mdsal-binding-generator/src/main/java/org/opendaylight/mdsal/binding/generator/impl/reactor/InputGenerator.java
+++ b/binding/mdsal-binding-generator/src/main/java/org/opendaylight/mdsal/binding/generator/impl/reactor/InputGenerator.java
@@ -25,8 +25,14 @@ class InputGenerator extends OperationContainerGenerator<InputEffectiveStatement
     }
 
     @Override
-    final InputRuntimeType createRuntimeType(final GeneratedType type, final InputEffectiveStatement statement,
-            final List<RuntimeType> children, final List<AugmentRuntimeType> augments) {
-        return new DefaultInputRuntimeType(type, statement, children, augments);
+    final CompositeRuntimeTypeBuilder<InputEffectiveStatement, InputRuntimeType> createBuilder(
+            final InputEffectiveStatement statement) {
+        return new CompositeRuntimeTypeBuilder<>(statement) {
+            @Override
+            InputRuntimeType build(final GeneratedType type, final InputEffectiveStatement statement,
+                    final List<RuntimeType> children, final List<AugmentRuntimeType> augments) {
+                return new DefaultInputRuntimeType(type, statement, children, augments);
+            }
+        };
     }
 }