Define ExecutionOrder
[yangtools.git] / yang / yang-parser-reactor / src / main / java / org / opendaylight / yangtools / yang / parser / stmt / reactor / StatementContextBase.java
index a331d6b6d91619da119fcc3323aa94dee5a478a5..71fdd107bf13c474fb8c3a983c7d0ed01d64415f 100644 (file)
@@ -10,6 +10,7 @@ package org.opendaylight.yangtools.yang.parser.stmt.reactor;
 import static com.google.common.base.Preconditions.checkArgument;
 import static com.google.common.base.Preconditions.checkState;
 import static com.google.common.base.Verify.verify;
+import static com.google.common.base.Verify.verifyNotNull;
 import static java.util.Objects.requireNonNull;
 
 import com.google.common.annotations.Beta;
@@ -29,7 +30,6 @@ import java.util.Map.Entry;
 import java.util.Optional;
 import java.util.stream.Stream;
 import org.eclipse.jdt.annotation.NonNull;
-import org.eclipse.jdt.annotation.Nullable;
 import org.opendaylight.yangtools.yang.common.QNameModule;
 import org.opendaylight.yangtools.yang.model.api.meta.DeclaredStatement;
 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
@@ -39,6 +39,7 @@ import org.opendaylight.yangtools.yang.parser.spi.meta.CopyType;
 import org.opendaylight.yangtools.yang.parser.spi.meta.ImplicitParentAwareStatementSupport;
 import org.opendaylight.yangtools.yang.parser.spi.meta.ModelActionBuilder;
 import org.opendaylight.yangtools.yang.parser.spi.meta.ModelProcessingPhase;
+import org.opendaylight.yangtools.yang.parser.spi.meta.ModelProcessingPhase.ExecutionOrder;
 import org.opendaylight.yangtools.yang.parser.spi.meta.MutableStatement;
 import org.opendaylight.yangtools.yang.parser.spi.meta.NamespaceBehaviour;
 import org.opendaylight.yangtools.yang.parser.spi.meta.NamespaceKeyCriterion;
@@ -99,19 +100,23 @@ public abstract class StatementContextBase<A, D extends DeclaredStatement<A>, E
     //       operations, hence we want to keep it close by.
     private final @NonNull StatementDefinitionContext<A, D, E> definition;
 
+    // TODO: consider keying by Byte equivalent of ExecutionOrder
     private Multimap<ModelProcessingPhase, OnPhaseFinished> phaseListeners = ImmutableMultimap.of();
     private Multimap<ModelProcessingPhase, ContextMutation> phaseMutation = ImmutableMultimap.of();
 
     private List<StmtContext<?, ?, ?>> effectOfStatement = ImmutableList.of();
 
-    private @Nullable ModelProcessingPhase completedPhase;
+    /**
+     * {@link ModelProcessingPhase.ExecutionOrder} value of current {@link ModelProcessingPhase} of this statement.
+     */
+    private byte executionOrder;
 
     // Copy constructor used by subclasses to implement reparent()
     StatementContextBase(final StatementContextBase<A, D, E> original) {
         super(original);
         this.copyHistory = original.copyHistory;
         this.definition = original.definition;
-        this.completedPhase = original.completedPhase;
+        this.executionOrder = original.executionOrder;
     }
 
     StatementContextBase(final StatementDefinitionContext<A, D, E> def) {
@@ -147,14 +152,14 @@ public abstract class StatementContextBase<A, D extends DeclaredStatement<A>, E
     }
 
     @Override
-    public final ModelProcessingPhase getCompletedPhase() {
-        return completedPhase;
+    final byte executionOrder() {
+        return executionOrder;
     }
 
     // FIXME: this should be propagated through a correct constructor
     @Deprecated
     final void setCompletedPhase(final ModelProcessingPhase completedPhase) {
-        this.completedPhase = completedPhase;
+        this.executionOrder = completedPhase.executionOrder();
     }
 
     @Override
@@ -256,10 +261,7 @@ public abstract class StatementContextBase<A, D extends DeclaredStatement<A>, E
 
         final List<ReactorStmtCtx<?, ?, ?>> resized = beforeAddEffectiveStatement(effective, 1);
         final ReactorStmtCtx<?, ?, ?> stmt = (ReactorStmtCtx<?, ?, ?>) substatement;
-        final ModelProcessingPhase phase = completedPhase;
-        if (phase != null) {
-            ensureCompletedPhase(stmt, phase);
-        }
+        ensureCompletedExecution(stmt);
         resized.add(stmt);
         return resized;
     }
@@ -287,10 +289,9 @@ public abstract class StatementContextBase<A, D extends DeclaredStatement<A>, E
         final List<ReactorStmtCtx<?, ?, ?>> resized = beforeAddEffectiveStatement(effective, statements.size());
         final Collection<? extends ReactorStmtCtx<?, ?, ?>> casted =
             (Collection<? extends ReactorStmtCtx<?, ?, ?>>) statements;
-        final ModelProcessingPhase phase = completedPhase;
-        if (phase != null) {
+        if (executionOrder != ExecutionOrder.NULL) {
             for (ReactorStmtCtx<?, ?, ?> stmt : casted) {
-                ensureCompletedPhase(stmt, phase);
+                ensureCompletedExecution(stmt, executionOrder);
             }
         }
 
@@ -303,17 +304,20 @@ public abstract class StatementContextBase<A, D extends DeclaredStatement<A>, E
     // exposed for InferredStatementContext only
     final void ensureCompletedPhase(final Mutable<?, ?, ?> stmt) {
         verifyStatement(stmt);
-        final ModelProcessingPhase phase = completedPhase;
-        if (phase != null) {
-            ensureCompletedPhase((ReactorStmtCtx<?, ?, ?>) stmt, phase);
+        ensureCompletedExecution((ReactorStmtCtx<?, ?, ?>) stmt);
+    }
+
+    // Make sure target statement has transitioned at least to our phase (if we have one). This method is just before we
+    // take allow a statement to become our substatement. This is needed to ensure that every statement tree does not
+    // contain any statements which did not complete the same phase as the root statement.
+    private void ensureCompletedExecution(final ReactorStmtCtx<?, ?, ?> stmt) {
+        if (executionOrder != ExecutionOrder.NULL) {
+            ensureCompletedExecution(stmt, executionOrder);
         }
     }
 
-    // Make sure target statement has transitioned at least to specified phase. This method is just before we take
-    // allow a statement to become our substatement. This is needed to ensure that every statement tree does not contain
-    // any statements which did not complete the same phase as the root statement.
-    private static void ensureCompletedPhase(final ReactorStmtCtx<?, ?, ?> stmt, final ModelProcessingPhase phase) {
-        verify(stmt.tryToCompletePhase(phase), "Statement %s cannot complete phase %s", stmt, phase);
+    private static void ensureCompletedExecution(final ReactorStmtCtx<?, ?, ?> stmt, final byte executionOrder) {
+        verify(stmt.tryToCompletePhase(executionOrder), "Statement %s cannot complete phase %s", stmt, executionOrder);
     }
 
     private static void verifyStatement(final Mutable<?, ?, ?> stmt) {
@@ -322,8 +326,10 @@ public abstract class StatementContextBase<A, D extends DeclaredStatement<A>, E
 
     private List<ReactorStmtCtx<?, ?, ?>> beforeAddEffectiveStatement(final List<ReactorStmtCtx<?, ?, ?>> effective,
             final int toAdd) {
-        // We cannot allow statement to be further mutated
-        verify(completedPhase != ModelProcessingPhase.EFFECTIVE_MODEL, "Cannot modify finished statement at %s",
+        // We cannot allow statement to be further mutated.
+        // TODO: we really want to say 'not NULL and not at or after EFFECTIVE_MODEL here. This will matter if we have
+        //       a phase after EFFECTIVE_MODEL
+        verify(executionOrder != ExecutionOrder.EFFECTIVE_MODEL, "Cannot modify finished statement at %s",
             sourceReference());
         return beforeAddEffectiveStatementUnsafe(effective, toAdd);
     }
@@ -376,27 +382,28 @@ public abstract class StatementContextBase<A, D extends DeclaredStatement<A>, E
     abstract Stream<? extends @NonNull StmtContext<?, ?, ?>> streamEffective();
 
     @Override
-    final boolean doTryToCompletePhase(final ModelProcessingPhase phase) {
-        final boolean finished = phaseMutation.isEmpty() ? true : runMutations(phase);
-        if (completeChildren(phase) && finished) {
-            onPhaseCompleted(phase);
+    final boolean doTryToCompletePhase(final byte targetOrder) {
+        final boolean finished = phaseMutation.isEmpty() ? true : runMutations(targetOrder);
+        if (completeChildren(targetOrder) && finished) {
+            onPhaseCompleted(targetOrder);
             return true;
         }
         return false;
     }
 
-    private boolean completeChildren(final ModelProcessingPhase phase) {
+    private boolean completeChildren(final byte targetOrder) {
         boolean finished = true;
         for (final StatementContextBase<?, ?, ?> child : mutableDeclaredSubstatements()) {
-            finished &= child.tryToCompletePhase(phase);
+            finished &= child.tryToCompletePhase(targetOrder);
         }
         for (final ReactorStmtCtx<?, ?, ?> child : effectiveChildrenToComplete()) {
-            finished &= child.tryToCompletePhase(phase);
+            finished &= child.tryToCompletePhase(targetOrder);
         }
         return finished;
     }
 
-    private boolean runMutations(final ModelProcessingPhase phase) {
+    private boolean runMutations(final byte targetOrder) {
+        final ModelProcessingPhase phase = verifyNotNull(ModelProcessingPhase.ofExecutionOrder(targetOrder));
         final Collection<ContextMutation> openMutations = phaseMutation.get(phase);
         return openMutations.isEmpty() ? true : runMutations(phase, openMutations);
     }
@@ -427,19 +434,20 @@ public abstract class StatementContextBase<A, D extends DeclaredStatement<A>, E
     }
 
     /**
-     * Occurs on end of {@link ModelProcessingPhase} of source parsing.
+     * Occurs on end of {@link ModelProcessingPhase} of source parsing. This method must not be called with
+     * {@code executionOrder} equal to {@link ExecutionOrder#NULL}.
      *
-     * @param phase
-     *            that was to be completed (finished)
-     * @throws SourceException
-     *             when an error occurred in source parsing
+     * @param phase that was to be completed (finished)
+     * @throws SourceException when an error occurred in source parsing
      */
-    private void onPhaseCompleted(final ModelProcessingPhase phase) {
-        completedPhase = phase;
-        if (phase == ModelProcessingPhase.EFFECTIVE_MODEL) {
+    private void onPhaseCompleted(final byte completedOrder) {
+        executionOrder = completedOrder;
+        if (completedOrder == ExecutionOrder.EFFECTIVE_MODEL) {
+            // We have completed effective model, substatements are guaranteed not to change
             summarizeSubstatementPolicy();
         }
 
+        final ModelProcessingPhase phase = verifyNotNull(ModelProcessingPhase.ofExecutionOrder(completedOrder));
         final Collection<OnPhaseFinished> listeners = phaseListeners.get(phase);
         if (!listeners.isEmpty()) {
             runPhaseListeners(phase, listeners);
@@ -609,7 +617,7 @@ public abstract class StatementContextBase<A, D extends DeclaredStatement<A>, E
         requireNonNull(phase, "Statement context processing phase cannot be null");
         requireNonNull(listener, "Statement context phase listener cannot be null");
 
-        ModelProcessingPhase finishedPhase = completedPhase;
+        ModelProcessingPhase finishedPhase = ModelProcessingPhase.ofExecutionOrder(executionOrder);
         while (finishedPhase != null) {
             if (phase.equals(finishedPhase)) {
                 listener.phaseFinished(this, finishedPhase);
@@ -630,12 +638,8 @@ public abstract class StatementContextBase<A, D extends DeclaredStatement<A>, E
      * @throws IllegalStateException when the mutation was registered after phase was completed
      */
     final void addMutation(final ModelProcessingPhase phase, final ContextMutation mutation) {
-        ModelProcessingPhase finishedPhase = completedPhase;
-        while (finishedPhase != null) {
-            checkState(!phase.equals(finishedPhase), "Mutation registered after phase was completed at: %s",
-                sourceReference());
-            finishedPhase = finishedPhase.getPreviousPhase();
-        }
+        checkState(executionOrder < phase.executionOrder(), "Mutation registered after phase was completed at: %s",
+            sourceReference());
 
         if (phaseMutation.isEmpty()) {
             phaseMutation = newMultimap();