commit 64a54c7062c8250c42c0b641c8788b6ea1e35ca5
author sonugupta-huawei <sonu.gupta1289@gmail.com> Mon Jul 31 21:13:01 2017 +0530
committer sonugupta-huawei <sonu.gupta1289@gmail.com> Mon Jul 31 21:13:16 2017 +0530
tree e3e2f899239f39049c7d76f75064124681bd1d6d
parent 5ccbac3fb0ccbbe1b07e4cd7809858cb0e89fa98

model registration using classes issue

Change-Id: I6eae322e3a4653ee5c403c32ac59cd4092c313ac

runtime/src/main/java/org/onosproject/yang/runtime/impl/DefaultYangModelRegistry.java

diff --git a/runtime/src/main/java/org/onosproject/yang/runtime/impl/DefaultYangModelRegistry.java b/runtime/src/main/java/org/onosproject/yang/runtime/impl/DefaultYangModelRegistry.java
index bd684e6..87f6c78 100644
--- a/runtime/src/main/java/org/onosproject/yang/runtime/impl/DefaultYangModelRegistry.java
+++ b/runtime/src/main/java/org/onosproject/yang/runtime/impl/DefaultYangModelRegistry.java
@@ -125,12 +125,22 @@
         Set<YangNode> curNodes = getNodes(model);
 
         //adding class info if added by application.
-        updateRegClassStore(param);
+        AppModuleInfo info = null;
+        for (YangModuleId id : param.getYangModel().getYangModulesId()) {
+            info = param.getAppModuleInfo(id);
+            if (info != null) {
+                break;
+            }
+        }
 
-        if (modelIdStore.containsKey(model.getYangModelId())) {
-            throw new IllegalArgumentException(E_MEXIST);
+        if (!modelIdStore.containsKey(model.getYangModelId())) {
+            updateRegClassStore(param);
+            modelIdStore.put(model.getYangModelId(), model);
+        } else if ((info != null) && (!registerClassStore
+                .containsKey(info.getModuleClass()))) {
+            updateRegClassStore(param);
         } else {
-            modelIdStore.put(model.getYangModelId(), param.getYangModel());
+            throw new IllegalArgumentException(E_MEXIST);
         }
 
         //Register all the YANG nodes.