*/
package org.opendaylight.yangtools.yang.model.api.stmt;
-import javax.annotation.Nullable;
+import java.util.Optional;
+import org.eclipse.jdt.annotation.Nullable;
import org.opendaylight.yangtools.yang.common.QName;
-import org.opendaylight.yangtools.yang.model.api.meta.DeclaredStatement;
-
-public interface ExtensionStatement extends DeclaredStatement<QName>, DocumentationGroup.WithStatus {
-
- @Nullable ArgumentStatement getArgument();
-
+public interface ExtensionStatement extends DocumentedDeclaredStatement.WithStatus<QName> {
+ default @Nullable ArgumentStatement getArgument() {
+ final Optional<ArgumentStatement> opt = findFirstDeclaredSubstatement(ArgumentStatement.class);
+ return opt.isPresent() ? opt.get() : null;
+ }
}