Retain grouping/uses instantiation vectors

[mdsal.git] / binding / mdsal-binding-generator / src / main / java / org / opendaylight / mdsal / binding / generator / impl / rt / DefaultGroupingRuntimeType.java

diff --git a/binding/mdsal-binding-generator/src/main/java/org/opendaylight/mdsal/binding/generator/impl/rt/DefaultGroupingRuntimeType.java b/binding/mdsal-binding-generator/src/main/java/org/opendaylight/mdsal/binding/generator/impl/rt/DefaultGroupingRuntimeType.java
index 5b81473851784efb1d746533f8f17acf81f9b089..a0abdcefbb02c7de44aebd796a250a253f4364e6 100644 (file)
--- a/binding/mdsal-binding-generator/src/main/java/org/opendaylight/mdsal/binding/generator/impl/rt/DefaultGroupingRuntimeType.java
+++ b/binding/mdsal-binding-generator/src/main/java/org/opendaylight/mdsal/binding/generator/impl/rt/DefaultGroupingRuntimeType.java
@@ -7,16 +7,49 @@
  */
 package org.opendaylight.mdsal.binding.generator.impl.rt;
 
+import java.util.Arrays;
+import java.util.Collections;
 import java.util.List;
+import java.util.Objects;
+import org.eclipse.jdt.annotation.Nullable;
 import org.opendaylight.mdsal.binding.model.api.GeneratedType;
+import org.opendaylight.mdsal.binding.runtime.api.CompositeRuntimeType;
 import org.opendaylight.mdsal.binding.runtime.api.GroupingRuntimeType;
 import org.opendaylight.mdsal.binding.runtime.api.RuntimeType;
 import org.opendaylight.yangtools.yang.model.api.stmt.GroupingEffectiveStatement;
 
 public final class DefaultGroupingRuntimeType extends AbstractCompositeRuntimeType<GroupingEffectiveStatement>
         implements GroupingRuntimeType {
+    /**
+     * These are vectors towards concrete instantiations of this type -- i.e. the manifestation in the effective data
+     * tree. Each item in this list represents either:
+     * <ul>
+     *   <li>a concrete instantiation, or<li>
+     *   <li>another {@link GroupingRuntimeType}</li>
+     * </ul>
+     * We use these vectors to create {@link #instantiations()}.
+     */
+    private final @Nullable Object instantiationVectors;
+
     public DefaultGroupingRuntimeType(final GeneratedType bindingType, final GroupingEffectiveStatement statement,
-            final List<RuntimeType> children) {
+            final List<RuntimeType> children, final List<? extends CompositeRuntimeType> instantiationVectors) {
         super(bindingType, statement, children);
+        this.instantiationVectors = switch (instantiationVectors.size()) {
+            case 0 -> null;
+            case 1 -> Objects.requireNonNull(instantiationVectors.get(0));
+            default -> instantiationVectors.stream().map(Objects::requireNonNull).toArray(CompositeRuntimeType[]::new);
+        };
+    }
+
+    @Override
+    public List<CompositeRuntimeType> directUsers() {
+        final var local = instantiationVectors;
+        if (local == null) {
+            return List.of();
+        } else if (local instanceof CompositeRuntimeType[] array) {
+            return Collections.unmodifiableList(Arrays.asList(array));
+        } else {
+            return List.of((CompositeRuntimeType) local);
+        }
     }
 }