Reformulate StatementContextFactory.createEffective()

[yangtools.git] / yang / yang-parser-rfc7950 / src / main / java / org / opendaylight / yangtools / yang / parser / rfc7950 / stmt / yang_version / YangVersionStatementSupport.java

diff --git a/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/yang_version/YangVersionStatementSupport.java b/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/yang_version/YangVersionStatementSupport.java
index 483e63951626b474493b3e9c890dc86f9c0646a9..3a938d79345ae20eca45b1d575fb81387a978a26 100644 (file)
--- a/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/yang_version/YangVersionStatementSupport.java
+++ b/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/yang_version/YangVersionStatementSupport.java
@@ -7,25 +7,39 @@
  */
 package org.opendaylight.yangtools.yang.parser.rfc7950.stmt.yang_version;
 
+import com.google.common.collect.ImmutableList;
+import org.eclipse.jdt.annotation.NonNull;
 import org.opendaylight.yangtools.yang.common.YangVersion;
 import org.opendaylight.yangtools.yang.model.api.YangStmtMapping;
+import org.opendaylight.yangtools.yang.model.api.meta.DeclaredStatement;
 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
+import org.opendaylight.yangtools.yang.model.api.stmt.YangVersionEffectiveStatement;
 import org.opendaylight.yangtools.yang.model.api.stmt.YangVersionStatement;
-import org.opendaylight.yangtools.yang.parser.spi.meta.AbstractStatementSupport;
+import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.BaseStatementSupport;
+import org.opendaylight.yangtools.yang.parser.spi.meta.EffectiveStmtCtx.Current;
 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext.Mutable;
 import org.opendaylight.yangtools.yang.parser.spi.meta.SubstatementValidator;
 import org.opendaylight.yangtools.yang.parser.spi.source.SourceException;
 
-public final class YangVersionStatementSupport extends AbstractStatementSupport<YangVersion, YangVersionStatement,
-        EffectiveStatement<YangVersion, YangVersionStatement>> {
+public final class YangVersionStatementSupport
+        extends BaseStatementSupport<YangVersion, YangVersionStatement, YangVersionEffectiveStatement> {
     private static final SubstatementValidator SUBSTATEMENT_VALIDATOR = SubstatementValidator.builder(YangStmtMapping
         .YANG_VERSION)
         .build();
     private static final YangVersionStatementSupport INSTANCE = new YangVersionStatementSupport();
 
+    private static final @NonNull EmptyYangVersionStatement EMPTY_VER1_DECL =
+            new EmptyYangVersionStatement(YangVersion.VERSION_1);
+    private static final @NonNull EmptyYangVersionStatement EMPTY_VER1_1_DECL =
+            new EmptyYangVersionStatement(YangVersion.VERSION_1_1);
+    private static final @NonNull EmptyYangVersionEffectiveStatement EMPTY_VER1_EFF =
+            new EmptyYangVersionEffectiveStatement(EMPTY_VER1_DECL);
+    private static final @NonNull EmptyYangVersionEffectiveStatement EMPTY_VER1_1_EFF =
+            new EmptyYangVersionEffectiveStatement(EMPTY_VER1_1_DECL);
+
     private YangVersionStatementSupport() {
-        super(YangStmtMapping.YANG_VERSION);
+        super(YangStmtMapping.YANG_VERSION, CopyPolicy.CONTEXT_INDEPENDENT);
     }
 
     public static YangVersionStatementSupport getInstance() {
@@ -39,24 +53,49 @@ public final class YangVersionStatementSupport extends AbstractStatementSupport<
     }
 
     @Override
-    public YangVersionStatement createDeclared(final StmtContext<YangVersion, YangVersionStatement, ?> ctx) {
-        return new YangVersionStatementImpl(ctx);
+    public void onPreLinkageDeclared(
+            final Mutable<YangVersion, YangVersionStatement, YangVersionEffectiveStatement> stmt) {
+        stmt.setRootVersion(stmt.getStatementArgument());
     }
 
     @Override
-    public void onPreLinkageDeclared(final Mutable<YangVersion, YangVersionStatement,
-            EffectiveStatement<YangVersion, YangVersionStatement>> stmt) {
-        stmt.setRootVersion(stmt.getStatementArgument());
+    protected SubstatementValidator getSubstatementValidator() {
+        return SUBSTATEMENT_VALIDATOR;
     }
 
     @Override
-    public EffectiveStatement<YangVersion, YangVersionStatement> createEffective(final StmtContext<YangVersion,
-            YangVersionStatement, EffectiveStatement<YangVersion, YangVersionStatement>> ctx) {
-        return new YangVersionEffectiveStatementImpl(ctx);
+    protected YangVersionStatement createDeclared(final StmtContext<YangVersion, YangVersionStatement, ?> ctx,
+            final ImmutableList<? extends DeclaredStatement<?>> substatements) {
+        return new RegularYangVersionStatement(ctx.coerceStatementArgument(), substatements);
     }
 
     @Override
-    protected SubstatementValidator getSubstatementValidator() {
-        return SUBSTATEMENT_VALIDATOR;
+    protected YangVersionStatement createEmptyDeclared(final StmtContext<YangVersion, YangVersionStatement, ?> ctx) {
+        final YangVersion argument = ctx.coerceStatementArgument();
+        switch (argument) {
+            case VERSION_1:
+                return EMPTY_VER1_DECL;
+            case VERSION_1_1:
+                return EMPTY_VER1_1_DECL;
+            default:
+                throw new IllegalStateException("Unhandled version " + argument);
+        }
+    }
+
+    @Override
+    protected YangVersionEffectiveStatement createEffective(final Current<YangVersion, YangVersionStatement> stmt,
+            final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
+        return substatements.isEmpty() ? createEmptyEffective(stmt.declared())
+            : new RegularYangVersionEffectiveStatement(stmt.declared(), substatements);
+    }
+
+    private static @NonNull YangVersionEffectiveStatement createEmptyEffective(final YangVersionStatement declared) {
+        if (EMPTY_VER1_DECL.equals(declared)) {
+            return EMPTY_VER1_EFF;
+        } else if (EMPTY_VER1_1_DECL.equals(declared)) {
+            return EMPTY_VER1_1_EFF;
+        } else {
+            return new EmptyYangVersionEffectiveStatement(declared);
+        }
     }
-}
\ No newline at end of file
+}