Migrate get{PublicDefinition,StatementSource} callers

[yangtools.git] / yang / yang-parser-reactor / src / main / java / org / opendaylight / yangtools / yang / parser / stmt / reactor / StatementContextWriter.java

diff --git a/yang/yang-parser-reactor/src/main/java/org/opendaylight/yangtools/yang/parser/stmt/reactor/StatementContextWriter.java b/yang/yang-parser-reactor/src/main/java/org/opendaylight/yangtools/yang/parser/stmt/reactor/StatementContextWriter.java
index c0ef1180af43447f01c608f4cf2f9cc6d1a0c682..6f29dff5f968a601535e439e7d27642705579747 100644 (file)
--- a/yang/yang-parser-reactor/src/main/java/org/opendaylight/yangtools/yang/parser/stmt/reactor/StatementContextWriter.java
+++ b/yang/yang-parser-reactor/src/main/java/org/opendaylight/yangtools/yang/parser/stmt/reactor/StatementContextWriter.java
@@ -88,7 +88,7 @@ final class StatementContextWriter implements StatementWriter {
         //       - allow endDeclared() to be moved to AbstractResumedStatement
         //       - remove the need for verify()
         StatementContextBase<?, ?, ?> parentContext = current.getParentContext();
-        while (parentContext != null && StatementSource.CONTEXT == parentContext.getStatementSource()) {
+        while (parentContext != null && StatementSource.CONTEXT == parentContext.source()) {
             parentContext.endDeclared(phase);
             parentContext = parentContext.getParentContext();
         }
@@ -108,7 +108,7 @@ final class StatementContextWriter implements StatementWriter {
 
         // Fast path: we are entering a statement which was emitted in previous phase
         AbstractResumedStatement<?, ?, ?> existing = current.lookupSubstatement(childId);
-        while (existing != null && StatementSource.CONTEXT == existing.getStatementSource()) {
+        while (existing != null && StatementSource.CONTEXT == existing.source()) {
             existing = existing.lookupSubstatement(childId);
         }