commit f70bf462f8ac59446d399be8866b2647d75ca4c8
author Jordan Halterman <jordan@opennetworking.org> Sat Jul 29 13:12:00 2017 -0700
committer Ray Milkey <ray@opennetworking.org> Tue Aug 01 16:03:57 2017 +0000
tree 8135610f8b9ff40585107d6a55a6b81affc15379
parent a966c34402da0f306576ce4cd90c8e2d418cb47d

[ONOS-6756] Replicate node version information for ISSU

Change-Id: Ibd31c573990f2732b7abf8615ca914ffb77615ec

incubator/store/src/main/java/org/onosproject/incubator/store/virtual/impl/SimpleVirtualMastershipStore.java

diff --git a/incubator/store/src/main/java/org/onosproject/incubator/store/virtual/impl/SimpleVirtualMastershipStore.java b/incubator/store/src/main/java/org/onosproject/incubator/store/virtual/impl/SimpleVirtualMastershipStore.java
index 47b9c6f..c390513 100644
--- a/incubator/store/src/main/java/org/onosproject/incubator/store/virtual/impl/SimpleVirtualMastershipStore.java
+++ b/incubator/store/src/main/java/org/onosproject/incubator/store/virtual/impl/SimpleVirtualMastershipStore.java
@@ -32,6 +32,8 @@
 import org.onosproject.cluster.DefaultControllerNode;
 import org.onosproject.cluster.NodeId;
 import org.onosproject.cluster.RoleInfo;
+import org.onosproject.core.Version;
+import org.onosproject.core.VersionService;
 import org.onosproject.incubator.net.virtual.NetworkId;
 import org.onosproject.incubator.net.virtual.VirtualNetworkMastershipStore;
 import org.onosproject.mastership.MastershipEvent;
@@ -74,6 +76,9 @@
     @Reference(cardinality = ReferenceCardinality.MANDATORY_UNARY)
     protected ClusterService clusterService;
 
+    @Reference(cardinality = ReferenceCardinality.MANDATORY_UNARY)
+    protected VersionService versionService;
+
     //devices mapped to their masters, to emulate multiple nodes
     protected final Map<NetworkId, Map<DeviceId, NodeId>> masterMapByNetwork =
             new HashMap<>();
@@ -479,6 +484,14 @@
             }
 
             @Override
+            public Version getVersion(NodeId nodeId) {
+                if (instance.id().equals(nodeId)) {
+                    return versionService.version();
+                }
+                return null;
+            }
+
+            @Override
             public DateTime getLastUpdated(NodeId nodeId) {
                 return creationTime;
             }