Seal ModelStatement

[yangtools.git] / model / yang-model-spi / src / main / java / org / opendaylight / yangtools / yang / model / spi / meta / ForwardingDeclaredStatement.java

diff --git a/model/yang-model-spi/src/main/java/org/opendaylight/yangtools/yang/model/spi/meta/ForwardingDeclaredStatement.java b/model/yang-model-spi/src/main/java/org/opendaylight/yangtools/yang/model/spi/meta/ForwardingDeclaredStatement.java
index 5ed27974b3b5367a1557f1100deebe74650297a1..f26a1355d1e9acddaebaeaa284a5228bfa791e14 100644 (file)
--- a/model/yang-model-spi/src/main/java/org/opendaylight/yangtools/yang/model/spi/meta/ForwardingDeclaredStatement.java
+++ b/model/yang-model-spi/src/main/java/org/opendaylight/yangtools/yang/model/spi/meta/ForwardingDeclaredStatement.java
@@ -8,18 +8,30 @@
 package org.opendaylight.yangtools.yang.model.spi.meta;
 
 import com.google.common.annotations.Beta;
+import com.google.common.collect.ForwardingObject;
 import java.util.List;
 import java.util.Optional;
 import org.eclipse.jdt.annotation.NonNull;
 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.StatementDefinition;
 
 /**
  * Common base class for forwarding implementations of {@link DeclaredStatement}.
  */
 @Beta
 public abstract class ForwardingDeclaredStatement<A, D extends DeclaredStatement<A>>
-        extends ForwardingModelStatement<A, D> implements DeclaredStatement<A> {
+        extends ForwardingObject implements DeclaredStatement<A> {
+    @Override
+    public StatementDefinition statementDefinition() {
+        return delegate().statementDefinition();
+    }
+
+    @Override
+    public A argument() {
+        return delegate().argument();
+    }
+
     @Override
     public String rawArgument() {
         return delegate().rawArgument();