Correct NonNull annotation

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

diff --git a/yang/yang-model-api/src/main/java/org/opendaylight/yangtools/yang/model/api/stmt/KeyStatement.java b/yang/yang-model-api/src/main/java/org/opendaylight/yangtools/yang/model/api/stmt/KeyStatement.java
index e015604600aa1545b66d40965657e2d5dbb12f78..ff82ba55bf188bbffca7c6d39e4ac37c1fa03585 100644 (file)
--- a/yang/yang-model-api/src/main/java/org/opendaylight/yangtools/yang/model/api/stmt/KeyStatement.java
+++ b/yang/yang-model-api/src/main/java/org/opendaylight/yangtools/yang/model/api/stmt/KeyStatement.java
@@ -7,10 +7,15 @@
  */
 package org.opendaylight.yangtools.yang.model.api.stmt;
 
-import java.util.Collection;
+import java.util.Set;
+import org.opendaylight.yangtools.yang.common.QName;
+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.StatementDefinition;
 
-public interface KeyStatement extends DeclaredStatement<Collection<SchemaNodeIdentifier>> {
-
+public interface KeyStatement extends DeclaredStatement<Set<QName>> {
+    @Override
+    default StatementDefinition statementDefinition() {
+        return YangStmtMapping.KEY;
+    }
 }
-