All StatementSupports must support DeclarationReference

[yangtools.git] / yang / odlext-parser-support / src / main / java / org / opendaylight / yangtools / odlext / parser / AugmentIdentifierStatementSupport.java

diff --git a/yang/odlext-parser-support/src/main/java/org/opendaylight/yangtools/odlext/parser/AugmentIdentifierStatementSupport.java b/yang/odlext-parser-support/src/main/java/org/opendaylight/yangtools/odlext/parser/AugmentIdentifierStatementSupport.java
index 3b6ff348447776201f1ee62ae8b63b4cca7d126e..dc1077901f5344a38d90ff846e24f3cf2a057980 100644 (file)
--- a/yang/odlext-parser-support/src/main/java/org/opendaylight/yangtools/odlext/parser/AugmentIdentifierStatementSupport.java
+++ b/yang/odlext-parser-support/src/main/java/org/opendaylight/yangtools/odlext/parser/AugmentIdentifierStatementSupport.java
@@ -13,6 +13,7 @@ import org.opendaylight.yangtools.odlext.model.api.AugmentIdentifierEffectiveSta
 import org.opendaylight.yangtools.odlext.model.api.AugmentIdentifierStatement;
 import org.opendaylight.yangtools.odlext.model.api.OpenDaylightExtensionsStatements;
 import org.opendaylight.yangtools.yang.common.UnqualifiedQName;
+import org.opendaylight.yangtools.yang.model.api.meta.DeclarationReference;
 import org.opendaylight.yangtools.yang.model.api.meta.DeclaredStatement;
 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
 import org.opendaylight.yangtools.yang.parser.api.YangParserConfiguration;
@@ -54,6 +55,12 @@ public final class AugmentIdentifierStatementSupport
         return new AugmentIdentifierStatementImpl(ctx.getArgument(), substatements);
     }
 
+    @Override
+    protected AugmentIdentifierStatement attachDeclarationReference(final AugmentIdentifierStatement stmt,
+            final DeclarationReference reference) {
+        return new RefAugmentIdentifierStatement(stmt, reference);
+    }
+
     @Override
     protected AugmentIdentifierEffectiveStatement createEffective(
             final Current<UnqualifiedQName, AugmentIdentifierStatement> stmt,