commit cdc4b8ccca7cde2c4383146ca0ff20d9c51f912a
author Claudine Chiu <cchiu@ciena.com> Thu Mar 30 00:34:39 2017 -0400
committer Yoonseon Han <yoonseon@onlab.us> Wed Apr 26 00:41:20 2017 +0000
tree 95e7ae3c04c6bae30e66e1fcf6c801b4c2f92d52
parent 90492529c3eaaceb1eb095a02bad78af73d3804f

ONOS-6204 initial impl of dist. flow objective store

Change-Id: I968bea6f95d91b5b30e9d0e11c6948502684022c

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

diff --git a/incubator/store/src/main/java/org/onosproject/incubator/store/virtual/impl/SimpleVirtualFlowObjectiveStore.java b/incubator/store/src/main/java/org/onosproject/incubator/store/virtual/impl/SimpleVirtualFlowObjectiveStore.java
index 1ac5396..e2ecdf3 100644
--- a/incubator/store/src/main/java/org/onosproject/incubator/store/virtual/impl/SimpleVirtualFlowObjectiveStore.java
+++ b/incubator/store/src/main/java/org/onosproject/incubator/store/virtual/impl/SimpleVirtualFlowObjectiveStore.java
@@ -72,13 +72,25 @@
         eventQ = new LinkedBlockingQueue<>();
         tpool.execute(new FlowObjectiveNotifier());
 
-        nextGroupsMap = Maps.newConcurrentMap();
+        initNextGroupsMap();
 
         nextIds = storageService.getAtomicCounter("next-objective-counter");
         log.info("Started");
     }
 
-    private ConcurrentMap<Integer, byte[]> getNextGroups(NetworkId networkId) {
+    public void deactivate() {
+        log.info("Stopped");
+    }
+
+    protected void initNextGroupsMap() {
+        nextGroupsMap = Maps.newConcurrentMap();
+    }
+
+    protected void updateNextGroupsMap(NetworkId networkId,
+                                       ConcurrentMap<Integer, byte[]> nextGroups) {
+    }
+
+    protected ConcurrentMap<Integer, byte[]> getNextGroups(NetworkId networkId) {
         nextGroupsMap.computeIfAbsent(networkId, n -> Maps.newConcurrentMap());
         return nextGroupsMap.get(networkId);
     }
@@ -87,6 +99,7 @@
     public void putNextGroup(NetworkId networkId, Integer nextId, NextGroup group) {
         ConcurrentMap<Integer, byte[]> nextGroups = getNextGroups(networkId);
         nextGroups.put(nextId, group.data());
+        updateNextGroupsMap(networkId, nextGroups);
 
         eventQ.add(new VirtualObjectiveEvent(networkId, ObjectiveEvent.Type.ADD, nextId));
     }
@@ -94,14 +107,22 @@
     @Override
     public NextGroup getNextGroup(NetworkId networkId, Integer nextId) {
         ConcurrentMap<Integer, byte[]> nextGroups = getNextGroups(networkId);
-        return new DefaultNextGroup(nextGroups.get(nextId));
+        byte[] groupData = nextGroups.get(nextId);
+        if (groupData != null) {
+            return new DefaultNextGroup(groupData);
+        }
+        return null;
     }
 
     @Override
     public NextGroup removeNextGroup(NetworkId networkId, Integer nextId) {
         ConcurrentMap<Integer, byte[]> nextGroups = getNextGroups(networkId);
+        byte[] nextGroup = nextGroups.remove(nextId);
+        updateNextGroupsMap(networkId, nextGroups);
+
         eventQ.add(new VirtualObjectiveEvent(networkId, ObjectiveEvent.Type.REMOVE, nextId));
-        return new DefaultNextGroup(nextGroups.remove(nextId));
+
+        return new DefaultNextGroup(nextGroup);
     }
 
     @Override