Refactor ExtensionStatementSupport

[yangtools.git] / yang / yang-model-api / src / main / java / org / opendaylight / yangtools / yang / model / api / stmt / ExtensionStatement.java

diff --git a/yang/yang-model-api/src/main/java/org/opendaylight/yangtools/yang/model/api/stmt/ExtensionStatement.java b/yang/yang-model-api/src/main/java/org/opendaylight/yangtools/yang/model/api/stmt/ExtensionStatement.java
index 01f712047d4dac462f0d5997b125bff45f17e4f3..b97835cbede43c17939445859b814c6a8f1252a0 100644 (file)
--- a/yang/yang-model-api/src/main/java/org/opendaylight/yangtools/yang/model/api/stmt/ExtensionStatement.java
+++ b/yang/yang-model-api/src/main/java/org/opendaylight/yangtools/yang/model/api/stmt/ExtensionStatement.java
@@ -10,8 +10,15 @@ package org.opendaylight.yangtools.yang.model.api.stmt;
 import java.util.Optional;
 import org.eclipse.jdt.annotation.Nullable;
 import org.opendaylight.yangtools.yang.common.QName;
+import org.opendaylight.yangtools.yang.model.api.YangStmtMapping;
+import org.opendaylight.yangtools.yang.model.api.meta.StatementDefinition;
 
 public interface ExtensionStatement extends DocumentedDeclaredStatement.WithStatus<QName> {
+    @Override
+    default StatementDefinition statementDefinition() {
+        return YangStmtMapping.EXTENSION;
+    }
+
     default @Nullable ArgumentStatement getArgument() {
         final Optional<ArgumentStatement> opt = findFirstDeclaredSubstatement(ArgumentStatement.class);
         return opt.isPresent() ? opt.get() : null;