commit 2737d2a168e0e563e78e9285614ed55136f2e12a
author Gaurav Agrawal <gaurav.agrawal@huawei.com> Sat Feb 13 14:23:40 2016 +0530
committer Gerrit Code Review <gerrit@onlab.us> Thu Feb 18 09:00:17 2016 +0000
tree 044800021b25dde06b86f7e45c92ec78bb860a93
parent 2d45f37915b3a6bdc735d556cbcf77e38081e328

[ONOS-3880, 3881] Yang Listener for Module and Sub-Module

Change-Id: Iee75c3e04af9b66ebc38acb3396aa4c54af5a268

src/main/java/org/onosproject/yangutils/parser/impl/listeners/BaseFileListener.java

diff --git a/src/main/java/org/onosproject/yangutils/parser/impl/listeners/BaseFileListener.java b/src/main/java/org/onosproject/yangutils/parser/impl/listeners/BaseFileListener.java
index 355c268..f53a767 100644
--- a/src/main/java/org/onosproject/yangutils/parser/impl/listeners/BaseFileListener.java
+++ b/src/main/java/org/onosproject/yangutils/parser/impl/listeners/BaseFileListener.java
@@ -16,8 +16,17 @@
 
 package org.onosproject.yangutils.parser.impl.listeners;
 
+import org.onosproject.yangutils.datamodel.YangModule;
+import org.onosproject.yangutils.datamodel.YangNode;
+import org.onosproject.yangutils.datamodel.YangSubModule;
+import org.onosproject.yangutils.parser.ParsableDataType;
 import org.onosproject.yangutils.parser.antlrgencode.GeneratedYangParser;
+import org.onosproject.yangutils.parser.exceptions.ParserException;
 import org.onosproject.yangutils.parser.impl.TreeWalkListener;
+import org.onosproject.yangutils.parser.impl.parserutils.ListenerErrorLocation;
+import org.onosproject.yangutils.parser.impl.parserutils.ListenerErrorMessageConstruction;
+import org.onosproject.yangutils.parser.impl.parserutils.ListenerErrorType;
+import org.onosproject.yangutils.parser.impl.parserutils.ListenerValidation;
 
 /*
  * Reference: RFC6020 and YANG ANTLR Grammar
@@ -48,7 +57,11 @@
      * @param ctx context object of the grammar rule.
      */
     public static void processYangFileEntry(TreeWalkListener listener, GeneratedYangParser.YangfileContext ctx) {
-        // TODO method implementation
+
+        // Check if stack is empty.
+        ListenerValidation.checkStackIsEmpty(listener, ListenerErrorType.INVALID_HOLDER,
+                                             ParsableDataType.YANGBASE_DATA, "", ListenerErrorLocation.ENTRY);
+
     }
 
     /**
@@ -59,6 +72,25 @@
      * @param ctx context object of the grammar rule.
      */
     public static void processYangFileExit(TreeWalkListener listener, GeneratedYangParser.YangfileContext ctx) {
-        // TODO method implementation
+
+        // Check for stack to be non empty.
+        ListenerValidation.checkStackIsNotEmpty(listener, ListenerErrorType.MISSING_HOLDER,
+                                                ParsableDataType.YANGBASE_DATA, "", ListenerErrorLocation.EXIT);
+
+        // Data Model tree root node is set.
+        if (listener.getParsedDataStack().peek() instanceof YangModule
+                | listener.getParsedDataStack().peek() instanceof YangSubModule) {
+            listener.setRootNode((YangNode) listener.getParsedDataStack().pop());
+        } else {
+            throw new ParserException(
+                                      ListenerErrorMessageConstruction
+                                              .constructListenerErrorMessage(ListenerErrorType.INVALID_CHILD,
+                                                                             ParsableDataType.YANGBASE_DATA, "",
+                                                                             ListenerErrorLocation.EXIT));
+        }
+
+        // Check if stack is empty.
+        ListenerValidation.checkStackIsEmpty(listener, ListenerErrorType.INVALID_HOLDER,
+                                             ParsableDataType.YANGBASE_DATA, "", ListenerErrorLocation.EXIT);
     }
-}
+}
\ No newline at end of file