Refactor simple statements

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

diff --git a/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/error_message/ErrorMessageStatementSupport.java b/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/error_message/ErrorMessageStatementSupport.java
index fa74543cba8fd5f13a189358e127b5d22f8742a0..0d404ef5dbc312310234da581d1ccb1a5f5f2313 100644 (file)
--- a/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/error_message/ErrorMessageStatementSupport.java
+++ b/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/error_message/ErrorMessageStatementSupport.java
@@ -7,18 +7,20 @@
  */
 package org.opendaylight.yangtools.yang.parser.rfc7950.stmt.error_message;
 
+import com.google.common.collect.ImmutableList;
 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.ErrorMessageEffectiveStatement;
 import org.opendaylight.yangtools.yang.model.api.stmt.ErrorMessageStatement;
-import org.opendaylight.yangtools.yang.parser.spi.meta.AbstractStatementSupport;
+import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.BaseStringStatementSupport;
 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
 import org.opendaylight.yangtools.yang.parser.spi.meta.SubstatementValidator;
 
-public final class ErrorMessageStatementSupport extends
-        AbstractStatementSupport<String, ErrorMessageStatement, EffectiveStatement<String, ErrorMessageStatement>> {
-    private static final SubstatementValidator SUBSTATEMENT_VALIDATOR = SubstatementValidator.builder(YangStmtMapping
-        .ERROR_MESSAGE)
-        .build();
+public final class ErrorMessageStatementSupport
+        extends BaseStringStatementSupport<ErrorMessageStatement, ErrorMessageEffectiveStatement> {
+    private static final SubstatementValidator SUBSTATEMENT_VALIDATOR =
+            SubstatementValidator.builder(YangStmtMapping.ERROR_MESSAGE).build();
     private static final ErrorMessageStatementSupport INSTANCE = new ErrorMessageStatementSupport();
 
     private ErrorMessageStatementSupport() {
@@ -30,24 +32,33 @@ public final class ErrorMessageStatementSupport extends
     }
 
     @Override
-    public String parseArgumentValue(final StmtContext<?, ?, ?> ctx, final String value) {
-        return value;
+    protected SubstatementValidator getSubstatementValidator() {
+        return SUBSTATEMENT_VALIDATOR;
     }
 
     @Override
-    public ErrorMessageStatement createDeclared(final StmtContext<String, ErrorMessageStatement, ?> ctx) {
-        return new ErrorMessageStatementImpl(ctx);
+    protected ErrorMessageStatement createDeclared(final StmtContext<String, ErrorMessageStatement, ?> ctx,
+            final ImmutableList<? extends DeclaredStatement<?>> substatements) {
+        return new RegularErrorMessageStatement(ctx.coerceRawStatementArgument(), substatements);
     }
 
     @Override
-    public EffectiveStatement<String, ErrorMessageStatement> createEffective(
-            final StmtContext<String, ErrorMessageStatement,
-            EffectiveStatement<String, ErrorMessageStatement>> ctx) {
-        return new ErrorMessageEffectiveStatementImpl(ctx);
+    protected ErrorMessageStatement createEmptyDeclared(final StmtContext<String, ErrorMessageStatement, ?> ctx) {
+        return new EmptyErrorMessageStatement(ctx.coerceRawStatementArgument());
     }
 
     @Override
-    protected SubstatementValidator getSubstatementValidator() {
-        return SUBSTATEMENT_VALIDATOR;
+    protected ErrorMessageEffectiveStatement createEffective(
+            final StmtContext<String, ErrorMessageStatement, ErrorMessageEffectiveStatement> ctx,
+            final ErrorMessageStatement declared,
+            final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
+        return new RegularErrorMessageEffectiveStatement(declared, substatements);
+    }
+
+    @Override
+    protected ErrorMessageEffectiveStatement createEmptyEffective(
+            final StmtContext<String, ErrorMessageStatement, ErrorMessageEffectiveStatement> ctx,
+            final ErrorMessageStatement declared) {
+        return new EmptyErrorMessageEffectiveStatement(declared);
     }
-}
\ No newline at end of file
+}