Convert trivial CopyPolicy users to StatementPolicy

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

diff --git a/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/base/BaseStatementSupport.java b/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/base/BaseStatementSupport.java
index ba212f5d8723e72446351bb161d123d5b11aec38..1f03b12a118ee6ccd33df04d73577477ed8e3ace 100644 (file)
--- a/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/base/BaseStatementSupport.java
+++ b/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/base/BaseStatementSupport.java
@@ -18,6 +18,7 @@ import org.opendaylight.yangtools.yang.model.api.stmt.BaseStatement;
 import org.opendaylight.yangtools.yang.model.api.stmt.IdentityStatement;
 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.BaseQNameStatementSupport;
 import org.opendaylight.yangtools.yang.parser.spi.IdentityNamespace;
+import org.opendaylight.yangtools.yang.parser.spi.meta.EffectiveStmtCtx.Current;
 import org.opendaylight.yangtools.yang.parser.spi.meta.InferenceException;
 import org.opendaylight.yangtools.yang.parser.spi.meta.ModelActionBuilder;
 import org.opendaylight.yangtools.yang.parser.spi.meta.ModelActionBuilder.InferenceAction;
@@ -35,7 +36,7 @@ public final class BaseStatementSupport extends BaseQNameStatementSupport<BaseSt
     private static final BaseStatementSupport INSTANCE = new BaseStatementSupport();
 
     private BaseStatementSupport() {
-        super(YangStmtMapping.BASE);
+        super(YangStmtMapping.BASE, StatementPolicy.reject());
     }
 
     public static BaseStatementSupport getInstance() {
@@ -51,7 +52,7 @@ public final class BaseStatementSupport extends BaseQNameStatementSupport<BaseSt
     public void onStatementDefinitionDeclared(final Mutable<QName, BaseStatement, BaseEffectiveStatement> baseStmtCtx) {
         final Mutable<?, ?, ?> baseParentCtx = baseStmtCtx.coerceParentContext();
         if (baseParentCtx.producesDeclared(IdentityStatement.class)) {
-            final QName baseIdentityQName = baseStmtCtx.coerceStatementArgument();
+            final QName baseIdentityQName = baseStmtCtx.getArgument();
             final ModelActionBuilder baseIdentityAction = baseStmtCtx.newInferenceAction(
                 ModelProcessingPhase.STATEMENT_DEFINITION);
             baseIdentityAction.requiresCtx(baseStmtCtx, IdentityNamespace.class, baseIdentityQName,
@@ -66,9 +67,8 @@ public final class BaseStatementSupport extends BaseQNameStatementSupport<BaseSt
 
                 @Override
                 public void prerequisiteFailed(final Collection<? extends Prerequisite<?>> failed) {
-                    throw new InferenceException(baseStmtCtx.getStatementSourceReference(),
-                        "Unable to resolve identity %s and base identity %s",
-                        baseParentCtx.getStatementArgument(), baseStmtCtx.getStatementArgument());
+                    throw new InferenceException(baseStmtCtx, "Unable to resolve identity %s and base identity %s",
+                        baseParentCtx.argument(), baseStmtCtx.argument());
                 }
             });
         }
@@ -82,24 +82,18 @@ public final class BaseStatementSupport extends BaseQNameStatementSupport<BaseSt
     @Override
     protected BaseStatement createDeclared(final StmtContext<QName, BaseStatement, ?> ctx,
             final ImmutableList<? extends DeclaredStatement<?>> substatements) {
-        return new RegularBaseStatement(ctx.coerceStatementArgument(), substatements);
+        return new RegularBaseStatement(ctx.getArgument(), substatements);
     }
 
     @Override
     protected BaseStatement createEmptyDeclared(final StmtContext<QName, BaseStatement, ?> ctx) {
-        return new EmptyBaseStatement(ctx.coerceStatementArgument());
+        return new EmptyBaseStatement(ctx.getArgument());
     }
 
     @Override
-    protected BaseEffectiveStatement createEffective(
-            final StmtContext<QName, BaseStatement, BaseEffectiveStatement> ctx, final BaseStatement declared,
+    protected BaseEffectiveStatement createEffective(final Current<QName, BaseStatement> stmt,
             final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
-        return new RegularBaseEffectiveStatement(declared, substatements);
-    }
-
-    @Override
-    protected BaseEffectiveStatement createEmptyEffective(
-            final StmtContext<QName, BaseStatement, BaseEffectiveStatement> ctx, final BaseStatement declared) {
-        return new EmptyBaseEffectiveStatement(declared);
+        return substatements.isEmpty() ? new EmptyBaseEffectiveStatement(stmt.declared())
+            : new RegularBaseEffectiveStatement(stmt.declared(), substatements);
     }
 }