Rework AugmentRuntimeType and Choice/Case linkage
[mdsal.git] / binding / mdsal-binding-generator / src / main / java / org / opendaylight / mdsal / binding / generator / impl / reactor / LeafListGenerator.java
index 68727b69c32cfc3cdfd74fd8421fa98c07f91b3b..aa6263a192696b724237e364020118151b64c883 100644 (file)
@@ -7,17 +7,19 @@
  */
 package org.opendaylight.mdsal.binding.generator.impl.reactor;
 
+import org.opendaylight.mdsal.binding.generator.impl.rt.DefaultLeafListRuntimeType;
 import org.opendaylight.mdsal.binding.model.api.Type;
 import org.opendaylight.mdsal.binding.model.ri.Types;
+import org.opendaylight.mdsal.binding.runtime.api.LeafListRuntimeType;
 import org.opendaylight.yangtools.yang.common.Ordering;
 import org.opendaylight.yangtools.yang.model.api.stmt.LeafListEffectiveStatement;
-import org.opendaylight.yangtools.yang.model.api.stmt.OrderedByEffectiveStatement;
 
 /**
  * Generator corresponding to a {@code leaf-list} statement.
  */
-final class LeafListGenerator extends AbstractTypeAwareGenerator<LeafListEffectiveStatement, LeafListGenerator> {
-    LeafListGenerator(final LeafListEffectiveStatement statement, final AbstractCompositeGenerator<?> parent) {
+final class LeafListGenerator
+        extends AbstractTypeAwareGenerator<LeafListEffectiveStatement, LeafListRuntimeType, LeafListGenerator> {
+    LeafListGenerator(final LeafListEffectiveStatement statement, final AbstractCompositeGenerator<?, ?> parent) {
         super(statement, parent);
     }
 
@@ -27,15 +29,21 @@ final class LeafListGenerator extends AbstractTypeAwareGenerator<LeafListEffecti
         // List<Object>, we will try to narrow the return type in subclasses.
         final Type type = super.methodReturnType(builderFactory);
         final boolean isObject = Types.objectType().equals(type);
-        final Ordering ordering = statement().findFirstEffectiveSubstatementArgument(OrderedByEffectiveStatement.class)
-            .orElse(Ordering.SYSTEM);
-        switch (ordering) {
-            case SYSTEM:
-                return isObject ? Types.setTypeWildcard() : Types.setTypeFor(type);
-            case USER:
-                return isObject ? Types.listTypeWildcard() : Types.listTypeFor(type);
-            default:
-                throw new IllegalStateException("Unexpected ordering " + ordering);
+
+        if (statement().ordering() == Ordering.SYSTEM) {
+            return isObject ? Types.setTypeWildcard() : Types.setTypeFor(type);
         }
+        return isObject ? Types.listTypeWildcard() : Types.listTypeFor(type);
+    }
+
+    @Override
+    LeafListRuntimeType createExternalRuntimeType(final Type type) {
+        return new DefaultLeafListRuntimeType(type, statement());
+    }
+
+    @Override
+    LeafListRuntimeType createInternalRuntimeType(final ChildLookup lookup, final LeafListEffectiveStatement statement,
+            final Type type) {
+        return new DefaultLeafListRuntimeType(type, statement);
     }
 }