yang/yang-parser-reactor/src/main/java/org/opendaylight/yangtools/yang/parser/stmt/reactor/StatementContextWriter.java

   1 /*
   2  * Copyright (c) 2015 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.yangtools.yang.parser.stmt.reactor;
   9
  10 import static com.google.common.base.Preconditions.checkArgument;
  11 import static com.google.common.base.Preconditions.checkState;
  12 import static com.google.common.base.Verify.verifyNotNull;
  13 import static java.util.Objects.requireNonNull;
  14
  15 import java.util.Optional;
  16 import org.opendaylight.yangtools.yang.common.QName;
  17 import org.opendaylight.yangtools.yang.model.api.meta.StatementSource;
  18 import org.opendaylight.yangtools.yang.parser.spi.meta.ModelProcessingPhase;
  19 import org.opendaylight.yangtools.yang.parser.spi.source.StatementSourceReference;
  20 import org.opendaylight.yangtools.yang.parser.spi.source.StatementWriter;
  21
  22 final class StatementContextWriter implements StatementWriter {
  23     private final ModelProcessingPhase phase;
  24     private final SourceSpecificContext ctx;
  25
  26     private StatementContextBase<?, ?, ?> current;
  27
  28     StatementContextWriter(final SourceSpecificContext ctx, final ModelProcessingPhase phase) {
  29         this.ctx = requireNonNull(ctx);
  30         this.phase = requireNonNull(phase);
  31     }
  32
  33     @Override
  34     public Optional<? extends ResumedStatement> resumeStatement(final int childId) {
  35         final Optional<StatementContextBase<?, ?, ?>> existing = ctx.lookupDeclaredChild(current, childId);
  36         existing.ifPresent(this::resumeStatement);
  37         return existing;
  38     }
  39
  40     private void resumeStatement(final StatementContextBase<?, ?, ?> child) {
  41         if (child.isFullyDefined()) {
  42             child.walkChildren(phase);
  43             child.endDeclared(phase);
  44         } else {
  45             current = child;
  46         }
  47     }
  48
  49     @Override
  50     public void storeStatement(final int expectedChildren, final boolean fullyDefined) {
  51         checkState(current != null);
  52         checkArgument(expectedChildren >= 0);
  53         current.resizeSubstatements(expectedChildren);
  54
  55         if (fullyDefined) {
  56             current.setFullyDefined();
  57         }
  58     }
  59
  60     @Override
  61     public void startStatement(final int childId, final QName name, final String argument,
  62             final StatementSourceReference ref) {
  63         final Optional<StatementContextBase<?, ?, ?>> existing = ctx.lookupDeclaredChild(current, childId);
  64         current = existing.isPresent() ? existing.get()
  65                 :  verifyNotNull(ctx.createDeclaredChild(current, childId, name, argument, ref));
  66     }
  67
  68     @Override
  69     public void endStatement(final StatementSourceReference ref) {
  70         checkState(current != null);
  71         current.endDeclared(phase);
  72         exitStatement();
  73     }
  74
  75     @Override
  76     public ModelProcessingPhase getPhase() {
  77         return phase;
  78     }
  79
  80     private void exitStatement() {
  81         StatementContextBase<?, ?, ?> parentContext = current.getParentContext();
  82         while (parentContext != null && StatementSource.CONTEXT == parentContext.getStatementSource()) {
  83             parentContext.endDeclared(phase);
  84             parentContext = parentContext.getParentContext();
  85         }
  86         current = parentContext;
  87     }
  88 }