Bug 7051 - moving of SubstatementValidator into spi.meta package

[yangtools.git] / yang / yang-parser-impl / src / main / java / org / opendaylight / yangtools / yang / parser / stmt / rfc6020 / IdentityRefSpecificationImpl.java

diff --git a/yang/yang-parser-impl/src/main/java/org/opendaylight/yangtools/yang/parser/stmt/rfc6020/IdentityRefSpecificationImpl.java b/yang/yang-parser-impl/src/main/java/org/opendaylight/yangtools/yang/parser/stmt/rfc6020/IdentityRefSpecificationImpl.java
index 2deb59dc921133516e27ae36c1d72de7e9855275..ea3f5425106c12d95a1e96006adccc3126f758d2 100644 (file)
--- a/yang/yang-parser-impl/src/main/java/org/opendaylight/yangtools/yang/parser/stmt/rfc6020/IdentityRefSpecificationImpl.java
+++ b/yang/yang-parser-impl/src/main/java/org/opendaylight/yangtools/yang/parser/stmt/rfc6020/IdentityRefSpecificationImpl.java
@@ -8,29 +8,30 @@
 package org.opendaylight.yangtools.yang.parser.stmt.rfc6020;
 
 import com.google.common.base.Preconditions;
+import java.util.Collection;
 import javax.annotation.Nonnull;
 import org.opendaylight.yangtools.yang.common.QName;
-import org.opendaylight.yangtools.yang.model.api.Rfc6020Mapping;
+import org.opendaylight.yangtools.yang.model.api.YangStmtMapping;
 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
 import org.opendaylight.yangtools.yang.model.api.stmt.BaseStatement;
 import org.opendaylight.yangtools.yang.model.api.stmt.IdentityStatement;
 import org.opendaylight.yangtools.yang.model.api.stmt.TypeStatement;
 import org.opendaylight.yangtools.yang.parser.spi.IdentityNamespace;
-import org.opendaylight.yangtools.yang.parser.spi.SubstatementValidator;
 import org.opendaylight.yangtools.yang.parser.spi.meta.AbstractDeclaredStatement;
 import org.opendaylight.yangtools.yang.parser.spi.meta.AbstractStatementSupport;
 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContextUtils;
+import org.opendaylight.yangtools.yang.parser.spi.meta.SubstatementValidator;
 import org.opendaylight.yangtools.yang.parser.stmt.rfc6020.effective.type.IdentityRefSpecificationEffectiveStatementImpl;
 
 public class IdentityRefSpecificationImpl extends AbstractDeclaredStatement<String> implements TypeStatement.IdentityRefSpecification {
-    private static final SubstatementValidator SUBSTATEMENT_VALIDATOR = SubstatementValidator.builder(Rfc6020Mapping
+    private static final SubstatementValidator SUBSTATEMENT_VALIDATOR = SubstatementValidator.builder(YangStmtMapping
             .TYPE)
-            .add(Rfc6020Mapping.BASE, 1, 1)
+            .addMandatory(YangStmtMapping.BASE)
             .build();
 
     protected IdentityRefSpecificationImpl(
-            StmtContext<String, TypeStatement.IdentityRefSpecification, ?> context) {
+            final StmtContext<String, TypeStatement.IdentityRefSpecification, ?> context) {
         super(context);
     }
 
@@ -39,51 +40,52 @@ public class IdentityRefSpecificationImpl extends AbstractDeclaredStatement<Stri
             AbstractStatementSupport<String, TypeStatement.IdentityRefSpecification, EffectiveStatement<String, TypeStatement.IdentityRefSpecification>> {
 
         public Definition() {
-            super(Rfc6020Mapping.TYPE);
+            super(YangStmtMapping.TYPE);
         }
 
         @Override
-        public String parseArgumentValue(StmtContext<?, ?, ?> ctx, String value) {
+        public String parseArgumentValue(final StmtContext<?, ?, ?> ctx, final String value) {
             return value;
         }
 
         @Override
         public TypeStatement.IdentityRefSpecification createDeclared(
-                StmtContext<String, TypeStatement.IdentityRefSpecification, ?> ctx) {
+                final StmtContext<String, TypeStatement.IdentityRefSpecification, ?> ctx) {
             return new IdentityRefSpecificationImpl(ctx);
         }
 
         @Override
         public EffectiveStatement<String, TypeStatement.IdentityRefSpecification> createEffective(
-                StmtContext<String, TypeStatement.IdentityRefSpecification, EffectiveStatement<String, TypeStatement
+                final StmtContext<String, TypeStatement.IdentityRefSpecification, EffectiveStatement<String, TypeStatement
                         .IdentityRefSpecification>> ctx) {
             return new IdentityRefSpecificationEffectiveStatementImpl(ctx);
         }
 
         @Override
-        public void onFullDefinitionDeclared(StmtContext.Mutable<String, IdentityRefSpecification,
+        public void onFullDefinitionDeclared(final StmtContext.Mutable<String, IdentityRefSpecification,
                 EffectiveStatement<String, IdentityRefSpecification>> stmt) {
-            final StmtContext<QName, ?, ?> baseStmt = StmtContextUtils.findFirstDeclaredSubstatement(stmt,
-                    BaseStatement.class);
-            Preconditions.checkArgument(baseStmt != null, "The \"base\" statement, which is a substatement to the " +
-                    "\"type\"\n statement, MUST be present if the type is \"identityref\" in source '%s'", stmt
-                    .getStatementSourceReference());
-            final QName baseIdentity = baseStmt.getStatementArgument();
-            final StmtContext<?, IdentityStatement, EffectiveStatement<QName, IdentityStatement>> stmtCtx = stmt
-                    .getFromNamespace(IdentityNamespace.class, baseIdentity);
-            Preconditions.checkArgument(stmtCtx != null, "Referenced base identity '%s' doesn't exist " +
-                        "in " + "given scope " + "(module, imported submodules), source: '%s'", baseIdentity
-                    .getLocalName(), stmt.getStatementSourceReference());
+            super.onFullDefinitionDeclared(stmt);
+
+            final Collection<StmtContext<QName, BaseStatement, ?>> baseStatements =
+                    StmtContextUtils.<QName, BaseStatement>findAllDeclaredSubstatements(stmt, BaseStatement.class);
+
+            for (StmtContext<QName, BaseStatement, ?> baseStmt : baseStatements) {
+                final QName baseIdentity = baseStmt.getStatementArgument();
+                final StmtContext<?, IdentityStatement, EffectiveStatement<QName, IdentityStatement>> stmtCtx =
+                        stmt.getFromNamespace(IdentityNamespace.class, baseIdentity);
+                Preconditions.checkArgument(stmtCtx != null, "Referenced base identity '%s' doesn't exist " +
+                        "in given scope (module, imported modules, submodules), source: '%s'",
+                        baseIdentity.getLocalName(), stmt.getStatementSourceReference());
+            }
         }
 
         @Override
-        public void onStatementDefinitionDeclared(StmtContext.Mutable<String, IdentityRefSpecification,
-                EffectiveStatement<String, IdentityRefSpecification>> stmt) {
-            super.onStatementDefinitionDeclared(stmt);
-            SUBSTATEMENT_VALIDATOR.validate(stmt);
+        protected SubstatementValidator getSubstatementValidator() {
+            return SUBSTATEMENT_VALIDATOR;
         }
     }
 
+    @Nonnull
     @Override
     public String getName() {
         return argument();
@@ -95,4 +97,10 @@ public class IdentityRefSpecificationImpl extends AbstractDeclaredStatement<Stri
         return firstDeclared(BaseStatement.class);
     }
 
+    @Nonnull
+    @Override
+    public Collection<? extends BaseStatement> getBases() {
+        return allDeclared(BaseStatement.class);
+    }
+
 }