blob: ab66bd47b33f5659d5160171fb8a37800381f884 [file] [log] [blame]
alshabib10580802015-02-18 18:30:33 -08001/*
Brian O'Connor5ab426f2016-04-09 01:19:45 -07002 * Copyright 2015-present Open Networking Laboratory
alshabib10580802015-02-18 18:30:33 -08003 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16package org.onosproject.store.group.impl;
17
Jonathan Hart6ec029a2015-03-24 17:12:35 -070018import com.google.common.collect.FluentIterable;
Charles Chanf4838a72015-12-07 18:13:45 -080019import com.google.common.collect.ImmutableSet;
Jonathan Hart6ec029a2015-03-24 17:12:35 -070020import com.google.common.collect.Iterables;
Srikanth Vavilapalli23181912015-05-04 09:48:09 -070021import com.google.common.collect.Sets;
alshabib10580802015-02-18 18:30:33 -080022import org.apache.felix.scr.annotations.Activate;
23import org.apache.felix.scr.annotations.Component;
24import org.apache.felix.scr.annotations.Deactivate;
alshabibb0285992016-03-28 23:30:37 -070025import org.apache.felix.scr.annotations.Modified;
26import org.apache.felix.scr.annotations.Property;
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -070027import org.apache.felix.scr.annotations.Reference;
28import org.apache.felix.scr.annotations.ReferenceCardinality;
alshabib10580802015-02-18 18:30:33 -080029import org.apache.felix.scr.annotations.Service;
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -070030import org.onlab.util.KryoNamespace;
alshabibb0285992016-03-28 23:30:37 -070031import org.onosproject.cfg.ComponentConfigService;
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -070032import org.onosproject.cluster.ClusterService;
Charles Chanf4838a72015-12-07 18:13:45 -080033import org.onosproject.cluster.NodeId;
alshabib10580802015-02-18 18:30:33 -080034import org.onosproject.core.DefaultGroupId;
35import org.onosproject.core.GroupId;
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -070036import org.onosproject.mastership.MastershipService;
alshabib10580802015-02-18 18:30:33 -080037import org.onosproject.net.DeviceId;
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -070038import org.onosproject.net.MastershipRole;
alshabib10580802015-02-18 18:30:33 -080039import org.onosproject.net.group.DefaultGroup;
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -070040import org.onosproject.net.group.DefaultGroupBucket;
alshabib10580802015-02-18 18:30:33 -080041import org.onosproject.net.group.DefaultGroupDescription;
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -070042import org.onosproject.net.group.DefaultGroupKey;
alshabib10580802015-02-18 18:30:33 -080043import org.onosproject.net.group.Group;
44import org.onosproject.net.group.Group.GroupState;
45import org.onosproject.net.group.GroupBucket;
46import org.onosproject.net.group.GroupBuckets;
47import org.onosproject.net.group.GroupDescription;
48import org.onosproject.net.group.GroupEvent;
49import org.onosproject.net.group.GroupEvent.Type;
50import org.onosproject.net.group.GroupKey;
51import org.onosproject.net.group.GroupOperation;
52import org.onosproject.net.group.GroupStore;
53import org.onosproject.net.group.GroupStoreDelegate;
Srikanth Vavilapalli10e75cd2015-04-13 16:21:24 -070054import org.onosproject.net.group.StoredGroupBucketEntry;
alshabib10580802015-02-18 18:30:33 -080055import org.onosproject.net.group.StoredGroupEntry;
56import org.onosproject.store.AbstractStore;
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -070057import org.onosproject.store.cluster.messaging.ClusterCommunicationService;
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -070058import org.onosproject.store.serializers.KryoNamespaces;
Madan Jampani0b847532016-03-03 13:44:15 -080059import org.onosproject.store.service.ConsistentMap;
60import org.onosproject.store.service.MapEvent;
61import org.onosproject.store.service.MapEventListener;
alshabibb0285992016-03-28 23:30:37 -070062import org.onosproject.store.service.MultiValuedTimestamp;
Madan Jampani0b847532016-03-03 13:44:15 -080063import org.onosproject.store.service.Serializer;
Jonathan Hart6ec029a2015-03-24 17:12:35 -070064import org.onosproject.store.service.StorageService;
helenyrwua1c41152016-08-18 16:16:14 -070065import org.onosproject.store.service.Topic;
Madan Jampani0b847532016-03-03 13:44:15 -080066import org.onosproject.store.service.Versioned;
alshabibb0285992016-03-28 23:30:37 -070067import org.osgi.service.component.ComponentContext;
alshabib10580802015-02-18 18:30:33 -080068import org.slf4j.Logger;
69
Jonathan Hart6ec029a2015-03-24 17:12:35 -070070import java.util.ArrayList;
Srikanth Vavilapalli23181912015-05-04 09:48:09 -070071import java.util.Collection;
Charles Chanf4838a72015-12-07 18:13:45 -080072import java.util.Collections;
alshabibb0285992016-03-28 23:30:37 -070073import java.util.Dictionary;
Jonathan Hart6ec029a2015-03-24 17:12:35 -070074import java.util.HashMap;
Charles Chan0c7c43b2016-01-14 17:39:20 -080075import java.util.HashSet;
Srikanth Vavilapalli23181912015-05-04 09:48:09 -070076import java.util.Iterator;
Jonathan Hart6ec029a2015-03-24 17:12:35 -070077import java.util.List;
Madan Jampani0b847532016-03-03 13:44:15 -080078import java.util.Map;
Charles Chan0c7c43b2016-01-14 17:39:20 -080079import java.util.Map.Entry;
Jonathan Hart6ec029a2015-03-24 17:12:35 -070080import java.util.Objects;
Sho SHIMIZU30d639b2015-05-05 09:30:35 -070081import java.util.Optional;
alshabibb0285992016-03-28 23:30:37 -070082import java.util.Properties;
Srikanth Vavilapalli23181912015-05-04 09:48:09 -070083import java.util.Set;
Jonathan Hart6ec029a2015-03-24 17:12:35 -070084import java.util.concurrent.ConcurrentHashMap;
85import java.util.concurrent.ConcurrentMap;
86import java.util.concurrent.ExecutorService;
87import java.util.concurrent.Executors;
88import java.util.concurrent.atomic.AtomicInteger;
Jonathan Hart6ec029a2015-03-24 17:12:35 -070089import java.util.stream.Collectors;
90
alshabibb0285992016-03-28 23:30:37 -070091import static com.google.common.base.Strings.isNullOrEmpty;
alshabibb0285992016-03-28 23:30:37 -070092import static org.onlab.util.Tools.get;
Jonathan Hart6ec029a2015-03-24 17:12:35 -070093import static org.onlab.util.Tools.groupedThreads;
94import static org.slf4j.LoggerFactory.getLogger;
alshabib10580802015-02-18 18:30:33 -080095
96/**
Saurav Das0fd79d92016-03-07 10:58:36 -080097 * Manages inventory of group entries using distributed group stores from the
98 * storage service.
alshabib10580802015-02-18 18:30:33 -080099 */
100@Component(immediate = true)
101@Service
102public class DistributedGroupStore
103 extends AbstractStore<GroupEvent, GroupStoreDelegate>
104 implements GroupStore {
105
106 private final Logger log = getLogger(getClass());
107
alshabibb0285992016-03-28 23:30:37 -0700108 private static final boolean GARBAGE_COLLECT = false;
109 private static final int GC_THRESH = 6;
110
alshabib10580802015-02-18 18:30:33 -0800111 private final int dummyId = 0xffffffff;
112 private final GroupId dummyGroupId = new DefaultGroupId(dummyId);
113
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700114 @Reference(cardinality = ReferenceCardinality.MANDATORY_UNARY)
115 protected ClusterCommunicationService clusterCommunicator;
116
117 @Reference(cardinality = ReferenceCardinality.MANDATORY_UNARY)
118 protected ClusterService clusterService;
119
120 @Reference(cardinality = ReferenceCardinality.MANDATORY_UNARY)
Jonathan Hart6ec029a2015-03-24 17:12:35 -0700121 protected StorageService storageService;
122
123 @Reference(cardinality = ReferenceCardinality.MANDATORY_UNARY)
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700124 protected MastershipService mastershipService;
125
alshabibb0285992016-03-28 23:30:37 -0700126 @Reference(cardinality = ReferenceCardinality.MANDATORY_UNARY)
127 protected ComponentConfigService cfgService;
128
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700129 // Per device group table with (device id + app cookie) as key
Madan Jampani0b847532016-03-03 13:44:15 -0800130 private ConsistentMap<GroupStoreKeyMapKey,
alshabibb0285992016-03-28 23:30:37 -0700131 StoredGroupEntry> groupStoreEntriesByKey = null;
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700132 // Per device group table with (device id + group id) as key
Srikanth Vavilapalli6a9d4e42015-03-30 19:41:56 -0700133 private final ConcurrentMap<DeviceId, ConcurrentMap<GroupId, StoredGroupEntry>>
alshabibb0285992016-03-28 23:30:37 -0700134 groupEntriesById = new ConcurrentHashMap<>();
Madan Jampani0b847532016-03-03 13:44:15 -0800135 private ConsistentMap<GroupStoreKeyMapKey,
alshabibb0285992016-03-28 23:30:37 -0700136 StoredGroupEntry> auditPendingReqQueue = null;
Frank Wange0eb5ce2016-07-01 18:21:25 +0800137 private MapEventListener<GroupStoreKeyMapKey, StoredGroupEntry>
138 mapListener = new GroupStoreKeyMapListener();
alshabib10580802015-02-18 18:30:33 -0800139 private final ConcurrentMap<DeviceId, ConcurrentMap<GroupId, Group>>
140 extraneousGroupEntriesById = new ConcurrentHashMap<>();
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700141 private ExecutorService messageHandlingExecutor;
142 private static final int MESSAGE_HANDLER_THREAD_POOL_SIZE = 1;
Sho SHIMIZU7a4087b2015-09-10 09:23:16 -0700143 private final HashMap<DeviceId, Boolean> deviceAuditStatus = new HashMap<>();
alshabib10580802015-02-18 18:30:33 -0800144
145 private final AtomicInteger groupIdGen = new AtomicInteger();
146
HIGUCHI Yuta180d70f2015-10-01 16:13:56 -0700147 private KryoNamespace clusterMsgSerializer;
148
helenyrwua1c41152016-08-18 16:16:14 -0700149 private static Topic<GroupStoreMessage> groupTopic;
150
alshabibb0285992016-03-28 23:30:37 -0700151 @Property(name = "garbageCollect", boolValue = GARBAGE_COLLECT,
152 label = "Enable group garbage collection")
153 private boolean garbageCollect = GARBAGE_COLLECT;
154
155 @Property(name = "gcThresh", intValue = GC_THRESH,
156 label = "Number of rounds for group garbage collection")
157 private int gcThresh = GC_THRESH;
158
159
alshabib10580802015-02-18 18:30:33 -0800160 @Activate
161 public void activate() {
alshabibb0285992016-03-28 23:30:37 -0700162 cfgService.registerProperties(getClass());
HIGUCHI Yuta3a84b322016-05-18 13:38:07 -0700163 KryoNamespace.Builder kryoBuilder = new KryoNamespace.Builder()
alshabibb0285992016-03-28 23:30:37 -0700164 .register(KryoNamespaces.API)
HIGUCHI Yuta3a84b322016-05-18 13:38:07 -0700165 .nextId(KryoNamespaces.BEGIN_USER_CUSTOM_ID)
alshabibb0285992016-03-28 23:30:37 -0700166 .register(DefaultGroup.class,
167 DefaultGroupBucket.class,
168 DefaultGroupDescription.class,
169 DefaultGroupKey.class,
170 GroupDescription.Type.class,
171 Group.GroupState.class,
172 GroupBuckets.class,
alshabibb0285992016-03-28 23:30:37 -0700173 GroupStoreMessage.class,
174 GroupStoreMessage.Type.class,
175 UpdateType.class,
176 GroupStoreMessageSubjects.class,
177 MultiValuedTimestamp.class,
178 GroupStoreKeyMapKey.class,
179 GroupStoreIdMapKey.class,
180 GroupStoreMapKey.class
181 );
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700182
HIGUCHI Yuta3a84b322016-05-18 13:38:07 -0700183 clusterMsgSerializer = kryoBuilder.build("GroupStore");
184 Serializer serializer = Serializer.using(clusterMsgSerializer);
HIGUCHI Yuta180d70f2015-10-01 16:13:56 -0700185
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700186 messageHandlingExecutor = Executors.
187 newFixedThreadPool(MESSAGE_HANDLER_THREAD_POOL_SIZE,
188 groupedThreads("onos/store/group",
HIGUCHI Yutad9e01052016-04-14 09:31:42 -0700189 "message-handlers",
190 log));
Madan Jampani01e05fb2015-08-13 13:29:36 -0700191
192 clusterCommunicator.addSubscriber(GroupStoreMessageSubjects.REMOTE_GROUP_OP_REQUEST,
alshabibb0285992016-03-28 23:30:37 -0700193 clusterMsgSerializer::deserialize,
194 this::process,
195 messageHandlingExecutor);
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700196
Madan Jampani0b847532016-03-03 13:44:15 -0800197 log.debug("Creating Consistent map onos-group-store-keymap");
Jonathan Hart6ec029a2015-03-24 17:12:35 -0700198
Madan Jampani0b847532016-03-03 13:44:15 -0800199 groupStoreEntriesByKey = storageService.<GroupStoreKeyMapKey, StoredGroupEntry>consistentMapBuilder()
200 .withName("onos-group-store-keymap")
HIGUCHI Yuta3a84b322016-05-18 13:38:07 -0700201 .withSerializer(serializer)
Jonathan Hart6ec029a2015-03-24 17:12:35 -0700202 .build();
Frank Wange0eb5ce2016-07-01 18:21:25 +0800203 groupStoreEntriesByKey.addListener(mapListener);
Srikanth Vavilapalli23181912015-05-04 09:48:09 -0700204 log.debug("Current size of groupstorekeymap:{}",
205 groupStoreEntriesByKey.size());
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700206
Madan Jampani0b847532016-03-03 13:44:15 -0800207 log.debug("Creating Consistent map pendinggroupkeymap");
Jonathan Hart6ec029a2015-03-24 17:12:35 -0700208
Madan Jampani0b847532016-03-03 13:44:15 -0800209 auditPendingReqQueue = storageService.<GroupStoreKeyMapKey, StoredGroupEntry>consistentMapBuilder()
210 .withName("onos-pending-group-keymap")
HIGUCHI Yuta3a84b322016-05-18 13:38:07 -0700211 .withSerializer(serializer)
Jonathan Hart6ec029a2015-03-24 17:12:35 -0700212 .build();
Srikanth Vavilapalli23181912015-05-04 09:48:09 -0700213 log.debug("Current size of pendinggroupkeymap:{}",
214 auditPendingReqQueue.size());
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700215
helenyrwua1c41152016-08-18 16:16:14 -0700216 groupTopic = getOrCreateGroupTopic(serializer);
217 groupTopic.subscribe(this::processGroupMessage);
218
alshabib10580802015-02-18 18:30:33 -0800219 log.info("Started");
220 }
221
222 @Deactivate
223 public void deactivate() {
Frank Wange0eb5ce2016-07-01 18:21:25 +0800224 groupStoreEntriesByKey.removeListener(mapListener);
alshabibb0285992016-03-28 23:30:37 -0700225 cfgService.unregisterProperties(getClass(), false);
HIGUCHI Yuta180d70f2015-10-01 16:13:56 -0700226 clusterCommunicator.removeSubscriber(GroupStoreMessageSubjects.REMOTE_GROUP_OP_REQUEST);
alshabib10580802015-02-18 18:30:33 -0800227 log.info("Stopped");
228 }
229
alshabibb0285992016-03-28 23:30:37 -0700230 @Modified
231 public void modified(ComponentContext context) {
232 Dictionary<?, ?> properties = context != null ? context.getProperties() : new Properties();
233
234 try {
235 String s = get(properties, "garbageCollect");
236 garbageCollect = isNullOrEmpty(s) ? GARBAGE_COLLECT : Boolean.parseBoolean(s.trim());
237
238 s = get(properties, "gcThresh");
239 gcThresh = isNullOrEmpty(s) ? GC_THRESH : Integer.parseInt(s.trim());
240 } catch (Exception e) {
241 gcThresh = GC_THRESH;
242 garbageCollect = GARBAGE_COLLECT;
243 }
244 }
245
helenyrwua1c41152016-08-18 16:16:14 -0700246 private Topic<GroupStoreMessage> getOrCreateGroupTopic(Serializer serializer) {
247 if (groupTopic == null) {
248 return storageService.getTopic("group-failover-notif", serializer);
249 } else {
250 return groupTopic;
251 }
252 };
253
alshabib10580802015-02-18 18:30:33 -0800254 /**
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700255 * Returns the group store eventual consistent key map.
alshabib10580802015-02-18 18:30:33 -0800256 *
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700257 * @return Map representing group key table.
alshabib10580802015-02-18 18:30:33 -0800258 */
Madan Jampani0b847532016-03-03 13:44:15 -0800259 private Map<GroupStoreKeyMapKey, StoredGroupEntry>
alshabibb0285992016-03-28 23:30:37 -0700260 getGroupStoreKeyMap() {
Madan Jampani0b847532016-03-03 13:44:15 -0800261 return groupStoreEntriesByKey.asJavaMap();
alshabib10580802015-02-18 18:30:33 -0800262 }
263
264 /**
Srikanth Vavilapalli6a9d4e42015-03-30 19:41:56 -0700265 * Returns the group id table for specified device.
alshabib10580802015-02-18 18:30:33 -0800266 *
Srikanth Vavilapalli6a9d4e42015-03-30 19:41:56 -0700267 * @param deviceId identifier of the device
268 * @return Map representing group key table of given device.
alshabib10580802015-02-18 18:30:33 -0800269 */
Srikanth Vavilapalli6a9d4e42015-03-30 19:41:56 -0700270 private ConcurrentMap<GroupId, StoredGroupEntry> getGroupIdTable(DeviceId deviceId) {
Yuta HIGUCHIc2e68152016-08-16 13:47:36 -0700271 return groupEntriesById.computeIfAbsent(deviceId, k -> new ConcurrentHashMap<>());
alshabib10580802015-02-18 18:30:33 -0800272 }
273
274 /**
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700275 * Returns the pending group request table.
alshabib10580802015-02-18 18:30:33 -0800276 *
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700277 * @return Map representing group key table.
alshabib10580802015-02-18 18:30:33 -0800278 */
Madan Jampani0b847532016-03-03 13:44:15 -0800279 private Map<GroupStoreKeyMapKey, StoredGroupEntry>
alshabibb0285992016-03-28 23:30:37 -0700280 getPendingGroupKeyTable() {
Madan Jampani0b847532016-03-03 13:44:15 -0800281 return auditPendingReqQueue.asJavaMap();
alshabib10580802015-02-18 18:30:33 -0800282 }
283
284 /**
285 * Returns the extraneous group id table for specified device.
286 *
287 * @param deviceId identifier of the device
288 * @return Map representing group key table of given device.
289 */
290 private ConcurrentMap<GroupId, Group>
291 getExtraneousGroupIdTable(DeviceId deviceId) {
Yuta HIGUCHIc2e68152016-08-16 13:47:36 -0700292 return extraneousGroupEntriesById.computeIfAbsent(deviceId, k -> new ConcurrentHashMap<>());
alshabib10580802015-02-18 18:30:33 -0800293 }
294
295 /**
296 * Returns the number of groups for the specified device in the store.
297 *
298 * @return number of groups for the specified device
299 */
300 @Override
301 public int getGroupCount(DeviceId deviceId) {
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700302 return (getGroups(deviceId) != null) ?
alshabibb0285992016-03-28 23:30:37 -0700303 Iterables.size(getGroups(deviceId)) : 0;
alshabib10580802015-02-18 18:30:33 -0800304 }
305
306 /**
307 * Returns the groups associated with a device.
308 *
309 * @param deviceId the device ID
alshabib10580802015-02-18 18:30:33 -0800310 * @return the group entries
311 */
312 @Override
313 public Iterable<Group> getGroups(DeviceId deviceId) {
Charles Chanf4838a72015-12-07 18:13:45 -0800314 // Let ImmutableSet.copyOf do the type conversion
315 return ImmutableSet.copyOf(getStoredGroups(deviceId));
alshabib10580802015-02-18 18:30:33 -0800316 }
317
Srikanth Vavilapalli23181912015-05-04 09:48:09 -0700318 private Iterable<StoredGroupEntry> getStoredGroups(DeviceId deviceId) {
Charles Chanf4838a72015-12-07 18:13:45 -0800319 NodeId master = mastershipService.getMasterFor(deviceId);
320 if (master == null) {
321 log.debug("Failed to getGroups: No master for {}", deviceId);
322 return Collections.emptySet();
323 }
324
325 Set<StoredGroupEntry> storedGroups = getGroupStoreKeyMap().values()
326 .stream()
327 .filter(input -> input.deviceId().equals(deviceId))
328 .collect(Collectors.toSet());
329 return ImmutableSet.copyOf(storedGroups);
Srikanth Vavilapalli23181912015-05-04 09:48:09 -0700330 }
331
alshabib10580802015-02-18 18:30:33 -0800332 /**
333 * Returns the stored group entry.
334 *
alshabibb0285992016-03-28 23:30:37 -0700335 * @param deviceId the device ID
alshabib10580802015-02-18 18:30:33 -0800336 * @param appCookie the group key
alshabib10580802015-02-18 18:30:33 -0800337 * @return a group associated with the key
338 */
339 @Override
340 public Group getGroup(DeviceId deviceId, GroupKey appCookie) {
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700341 return getStoredGroupEntry(deviceId, appCookie);
342 }
343
344 private StoredGroupEntry getStoredGroupEntry(DeviceId deviceId,
345 GroupKey appCookie) {
346 return getGroupStoreKeyMap().get(new GroupStoreKeyMapKey(deviceId,
347 appCookie));
348 }
349
350 @Override
351 public Group getGroup(DeviceId deviceId, GroupId groupId) {
352 return getStoredGroupEntry(deviceId, groupId);
353 }
354
355 private StoredGroupEntry getStoredGroupEntry(DeviceId deviceId,
356 GroupId groupId) {
Srikanth Vavilapalli6a9d4e42015-03-30 19:41:56 -0700357 return getGroupIdTable(deviceId).get(groupId);
alshabib10580802015-02-18 18:30:33 -0800358 }
359
360 private int getFreeGroupIdValue(DeviceId deviceId) {
361 int freeId = groupIdGen.incrementAndGet();
362
363 while (true) {
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700364 Group existing = getGroup(deviceId, new DefaultGroupId(freeId));
alshabib10580802015-02-18 18:30:33 -0800365 if (existing == null) {
366 existing = (
367 extraneousGroupEntriesById.get(deviceId) != null) ?
368 extraneousGroupEntriesById.get(deviceId).
369 get(new DefaultGroupId(freeId)) :
370 null;
371 }
372 if (existing != null) {
373 freeId = groupIdGen.incrementAndGet();
374 } else {
375 break;
376 }
377 }
Srikanth Vavilapalli23181912015-05-04 09:48:09 -0700378 log.debug("getFreeGroupIdValue: Next Free ID is {}", freeId);
alshabib10580802015-02-18 18:30:33 -0800379 return freeId;
380 }
381
382 /**
383 * Stores a new group entry using the information from group description.
384 *
385 * @param groupDesc group description to be used to create group entry
386 */
387 @Override
388 public void storeGroupDescription(GroupDescription groupDesc) {
Srikanth Vavilapalli23181912015-05-04 09:48:09 -0700389 log.debug("In storeGroupDescription");
alshabib10580802015-02-18 18:30:33 -0800390 // Check if a group is existing with the same key
Saurav Das8a0732e2015-11-20 15:27:53 -0800391 Group existingGroup = getGroup(groupDesc.deviceId(), groupDesc.appCookie());
392 if (existingGroup != null) {
Charles Chan216e3c82016-04-23 14:48:16 -0700393 log.info("Group already exists with the same key {} in dev:{} with id:0x{}",
Saurav Das8a0732e2015-11-20 15:27:53 -0800394 groupDesc.appCookie(), groupDesc.deviceId(),
395 Integer.toHexString(existingGroup.id().id()));
alshabib10580802015-02-18 18:30:33 -0800396 return;
397 }
398
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700399 // Check if group to be created by a remote instance
Madan Jampani175e8fd2015-05-20 14:10:45 -0700400 if (mastershipService.getLocalRole(groupDesc.deviceId()) != MastershipRole.MASTER) {
Srikanth Vavilapalli23181912015-05-04 09:48:09 -0700401 log.debug("storeGroupDescription: Device {} local role is not MASTER",
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700402 groupDesc.deviceId());
Srikanth Vavilapalli23181912015-05-04 09:48:09 -0700403 if (mastershipService.getMasterFor(groupDesc.deviceId()) == null) {
404 log.error("No Master for device {}..."
alshabibb0285992016-03-28 23:30:37 -0700405 + "Can not perform add group operation",
406 groupDesc.deviceId());
Srikanth Vavilapalli23181912015-05-04 09:48:09 -0700407 //TODO: Send Group operation failure event
408 return;
409 }
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700410 GroupStoreMessage groupOp = GroupStoreMessage.
411 createGroupAddRequestMsg(groupDesc.deviceId(),
412 groupDesc);
Madan Jampani2bfa94c2015-04-11 05:03:49 -0700413
Madan Jampani175e8fd2015-05-20 14:10:45 -0700414 clusterCommunicator.unicast(groupOp,
alshabibb0285992016-03-28 23:30:37 -0700415 GroupStoreMessageSubjects.REMOTE_GROUP_OP_REQUEST,
416 clusterMsgSerializer::serialize,
417 mastershipService.getMasterFor(groupDesc.deviceId()))
418 .whenComplete((result, error) -> {
Madan Jampani175e8fd2015-05-20 14:10:45 -0700419 if (error != null) {
420 log.warn("Failed to send request to master: {} to {}",
alshabibb0285992016-03-28 23:30:37 -0700421 groupOp,
422 mastershipService.getMasterFor(groupDesc.deviceId()));
Madan Jampani175e8fd2015-05-20 14:10:45 -0700423 //TODO: Send Group operation failure event
424 } else {
425 log.debug("Sent Group operation request for device {} "
alshabibb0285992016-03-28 23:30:37 -0700426 + "to remote MASTER {}",
427 groupDesc.deviceId(),
428 mastershipService.getMasterFor(groupDesc.deviceId()));
Madan Jampani175e8fd2015-05-20 14:10:45 -0700429 }
430 });
alshabib10580802015-02-18 18:30:33 -0800431 return;
432 }
433
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700434 log.debug("Store group for device {} is getting handled locally",
435 groupDesc.deviceId());
alshabib10580802015-02-18 18:30:33 -0800436 storeGroupDescriptionInternal(groupDesc);
437 }
438
Srikanth Vavilapallie48b3cf2015-07-06 11:43:07 -0700439 private Group getMatchingExtraneousGroupbyId(DeviceId deviceId, Integer groupId) {
440 ConcurrentMap<GroupId, Group> extraneousMap =
441 extraneousGroupEntriesById.get(deviceId);
442 if (extraneousMap == null) {
443 return null;
444 }
445 return extraneousMap.get(new DefaultGroupId(groupId));
446 }
447
448 private Group getMatchingExtraneousGroupbyBuckets(DeviceId deviceId,
449 GroupBuckets buckets) {
450 ConcurrentMap<GroupId, Group> extraneousMap =
451 extraneousGroupEntriesById.get(deviceId);
452 if (extraneousMap == null) {
453 return null;
454 }
455
alshabibb0285992016-03-28 23:30:37 -0700456 for (Group extraneousGroup : extraneousMap.values()) {
Srikanth Vavilapallie48b3cf2015-07-06 11:43:07 -0700457 if (extraneousGroup.buckets().equals(buckets)) {
458 return extraneousGroup;
459 }
460 }
461 return null;
462 }
463
alshabib10580802015-02-18 18:30:33 -0800464 private void storeGroupDescriptionInternal(GroupDescription groupDesc) {
465 // Check if a group is existing with the same key
466 if (getGroup(groupDesc.deviceId(), groupDesc.appCookie()) != null) {
467 return;
468 }
469
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700470 if (deviceAuditStatus.get(groupDesc.deviceId()) == null) {
471 // Device group audit has not completed yet
472 // Add this group description to pending group key table
473 // Create a group entry object with Dummy Group ID
Srikanth Vavilapalli23181912015-05-04 09:48:09 -0700474 log.debug("storeGroupDescriptionInternal: Device {} AUDIT pending...Queuing Group ADD request",
alshabibb0285992016-03-28 23:30:37 -0700475 groupDesc.deviceId());
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700476 StoredGroupEntry group = new DefaultGroup(dummyGroupId, groupDesc);
477 group.setState(GroupState.WAITING_AUDIT_COMPLETE);
Madan Jampani0b847532016-03-03 13:44:15 -0800478 Map<GroupStoreKeyMapKey, StoredGroupEntry> pendingKeyTable =
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700479 getPendingGroupKeyTable();
480 pendingKeyTable.put(new GroupStoreKeyMapKey(groupDesc.deviceId(),
481 groupDesc.appCookie()),
482 group);
483 return;
484 }
485
Srikanth Vavilapallie48b3cf2015-07-06 11:43:07 -0700486 Group matchingExtraneousGroup = null;
487 if (groupDesc.givenGroupId() != null) {
488 //Check if there is a extraneous group existing with the same Id
489 matchingExtraneousGroup = getMatchingExtraneousGroupbyId(
alshabibb0285992016-03-28 23:30:37 -0700490 groupDesc.deviceId(), groupDesc.givenGroupId());
Srikanth Vavilapallie48b3cf2015-07-06 11:43:07 -0700491 if (matchingExtraneousGroup != null) {
Saurav Das0fd79d92016-03-07 10:58:36 -0800492 log.debug("storeGroupDescriptionInternal: Matching extraneous group "
alshabibb0285992016-03-28 23:30:37 -0700493 + "found in Device {} for group id 0x{}",
Srikanth Vavilapallie48b3cf2015-07-06 11:43:07 -0700494 groupDesc.deviceId(),
Saurav Das0fd79d92016-03-07 10:58:36 -0800495 Integer.toHexString(groupDesc.givenGroupId()));
Srikanth Vavilapallie48b3cf2015-07-06 11:43:07 -0700496 //Check if the group buckets matches with user provided buckets
497 if (matchingExtraneousGroup.buckets().equals(groupDesc.buckets())) {
498 //Group is already existing with the same buckets and Id
499 // Create a group entry object
Saurav Das0fd79d92016-03-07 10:58:36 -0800500 log.debug("storeGroupDescriptionInternal: Buckets also matching "
alshabibb0285992016-03-28 23:30:37 -0700501 + "in Device {} for group id 0x{}",
Srikanth Vavilapallie48b3cf2015-07-06 11:43:07 -0700502 groupDesc.deviceId(),
Saurav Das0fd79d92016-03-07 10:58:36 -0800503 Integer.toHexString(groupDesc.givenGroupId()));
Srikanth Vavilapallie48b3cf2015-07-06 11:43:07 -0700504 StoredGroupEntry group = new DefaultGroup(
alshabibb0285992016-03-28 23:30:37 -0700505 matchingExtraneousGroup.id(), groupDesc);
Srikanth Vavilapallie48b3cf2015-07-06 11:43:07 -0700506 // Insert the newly created group entry into key and id maps
507 getGroupStoreKeyMap().
alshabibb0285992016-03-28 23:30:37 -0700508 put(new GroupStoreKeyMapKey(groupDesc.deviceId(),
509 groupDesc.appCookie()), group);
Srikanth Vavilapallie48b3cf2015-07-06 11:43:07 -0700510 // Ensure it also inserted into group id based table to
511 // avoid any chances of duplication in group id generation
512 getGroupIdTable(groupDesc.deviceId()).
alshabibb0285992016-03-28 23:30:37 -0700513 put(matchingExtraneousGroup.id(), group);
Srikanth Vavilapallie48b3cf2015-07-06 11:43:07 -0700514 addOrUpdateGroupEntry(matchingExtraneousGroup);
515 removeExtraneousGroupEntry(matchingExtraneousGroup);
516 return;
517 } else {
518 //Group buckets are not matching. Update group
519 //with user provided buckets.
Saurav Das0fd79d92016-03-07 10:58:36 -0800520 log.debug("storeGroupDescriptionInternal: Buckets are not "
alshabibb0285992016-03-28 23:30:37 -0700521 + "matching in Device {} for group id 0x{}",
Srikanth Vavilapallie48b3cf2015-07-06 11:43:07 -0700522 groupDesc.deviceId(),
Saurav Das0fd79d92016-03-07 10:58:36 -0800523 Integer.toHexString(groupDesc.givenGroupId()));
524 StoredGroupEntry modifiedGroup = new DefaultGroup(
alshabibb0285992016-03-28 23:30:37 -0700525 matchingExtraneousGroup.id(), groupDesc);
Saurav Das0fd79d92016-03-07 10:58:36 -0800526 modifiedGroup.setState(GroupState.PENDING_UPDATE);
527 getGroupStoreKeyMap().
alshabibb0285992016-03-28 23:30:37 -0700528 put(new GroupStoreKeyMapKey(groupDesc.deviceId(),
529 groupDesc.appCookie()), modifiedGroup);
Saurav Das0fd79d92016-03-07 10:58:36 -0800530 // Ensure it also inserted into group id based table to
531 // avoid any chances of duplication in group id generation
532 getGroupIdTable(groupDesc.deviceId()).
alshabibb0285992016-03-28 23:30:37 -0700533 put(matchingExtraneousGroup.id(), modifiedGroup);
Saurav Das0fd79d92016-03-07 10:58:36 -0800534 removeExtraneousGroupEntry(matchingExtraneousGroup);
535 log.debug("storeGroupDescriptionInternal: Triggering Group "
alshabibb0285992016-03-28 23:30:37 -0700536 + "UPDATE request for {} in device {}",
Saurav Das0fd79d92016-03-07 10:58:36 -0800537 matchingExtraneousGroup.id(),
538 groupDesc.deviceId());
539 notifyDelegate(new GroupEvent(Type.GROUP_UPDATE_REQUESTED, modifiedGroup));
540 return;
Srikanth Vavilapallie48b3cf2015-07-06 11:43:07 -0700541 }
542 }
543 } else {
544 //Check if there is an extraneous group with user provided buckets
545 matchingExtraneousGroup = getMatchingExtraneousGroupbyBuckets(
alshabibb0285992016-03-28 23:30:37 -0700546 groupDesc.deviceId(), groupDesc.buckets());
Srikanth Vavilapallie48b3cf2015-07-06 11:43:07 -0700547 if (matchingExtraneousGroup != null) {
548 //Group is already existing with the same buckets.
549 //So reuse this group.
550 log.debug("storeGroupDescriptionInternal: Matching extraneous group found in Device {}",
551 groupDesc.deviceId());
552 //Create a group entry object
553 StoredGroupEntry group = new DefaultGroup(
alshabibb0285992016-03-28 23:30:37 -0700554 matchingExtraneousGroup.id(), groupDesc);
Srikanth Vavilapallie48b3cf2015-07-06 11:43:07 -0700555 // Insert the newly created group entry into key and id maps
556 getGroupStoreKeyMap().
alshabibb0285992016-03-28 23:30:37 -0700557 put(new GroupStoreKeyMapKey(groupDesc.deviceId(),
558 groupDesc.appCookie()), group);
Srikanth Vavilapallie48b3cf2015-07-06 11:43:07 -0700559 // Ensure it also inserted into group id based table to
560 // avoid any chances of duplication in group id generation
561 getGroupIdTable(groupDesc.deviceId()).
alshabibb0285992016-03-28 23:30:37 -0700562 put(matchingExtraneousGroup.id(), group);
Srikanth Vavilapallie48b3cf2015-07-06 11:43:07 -0700563 addOrUpdateGroupEntry(matchingExtraneousGroup);
564 removeExtraneousGroupEntry(matchingExtraneousGroup);
565 return;
566 } else {
567 //TODO: Check if there are any empty groups that can be used here
568 log.debug("storeGroupDescriptionInternal: No matching extraneous groups found in Device {}",
569 groupDesc.deviceId());
570 }
571 }
572
Saurav Das100e3b82015-04-30 11:12:10 -0700573 GroupId id = null;
574 if (groupDesc.givenGroupId() == null) {
575 // Get a new group identifier
576 id = new DefaultGroupId(getFreeGroupIdValue(groupDesc.deviceId()));
577 } else {
Saurav Das8be4e3a2016-03-11 17:19:07 -0800578 // we need to use the identifier passed in by caller, but check if
579 // already used
580 Group existing = getGroup(groupDesc.deviceId(),
581 new DefaultGroupId(groupDesc.givenGroupId()));
582 if (existing != null) {
583 log.warn("Group already exists with the same id: 0x{} in dev:{} "
alshabibb0285992016-03-28 23:30:37 -0700584 + "but with different key: {} (request gkey: {})",
585 Integer.toHexString(groupDesc.givenGroupId()),
586 groupDesc.deviceId(),
587 existing.appCookie(),
588 groupDesc.appCookie());
Saurav Das8be4e3a2016-03-11 17:19:07 -0800589 return;
590 }
Saurav Das100e3b82015-04-30 11:12:10 -0700591 id = new DefaultGroupId(groupDesc.givenGroupId());
592 }
alshabib10580802015-02-18 18:30:33 -0800593 // Create a group entry object
594 StoredGroupEntry group = new DefaultGroup(id, groupDesc);
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700595 // Insert the newly created group entry into key and id maps
596 getGroupStoreKeyMap().
alshabibb0285992016-03-28 23:30:37 -0700597 put(new GroupStoreKeyMapKey(groupDesc.deviceId(),
598 groupDesc.appCookie()), group);
Srikanth Vavilapalli6a9d4e42015-03-30 19:41:56 -0700599 // Ensure it also inserted into group id based table to
600 // avoid any chances of duplication in group id generation
601 getGroupIdTable(groupDesc.deviceId()).
alshabibb0285992016-03-28 23:30:37 -0700602 put(id, group);
Srikanth Vavilapalli23181912015-05-04 09:48:09 -0700603 log.debug("storeGroupDescriptionInternal: Processing Group ADD request for Id {} in device {}",
alshabibb0285992016-03-28 23:30:37 -0700604 id,
605 groupDesc.deviceId());
alshabib10580802015-02-18 18:30:33 -0800606 notifyDelegate(new GroupEvent(GroupEvent.Type.GROUP_ADD_REQUESTED,
607 group));
608 }
609
610 /**
611 * Updates the existing group entry with the information
612 * from group description.
613 *
alshabibb0285992016-03-28 23:30:37 -0700614 * @param deviceId the device ID
alshabib10580802015-02-18 18:30:33 -0800615 * @param oldAppCookie the current group key
alshabibb0285992016-03-28 23:30:37 -0700616 * @param type update type
617 * @param newBuckets group buckets for updates
alshabib10580802015-02-18 18:30:33 -0800618 * @param newAppCookie optional new group key
619 */
620 @Override
621 public void updateGroupDescription(DeviceId deviceId,
622 GroupKey oldAppCookie,
623 UpdateType type,
624 GroupBuckets newBuckets,
625 GroupKey newAppCookie) {
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700626 // Check if group update to be done by a remote instance
sangho52abe3a2015-05-05 14:13:34 -0700627 if (mastershipService.getMasterFor(deviceId) != null &&
628 mastershipService.getLocalRole(deviceId) != MastershipRole.MASTER) {
Srikanth Vavilapalli23181912015-05-04 09:48:09 -0700629 log.debug("updateGroupDescription: Device {} local role is not MASTER",
630 deviceId);
631 if (mastershipService.getMasterFor(deviceId) == null) {
632 log.error("No Master for device {}..."
alshabibb0285992016-03-28 23:30:37 -0700633 + "Can not perform update group operation",
634 deviceId);
Srikanth Vavilapalli23181912015-05-04 09:48:09 -0700635 //TODO: Send Group operation failure event
636 return;
637 }
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700638 GroupStoreMessage groupOp = GroupStoreMessage.
639 createGroupUpdateRequestMsg(deviceId,
640 oldAppCookie,
641 type,
642 newBuckets,
643 newAppCookie);
Madan Jampani2bfa94c2015-04-11 05:03:49 -0700644
Madan Jampani175e8fd2015-05-20 14:10:45 -0700645 clusterCommunicator.unicast(groupOp,
alshabibb0285992016-03-28 23:30:37 -0700646 GroupStoreMessageSubjects.REMOTE_GROUP_OP_REQUEST,
647 clusterMsgSerializer::serialize,
648 mastershipService.getMasterFor(deviceId)).whenComplete((result, error) -> {
649 if (error != null) {
650 log.warn("Failed to send request to master: {} to {}",
651 groupOp,
652 mastershipService.getMasterFor(deviceId), error);
653 }
654 //TODO: Send Group operation failure event
655 });
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700656 return;
657 }
Srikanth Vavilapalli23181912015-05-04 09:48:09 -0700658 log.debug("updateGroupDescription for device {} is getting handled locally",
659 deviceId);
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700660 updateGroupDescriptionInternal(deviceId,
661 oldAppCookie,
662 type,
663 newBuckets,
664 newAppCookie);
665 }
666
667 private void updateGroupDescriptionInternal(DeviceId deviceId,
alshabibb0285992016-03-28 23:30:37 -0700668 GroupKey oldAppCookie,
669 UpdateType type,
670 GroupBuckets newBuckets,
671 GroupKey newAppCookie) {
alshabib10580802015-02-18 18:30:33 -0800672 // Check if a group is existing with the provided key
673 Group oldGroup = getGroup(deviceId, oldAppCookie);
674 if (oldGroup == null) {
Saurav Das8be4e3a2016-03-11 17:19:07 -0800675 log.warn("updateGroupDescriptionInternal: Group not found...strange. "
alshabibb0285992016-03-28 23:30:37 -0700676 + "GroupKey:{} DeviceId:{}", oldAppCookie, deviceId);
alshabib10580802015-02-18 18:30:33 -0800677 return;
678 }
679
680 List<GroupBucket> newBucketList = getUpdatedBucketList(oldGroup,
681 type,
682 newBuckets);
683 if (newBucketList != null) {
684 // Create a new group object from the old group
685 GroupBuckets updatedBuckets = new GroupBuckets(newBucketList);
686 GroupKey newCookie = (newAppCookie != null) ? newAppCookie : oldAppCookie;
687 GroupDescription updatedGroupDesc = new DefaultGroupDescription(
688 oldGroup.deviceId(),
689 oldGroup.type(),
690 updatedBuckets,
691 newCookie,
Saurav Das100e3b82015-04-30 11:12:10 -0700692 oldGroup.givenGroupId(),
alshabib10580802015-02-18 18:30:33 -0800693 oldGroup.appId());
694 StoredGroupEntry newGroup = new DefaultGroup(oldGroup.id(),
695 updatedGroupDesc);
Srikanth Vavilapalli23181912015-05-04 09:48:09 -0700696 log.debug("updateGroupDescriptionInternal: group entry {} in device {} moving from {} to PENDING_UPDATE",
alshabibb0285992016-03-28 23:30:37 -0700697 oldGroup.id(),
698 oldGroup.deviceId(),
699 oldGroup.state());
alshabib10580802015-02-18 18:30:33 -0800700 newGroup.setState(GroupState.PENDING_UPDATE);
701 newGroup.setLife(oldGroup.life());
702 newGroup.setPackets(oldGroup.packets());
703 newGroup.setBytes(oldGroup.bytes());
Srikanth Vavilapalli6a9d4e42015-03-30 19:41:56 -0700704 //Update the group entry in groupkey based map.
705 //Update to groupid based map will happen in the
706 //groupkey based map update listener
Srikanth Vavilapalli23181912015-05-04 09:48:09 -0700707 log.debug("updateGroupDescriptionInternal with type {}: Group updated with buckets",
708 type);
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700709 getGroupStoreKeyMap().
alshabibb0285992016-03-28 23:30:37 -0700710 put(new GroupStoreKeyMapKey(newGroup.deviceId(),
711 newGroup.appCookie()), newGroup);
alshabib10580802015-02-18 18:30:33 -0800712 notifyDelegate(new GroupEvent(Type.GROUP_UPDATE_REQUESTED, newGroup));
Srikanth Vavilapalli23181912015-05-04 09:48:09 -0700713 } else {
714 log.warn("updateGroupDescriptionInternal with type {}: No "
alshabibb0285992016-03-28 23:30:37 -0700715 + "change in the buckets in update", type);
alshabib10580802015-02-18 18:30:33 -0800716 }
717 }
718
719 private List<GroupBucket> getUpdatedBucketList(Group oldGroup,
720 UpdateType type,
721 GroupBuckets buckets) {
Victor Silvadf1eeae2016-08-12 15:28:57 -0300722 List<GroupBucket> oldBuckets = oldGroup.buckets().buckets();
723 List<GroupBucket> updatedBucketList = new ArrayList<>();
alshabib10580802015-02-18 18:30:33 -0800724 boolean groupDescUpdated = false;
725
726 if (type == UpdateType.ADD) {
Victor Silvadf1eeae2016-08-12 15:28:57 -0300727 List<GroupBucket> newBuckets = buckets.buckets();
728
729 // Add old buckets that will not be updated and check if any will be updated.
730 for (GroupBucket oldBucket : oldBuckets) {
731 int newBucketIndex = newBuckets.indexOf(oldBucket);
732
733 if (newBucketIndex != -1) {
734 GroupBucket newBucket = newBuckets.get(newBucketIndex);
735 if (!newBucket.hasSameParameters(oldBucket)) {
736 // Bucket will be updated
737 groupDescUpdated = true;
738 }
739 } else {
740 // Old bucket will remain the same - add it.
741 updatedBucketList.add(oldBucket);
alshabib10580802015-02-18 18:30:33 -0800742 }
743 }
Victor Silvadf1eeae2016-08-12 15:28:57 -0300744
745 // Add all new buckets
746 updatedBucketList.addAll(newBuckets);
747 if (!oldBuckets.containsAll(newBuckets)) {
748 groupDescUpdated = true;
749 }
750
alshabib10580802015-02-18 18:30:33 -0800751 } else if (type == UpdateType.REMOVE) {
Victor Silvadf1eeae2016-08-12 15:28:57 -0300752 List<GroupBucket> bucketsToRemove = buckets.buckets();
753
754 // Check which old buckets should remain
755 for (GroupBucket oldBucket : oldBuckets) {
756 if (!bucketsToRemove.contains(oldBucket)) {
757 updatedBucketList.add(oldBucket);
758 } else {
alshabib10580802015-02-18 18:30:33 -0800759 groupDescUpdated = true;
760 }
761 }
762 }
763
764 if (groupDescUpdated) {
Victor Silvadf1eeae2016-08-12 15:28:57 -0300765 return updatedBucketList;
alshabib10580802015-02-18 18:30:33 -0800766 } else {
767 return null;
768 }
769 }
770
771 /**
772 * Triggers deleting the existing group entry.
773 *
alshabibb0285992016-03-28 23:30:37 -0700774 * @param deviceId the device ID
alshabib10580802015-02-18 18:30:33 -0800775 * @param appCookie the group key
776 */
777 @Override
778 public void deleteGroupDescription(DeviceId deviceId,
779 GroupKey appCookie) {
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700780 // Check if group to be deleted by a remote instance
781 if (mastershipService.
782 getLocalRole(deviceId) != MastershipRole.MASTER) {
Srikanth Vavilapalli23181912015-05-04 09:48:09 -0700783 log.debug("deleteGroupDescription: Device {} local role is not MASTER",
784 deviceId);
785 if (mastershipService.getMasterFor(deviceId) == null) {
786 log.error("No Master for device {}..."
alshabibb0285992016-03-28 23:30:37 -0700787 + "Can not perform delete group operation",
788 deviceId);
Srikanth Vavilapalli23181912015-05-04 09:48:09 -0700789 //TODO: Send Group operation failure event
790 return;
791 }
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700792 GroupStoreMessage groupOp = GroupStoreMessage.
793 createGroupDeleteRequestMsg(deviceId,
794 appCookie);
Madan Jampani2bfa94c2015-04-11 05:03:49 -0700795
Madan Jampani175e8fd2015-05-20 14:10:45 -0700796 clusterCommunicator.unicast(groupOp,
alshabibb0285992016-03-28 23:30:37 -0700797 GroupStoreMessageSubjects.REMOTE_GROUP_OP_REQUEST,
798 clusterMsgSerializer::serialize,
799 mastershipService.getMasterFor(deviceId)).whenComplete((result, error) -> {
800 if (error != null) {
801 log.warn("Failed to send request to master: {} to {}",
802 groupOp,
803 mastershipService.getMasterFor(deviceId), error);
804 }
805 //TODO: Send Group operation failure event
806 });
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700807 return;
808 }
Srikanth Vavilapalli23181912015-05-04 09:48:09 -0700809 log.debug("deleteGroupDescription in device {} is getting handled locally",
810 deviceId);
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700811 deleteGroupDescriptionInternal(deviceId, appCookie);
812 }
813
814 private void deleteGroupDescriptionInternal(DeviceId deviceId,
815 GroupKey appCookie) {
alshabib10580802015-02-18 18:30:33 -0800816 // Check if a group is existing with the provided key
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700817 StoredGroupEntry existing = getStoredGroupEntry(deviceId, appCookie);
alshabib10580802015-02-18 18:30:33 -0800818 if (existing == null) {
819 return;
820 }
821
Srikanth Vavilapalli23181912015-05-04 09:48:09 -0700822 log.debug("deleteGroupDescriptionInternal: group entry {} in device {} moving from {} to PENDING_DELETE",
alshabibb0285992016-03-28 23:30:37 -0700823 existing.id(),
824 existing.deviceId(),
825 existing.state());
alshabib10580802015-02-18 18:30:33 -0800826 synchronized (existing) {
827 existing.setState(GroupState.PENDING_DELETE);
Saurav Das80980c72016-03-23 11:22:49 -0700828 getGroupStoreKeyMap().
alshabibb0285992016-03-28 23:30:37 -0700829 put(new GroupStoreKeyMapKey(existing.deviceId(), existing.appCookie()),
830 existing);
alshabib10580802015-02-18 18:30:33 -0800831 }
Srikanth Vavilapalli23181912015-05-04 09:48:09 -0700832 log.debug("deleteGroupDescriptionInternal: in device {} issuing GROUP_REMOVE_REQUESTED",
833 deviceId);
alshabib10580802015-02-18 18:30:33 -0800834 notifyDelegate(new GroupEvent(Type.GROUP_REMOVE_REQUESTED, existing));
835 }
836
837 /**
838 * Stores a new group entry, or updates an existing entry.
839 *
840 * @param group group entry
841 */
842 @Override
843 public void addOrUpdateGroupEntry(Group group) {
844 // check if this new entry is an update to an existing entry
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700845 StoredGroupEntry existing = getStoredGroupEntry(group.deviceId(),
846 group.id());
alshabib10580802015-02-18 18:30:33 -0800847 GroupEvent event = null;
848
849 if (existing != null) {
Saurav Das0fd79d92016-03-07 10:58:36 -0800850 log.trace("addOrUpdateGroupEntry: updating group entry {} in device {}",
alshabibb0285992016-03-28 23:30:37 -0700851 group.id(),
852 group.deviceId());
alshabib10580802015-02-18 18:30:33 -0800853 synchronized (existing) {
alshabibb0285992016-03-28 23:30:37 -0700854 for (GroupBucket bucket : group.buckets().buckets()) {
Sho SHIMIZU30d639b2015-05-05 09:30:35 -0700855 Optional<GroupBucket> matchingBucket =
Srikanth Vavilapalli10e75cd2015-04-13 16:21:24 -0700856 existing.buckets().buckets()
alshabibb0285992016-03-28 23:30:37 -0700857 .stream()
858 .filter((existingBucket) -> (existingBucket.equals(bucket)))
859 .findFirst();
Srikanth Vavilapalli10e75cd2015-04-13 16:21:24 -0700860 if (matchingBucket.isPresent()) {
861 ((StoredGroupBucketEntry) matchingBucket.
862 get()).setPackets(bucket.packets());
863 ((StoredGroupBucketEntry) matchingBucket.
864 get()).setBytes(bucket.bytes());
865 } else {
866 log.warn("addOrUpdateGroupEntry: No matching "
alshabibb0285992016-03-28 23:30:37 -0700867 + "buckets to update stats");
Srikanth Vavilapalli10e75cd2015-04-13 16:21:24 -0700868 }
869 }
alshabib10580802015-02-18 18:30:33 -0800870 existing.setLife(group.life());
871 existing.setPackets(group.packets());
872 existing.setBytes(group.bytes());
alshabibb0285992016-03-28 23:30:37 -0700873 existing.setReferenceCount(group.referenceCount());
Srikanth Vavilapalli5428b6c2015-05-14 20:22:47 -0700874 if ((existing.state() == GroupState.PENDING_ADD) ||
alshabibb0285992016-03-28 23:30:37 -0700875 (existing.state() == GroupState.PENDING_ADD_RETRY)) {
Saurav Das0fd79d92016-03-07 10:58:36 -0800876 log.trace("addOrUpdateGroupEntry: group entry {} in device {} moving from {} to ADDED",
alshabibb0285992016-03-28 23:30:37 -0700877 existing.id(),
878 existing.deviceId(),
879 existing.state());
alshabib10580802015-02-18 18:30:33 -0800880 existing.setState(GroupState.ADDED);
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700881 existing.setIsGroupStateAddedFirstTime(true);
alshabib10580802015-02-18 18:30:33 -0800882 event = new GroupEvent(Type.GROUP_ADDED, existing);
883 } else {
Saurav Das0fd79d92016-03-07 10:58:36 -0800884 log.trace("addOrUpdateGroupEntry: group entry {} in device {} moving from {} to ADDED",
alshabibb0285992016-03-28 23:30:37 -0700885 existing.id(),
886 existing.deviceId(),
887 GroupState.PENDING_UPDATE);
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700888 existing.setState(GroupState.ADDED);
889 existing.setIsGroupStateAddedFirstTime(false);
alshabib10580802015-02-18 18:30:33 -0800890 event = new GroupEvent(Type.GROUP_UPDATED, existing);
891 }
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700892 //Re-PUT map entries to trigger map update events
893 getGroupStoreKeyMap().
alshabibb0285992016-03-28 23:30:37 -0700894 put(new GroupStoreKeyMapKey(existing.deviceId(),
895 existing.appCookie()), existing);
alshabib10580802015-02-18 18:30:33 -0800896 }
Srikanth Vavilapalli6a9d4e42015-03-30 19:41:56 -0700897 } else {
898 log.warn("addOrUpdateGroupEntry: Group update "
alshabibb0285992016-03-28 23:30:37 -0700899 + "happening for a non-existing entry in the map");
alshabib10580802015-02-18 18:30:33 -0800900 }
901
902 if (event != null) {
903 notifyDelegate(event);
904 }
905 }
906
907 /**
908 * Removes the group entry from store.
909 *
910 * @param group group entry
911 */
912 @Override
913 public void removeGroupEntry(Group group) {
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700914 StoredGroupEntry existing = getStoredGroupEntry(group.deviceId(),
915 group.id());
alshabib10580802015-02-18 18:30:33 -0800916
917 if (existing != null) {
Srikanth Vavilapalli23181912015-05-04 09:48:09 -0700918 log.debug("removeGroupEntry: removing group entry {} in device {}",
alshabibb0285992016-03-28 23:30:37 -0700919 group.id(),
920 group.deviceId());
Srikanth Vavilapalli6a9d4e42015-03-30 19:41:56 -0700921 //Removal from groupid based map will happen in the
922 //map update listener
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700923 getGroupStoreKeyMap().remove(new GroupStoreKeyMapKey(existing.deviceId(),
924 existing.appCookie()));
alshabib10580802015-02-18 18:30:33 -0800925 notifyDelegate(new GroupEvent(Type.GROUP_REMOVED, existing));
Srikanth Vavilapalli23181912015-05-04 09:48:09 -0700926 } else {
927 log.warn("removeGroupEntry for {} in device{} is "
alshabibb0285992016-03-28 23:30:37 -0700928 + "not existing in our maps",
929 group.id(),
930 group.deviceId());
alshabib10580802015-02-18 18:30:33 -0800931 }
932 }
933
934 @Override
Charles Chan0c7c43b2016-01-14 17:39:20 -0800935 public void purgeGroupEntry(DeviceId deviceId) {
936 Set<Entry<GroupStoreKeyMapKey, StoredGroupEntry>> entryPendingRemove =
937 new HashSet<>();
938
Madan Jampani0b847532016-03-03 13:44:15 -0800939 getGroupStoreKeyMap().entrySet().stream()
Charles Chan0c7c43b2016-01-14 17:39:20 -0800940 .filter(entry -> entry.getKey().deviceId().equals(deviceId))
941 .forEach(entryPendingRemove::add);
942
943 entryPendingRemove.forEach(entry -> {
944 groupStoreEntriesByKey.remove(entry.getKey());
945 notifyDelegate(new GroupEvent(Type.GROUP_REMOVED, entry.getValue()));
946 });
947 }
948
949 @Override
alshabib10580802015-02-18 18:30:33 -0800950 public void deviceInitialAuditCompleted(DeviceId deviceId,
951 boolean completed) {
952 synchronized (deviceAuditStatus) {
953 if (completed) {
Srikanth Vavilapalli23181912015-05-04 09:48:09 -0700954 log.debug("AUDIT completed for device {}",
955 deviceId);
alshabib10580802015-02-18 18:30:33 -0800956 deviceAuditStatus.put(deviceId, true);
957 // Execute all pending group requests
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700958 List<StoredGroupEntry> pendingGroupRequests =
959 getPendingGroupKeyTable().values()
alshabibb0285992016-03-28 23:30:37 -0700960 .stream()
961 .filter(g -> g.deviceId().equals(deviceId))
962 .collect(Collectors.toList());
Srikanth Vavilapalli23181912015-05-04 09:48:09 -0700963 log.debug("processing pending group add requests for device {} and number of pending requests {}",
alshabibb0285992016-03-28 23:30:37 -0700964 deviceId,
965 pendingGroupRequests.size());
966 for (Group group : pendingGroupRequests) {
alshabib10580802015-02-18 18:30:33 -0800967 GroupDescription tmp = new DefaultGroupDescription(
968 group.deviceId(),
969 group.type(),
970 group.buckets(),
971 group.appCookie(),
Saurav Das100e3b82015-04-30 11:12:10 -0700972 group.givenGroupId(),
alshabib10580802015-02-18 18:30:33 -0800973 group.appId());
974 storeGroupDescriptionInternal(tmp);
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700975 getPendingGroupKeyTable().
alshabibb0285992016-03-28 23:30:37 -0700976 remove(new GroupStoreKeyMapKey(deviceId, group.appCookie()));
alshabib10580802015-02-18 18:30:33 -0800977 }
alshabib10580802015-02-18 18:30:33 -0800978 } else {
Thomas Vachuskac40d4632015-04-09 16:55:03 -0700979 Boolean audited = deviceAuditStatus.get(deviceId);
980 if (audited != null && audited) {
Srikanth Vavilapalli23181912015-05-04 09:48:09 -0700981 log.debug("Clearing AUDIT status for device {}", deviceId);
alshabib10580802015-02-18 18:30:33 -0800982 deviceAuditStatus.put(deviceId, false);
983 }
984 }
985 }
986 }
987
988 @Override
989 public boolean deviceInitialAuditStatus(DeviceId deviceId) {
990 synchronized (deviceAuditStatus) {
Thomas Vachuskac40d4632015-04-09 16:55:03 -0700991 Boolean audited = deviceAuditStatus.get(deviceId);
992 return audited != null && audited;
alshabib10580802015-02-18 18:30:33 -0800993 }
994 }
995
996 @Override
997 public void groupOperationFailed(DeviceId deviceId, GroupOperation operation) {
998
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -0700999 StoredGroupEntry existing = getStoredGroupEntry(deviceId,
1000 operation.groupId());
alshabib10580802015-02-18 18:30:33 -08001001
1002 if (existing == null) {
1003 log.warn("No group entry with ID {} found ", operation.groupId());
1004 return;
1005 }
1006
Srikanth Vavilapalli23181912015-05-04 09:48:09 -07001007 log.warn("groupOperationFailed: group operation {} failed"
alshabibb0285992016-03-28 23:30:37 -07001008 + "for group {} in device {} with code {}",
1009 operation.opType(),
1010 existing.id(),
1011 existing.deviceId(),
1012 operation.failureCode());
Saurav Das0fd79d92016-03-07 10:58:36 -08001013 if (operation.failureCode() == GroupOperation.GroupMsgErrorCode.GROUP_EXISTS) {
1014 log.warn("Current extraneous groups in device:{} are: {}",
1015 deviceId,
1016 getExtraneousGroups(deviceId));
Saurav Das8be4e3a2016-03-11 17:19:07 -08001017 if (operation.buckets().equals(existing.buckets())) {
1018 if (existing.state() == GroupState.PENDING_ADD) {
1019 log.info("GROUP_EXISTS: GroupID and Buckets match for group in pending "
alshabibb0285992016-03-28 23:30:37 -07001020 + "add state - moving to ADDED for group {} in device {}",
1021 existing.id(), deviceId);
Saurav Das8be4e3a2016-03-11 17:19:07 -08001022 addOrUpdateGroupEntry(existing);
1023 return;
1024 } else {
1025 log.warn("GROUP EXISTS: Group ID matched but buckets did not. "
alshabibb0285992016-03-28 23:30:37 -07001026 + "Operation: {} Existing: {}", operation.buckets(),
1027 existing.buckets());
Saurav Das8be4e3a2016-03-11 17:19:07 -08001028 }
1029 }
Saurav Das0fd79d92016-03-07 10:58:36 -08001030 }
alshabib10580802015-02-18 18:30:33 -08001031 switch (operation.opType()) {
1032 case ADD:
Srikanth Vavilapalli5428b6c2015-05-14 20:22:47 -07001033 if (existing.state() == GroupState.PENDING_ADD) {
Srikanth Vavilapalli5428b6c2015-05-14 20:22:47 -07001034 notifyDelegate(new GroupEvent(Type.GROUP_ADD_FAILED, existing));
1035 log.warn("groupOperationFailed: cleaningup "
alshabibb0285992016-03-28 23:30:37 -07001036 + "group {} from store in device {}....",
1037 existing.id(),
1038 existing.deviceId());
Srikanth Vavilapalli5428b6c2015-05-14 20:22:47 -07001039 //Removal from groupid based map will happen in the
1040 //map update listener
1041 getGroupStoreKeyMap().remove(new GroupStoreKeyMapKey(existing.deviceId(),
1042 existing.appCookie()));
1043 }
alshabib10580802015-02-18 18:30:33 -08001044 break;
1045 case MODIFY:
1046 notifyDelegate(new GroupEvent(Type.GROUP_UPDATE_FAILED, existing));
1047 break;
1048 case DELETE:
1049 notifyDelegate(new GroupEvent(Type.GROUP_REMOVE_FAILED, existing));
1050 break;
1051 default:
1052 log.warn("Unknown group operation type {}", operation.opType());
1053 }
alshabib10580802015-02-18 18:30:33 -08001054 }
1055
1056 @Override
1057 public void addOrUpdateExtraneousGroupEntry(Group group) {
Srikanth Vavilapalli23181912015-05-04 09:48:09 -07001058 log.debug("add/update extraneous group entry {} in device {}",
alshabibb0285992016-03-28 23:30:37 -07001059 group.id(),
1060 group.deviceId());
alshabib10580802015-02-18 18:30:33 -08001061 ConcurrentMap<GroupId, Group> extraneousIdTable =
1062 getExtraneousGroupIdTable(group.deviceId());
1063 extraneousIdTable.put(group.id(), group);
Srikanth Vavilapallie48b3cf2015-07-06 11:43:07 -07001064 // Don't remove the extraneous groups, instead re-use it when
1065 // a group request comes with the same set of buckets
alshabib10580802015-02-18 18:30:33 -08001066 }
1067
1068 @Override
1069 public void removeExtraneousGroupEntry(Group group) {
Srikanth Vavilapalli23181912015-05-04 09:48:09 -07001070 log.debug("remove extraneous group entry {} of device {} from store",
alshabibb0285992016-03-28 23:30:37 -07001071 group.id(),
1072 group.deviceId());
alshabib10580802015-02-18 18:30:33 -08001073 ConcurrentMap<GroupId, Group> extraneousIdTable =
1074 getExtraneousGroupIdTable(group.deviceId());
1075 extraneousIdTable.remove(group.id());
1076 }
1077
1078 @Override
1079 public Iterable<Group> getExtraneousGroups(DeviceId deviceId) {
1080 // flatten and make iterator unmodifiable
1081 return FluentIterable.from(
1082 getExtraneousGroupIdTable(deviceId).values());
1083 }
1084
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -07001085 /**
Srikanth Vavilapalli6a9d4e42015-03-30 19:41:56 -07001086 * Map handler to receive any events when the group key map is updated.
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -07001087 */
Srikanth Vavilapalli6a9d4e42015-03-30 19:41:56 -07001088 private class GroupStoreKeyMapListener implements
Madan Jampani0b847532016-03-03 13:44:15 -08001089 MapEventListener<GroupStoreKeyMapKey, StoredGroupEntry> {
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -07001090
1091 @Override
Madan Jampani0b847532016-03-03 13:44:15 -08001092 public void event(MapEvent<GroupStoreKeyMapKey, StoredGroupEntry> mapEvent) {
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -07001093 GroupEvent groupEvent = null;
Srikanth Vavilapalli23181912015-05-04 09:48:09 -07001094 GroupStoreKeyMapKey key = mapEvent.key();
Madan Jampani0b847532016-03-03 13:44:15 -08001095 StoredGroupEntry group = Versioned.valueOrNull(mapEvent.newValue());
Srikanth Vavilapalli23181912015-05-04 09:48:09 -07001096 if ((key == null) && (group == null)) {
1097 log.error("GroupStoreKeyMapListener: Received "
alshabibb0285992016-03-28 23:30:37 -07001098 + "event {} with null entry", mapEvent.type());
Srikanth Vavilapalli23181912015-05-04 09:48:09 -07001099 return;
1100 } else if (group == null) {
1101 group = getGroupIdTable(key.deviceId()).values()
1102 .stream()
1103 .filter((storedGroup) -> (storedGroup.appCookie().equals(key.appCookie)))
1104 .findFirst().get();
1105 if (group == null) {
1106 log.error("GroupStoreKeyMapListener: Received "
alshabibb0285992016-03-28 23:30:37 -07001107 + "event {} with null entry... can not process", mapEvent.type());
Srikanth Vavilapalli23181912015-05-04 09:48:09 -07001108 return;
1109 }
1110 }
1111 log.trace("received groupid map event {} for id {} in device {}",
1112 mapEvent.type(),
1113 group.id(),
1114 key.deviceId());
Madan Jampani0b847532016-03-03 13:44:15 -08001115 if (mapEvent.type() == MapEvent.Type.INSERT || mapEvent.type() == MapEvent.Type.UPDATE) {
Srikanth Vavilapalli6a9d4e42015-03-30 19:41:56 -07001116 // Update the group ID table
1117 getGroupIdTable(group.deviceId()).put(group.id(), group);
Madan Jampani0b847532016-03-03 13:44:15 -08001118 StoredGroupEntry value = Versioned.valueOrNull(mapEvent.newValue());
1119 if (value.state() == Group.GroupState.ADDED) {
1120 if (value.isGroupStateAddedFirstTime()) {
1121 groupEvent = new GroupEvent(Type.GROUP_ADDED, value);
Srikanth Vavilapalli23181912015-05-04 09:48:09 -07001122 log.trace("Received first time GROUP_ADDED state update for id {} in device {}",
alshabibb0285992016-03-28 23:30:37 -07001123 group.id(),
1124 group.deviceId());
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -07001125 } else {
Madan Jampani0b847532016-03-03 13:44:15 -08001126 groupEvent = new GroupEvent(Type.GROUP_UPDATED, value);
Srikanth Vavilapalli23181912015-05-04 09:48:09 -07001127 log.trace("Received following GROUP_ADDED state update for id {} in device {}",
alshabibb0285992016-03-28 23:30:37 -07001128 group.id(),
1129 group.deviceId());
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -07001130 }
1131 }
Madan Jampani0b847532016-03-03 13:44:15 -08001132 } else if (mapEvent.type() == MapEvent.Type.REMOVE) {
Srikanth Vavilapalli23181912015-05-04 09:48:09 -07001133 groupEvent = new GroupEvent(Type.GROUP_REMOVED, group);
Srikanth Vavilapalli6a9d4e42015-03-30 19:41:56 -07001134 // Remove the entry from the group ID table
1135 getGroupIdTable(group.deviceId()).remove(group.id(), group);
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -07001136 }
1137
1138 if (groupEvent != null) {
1139 notifyDelegate(groupEvent);
1140 }
1141 }
1142 }
Madan Jampani01e05fb2015-08-13 13:29:36 -07001143
helenyrwua1c41152016-08-18 16:16:14 -07001144 private void processGroupMessage(GroupStoreMessage message) {
1145 if (message.type() == GroupStoreMessage.Type.FAILOVER) {
1146 // FIXME: groupStoreEntriesByKey inaccessible here
1147 getGroupIdTable(message.deviceId()).values()
1148 .stream()
1149 .filter((storedGroup) -> (storedGroup.appCookie().equals(message.appCookie())))
1150 .findFirst().ifPresent(group -> notifyDelegate(new GroupEvent(Type.GROUP_BUCKET_FAILOVER, group)));
1151 }
1152 }
1153
Madan Jampani01e05fb2015-08-13 13:29:36 -07001154 private void process(GroupStoreMessage groupOp) {
1155 log.debug("Received remote group operation {} request for device {}",
alshabibb0285992016-03-28 23:30:37 -07001156 groupOp.type(),
1157 groupOp.deviceId());
1158 if (!mastershipService.isLocalMaster(groupOp.deviceId())) {
1159 log.warn("This node is not MASTER for device {}", groupOp.deviceId());
1160 return;
1161 }
1162 if (groupOp.type() == GroupStoreMessage.Type.ADD) {
1163 storeGroupDescriptionInternal(groupOp.groupDesc());
1164 } else if (groupOp.type() == GroupStoreMessage.Type.UPDATE) {
1165 updateGroupDescriptionInternal(groupOp.deviceId(),
1166 groupOp.appCookie(),
1167 groupOp.updateType(),
1168 groupOp.updateBuckets(),
1169 groupOp.newAppCookie());
1170 } else if (groupOp.type() == GroupStoreMessage.Type.DELETE) {
1171 deleteGroupDescriptionInternal(groupOp.deviceId(),
1172 groupOp.appCookie());
1173 }
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -07001174 }
1175
1176 /**
1177 * Flattened map key to be used to store group entries.
1178 */
Ray Milkeyb3c5ce22015-08-10 09:07:36 -07001179 protected static class GroupStoreMapKey {
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -07001180 private final DeviceId deviceId;
1181
1182 public GroupStoreMapKey(DeviceId deviceId) {
1183 this.deviceId = deviceId;
1184 }
1185
Srikanth Vavilapalli23181912015-05-04 09:48:09 -07001186 public DeviceId deviceId() {
1187 return deviceId;
1188 }
1189
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -07001190 @Override
1191 public boolean equals(Object o) {
1192 if (this == o) {
1193 return true;
1194 }
1195 if (!(o instanceof GroupStoreMapKey)) {
1196 return false;
1197 }
1198 GroupStoreMapKey that = (GroupStoreMapKey) o;
1199 return this.deviceId.equals(that.deviceId);
1200 }
1201
1202 @Override
1203 public int hashCode() {
1204 int result = 17;
1205
1206 result = 31 * result + Objects.hash(this.deviceId);
1207
1208 return result;
1209 }
1210 }
1211
Ray Milkeyb3c5ce22015-08-10 09:07:36 -07001212 protected static class GroupStoreKeyMapKey extends GroupStoreMapKey {
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -07001213 private final GroupKey appCookie;
alshabibb0285992016-03-28 23:30:37 -07001214
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -07001215 public GroupStoreKeyMapKey(DeviceId deviceId,
1216 GroupKey appCookie) {
1217 super(deviceId);
1218 this.appCookie = appCookie;
1219 }
1220
1221 @Override
1222 public boolean equals(Object o) {
1223 if (this == o) {
1224 return true;
1225 }
1226 if (!(o instanceof GroupStoreKeyMapKey)) {
1227 return false;
1228 }
1229 GroupStoreKeyMapKey that = (GroupStoreKeyMapKey) o;
1230 return (super.equals(that) &&
1231 this.appCookie.equals(that.appCookie));
1232 }
1233
1234 @Override
1235 public int hashCode() {
1236 int result = 17;
1237
1238 result = 31 * result + super.hashCode() + Objects.hash(this.appCookie);
1239
1240 return result;
1241 }
1242 }
1243
Ray Milkeyb3c5ce22015-08-10 09:07:36 -07001244 protected static class GroupStoreIdMapKey extends GroupStoreMapKey {
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -07001245 private final GroupId groupId;
alshabibb0285992016-03-28 23:30:37 -07001246
Srikanth Vavilapalli717361f2015-03-16 12:06:04 -07001247 public GroupStoreIdMapKey(DeviceId deviceId,
1248 GroupId groupId) {
1249 super(deviceId);
1250 this.groupId = groupId;
1251 }
1252
1253 @Override
1254 public boolean equals(Object o) {
1255 if (this == o) {
1256 return true;
1257 }
1258 if (!(o instanceof GroupStoreIdMapKey)) {
1259 return false;
1260 }
1261 GroupStoreIdMapKey that = (GroupStoreIdMapKey) o;
1262 return (super.equals(that) &&
1263 this.groupId.equals(that.groupId));
1264 }
1265
1266 @Override
1267 public int hashCode() {
1268 int result = 17;
1269
1270 result = 31 * result + super.hashCode() + Objects.hash(this.groupId);
1271
1272 return result;
1273 }
1274 }
Srikanth Vavilapalli23181912015-05-04 09:48:09 -07001275
1276 @Override
1277 public void pushGroupMetrics(DeviceId deviceId,
1278 Collection<Group> groupEntries) {
1279 boolean deviceInitialAuditStatus =
1280 deviceInitialAuditStatus(deviceId);
1281 Set<Group> southboundGroupEntries =
1282 Sets.newHashSet(groupEntries);
1283 Set<StoredGroupEntry> storedGroupEntries =
1284 Sets.newHashSet(getStoredGroups(deviceId));
1285 Set<Group> extraneousStoredEntries =
1286 Sets.newHashSet(getExtraneousGroups(deviceId));
1287
Sho SHIMIZU695bac62016-08-15 12:41:59 -07001288 if (log.isTraceEnabled()) {
1289 log.trace("pushGroupMetrics: Displaying all ({}) southboundGroupEntries for device {}",
1290 southboundGroupEntries.size(),
1291 deviceId);
1292 for (Group group : southboundGroupEntries) {
1293 log.trace("Group {} in device {}", group, deviceId);
1294 }
Srikanth Vavilapalli23181912015-05-04 09:48:09 -07001295
Sho SHIMIZU695bac62016-08-15 12:41:59 -07001296 log.trace("Displaying all ({}) stored group entries for device {}",
1297 storedGroupEntries.size(),
1298 deviceId);
1299 for (StoredGroupEntry group : storedGroupEntries) {
1300 log.trace("Stored Group {} for device {}", group, deviceId);
1301 }
Srikanth Vavilapalli23181912015-05-04 09:48:09 -07001302 }
1303
alshabibb0285992016-03-28 23:30:37 -07001304 garbageCollect(deviceId, southboundGroupEntries, storedGroupEntries);
1305
Srikanth Vavilapalli23181912015-05-04 09:48:09 -07001306 for (Iterator<Group> it2 = southboundGroupEntries.iterator(); it2.hasNext();) {
1307 Group group = it2.next();
1308 if (storedGroupEntries.remove(group)) {
1309 // we both have the group, let's update some info then.
1310 log.trace("Group AUDIT: group {} exists in both planes for device {}",
alshabibb0285992016-03-28 23:30:37 -07001311 group.id(), deviceId);
1312
Srikanth Vavilapalli23181912015-05-04 09:48:09 -07001313 groupAdded(group);
1314 it2.remove();
1315 }
1316 }
1317 for (Group group : southboundGroupEntries) {
1318 if (getGroup(group.deviceId(), group.id()) != null) {
1319 // There is a group existing with the same id
1320 // It is possible that group update is
1321 // in progress while we got a stale info from switch
1322 if (!storedGroupEntries.remove(getGroup(
alshabibb0285992016-03-28 23:30:37 -07001323 group.deviceId(), group.id()))) {
Srikanth Vavilapalli23181912015-05-04 09:48:09 -07001324 log.warn("Group AUDIT: Inconsistent state:"
alshabibb0285992016-03-28 23:30:37 -07001325 + "Group exists in ID based table while "
1326 + "not present in key based table");
Srikanth Vavilapalli23181912015-05-04 09:48:09 -07001327 }
1328 } else {
1329 // there are groups in the switch that aren't in the store
1330 log.debug("Group AUDIT: extraneous group {} exists in data plane for device {}",
alshabibb0285992016-03-28 23:30:37 -07001331 group.id(), deviceId);
Srikanth Vavilapalli23181912015-05-04 09:48:09 -07001332 extraneousStoredEntries.remove(group);
1333 extraneousGroup(group);
1334 }
1335 }
1336 for (Group group : storedGroupEntries) {
1337 // there are groups in the store that aren't in the switch
1338 log.debug("Group AUDIT: group {} missing in data plane for device {}",
alshabibb0285992016-03-28 23:30:37 -07001339 group.id(), deviceId);
Srikanth Vavilapalli23181912015-05-04 09:48:09 -07001340 groupMissing(group);
1341 }
1342 for (Group group : extraneousStoredEntries) {
1343 // there are groups in the extraneous store that
1344 // aren't in the switch
Saurav Das0fd79d92016-03-07 10:58:36 -08001345 log.debug("Group AUDIT: clearing extraneous group {} from store for device {}",
alshabibb0285992016-03-28 23:30:37 -07001346 group.id(), deviceId);
Srikanth Vavilapalli23181912015-05-04 09:48:09 -07001347 removeExtraneousGroupEntry(group);
1348 }
1349
1350 if (!deviceInitialAuditStatus) {
Saurav Das0fd79d92016-03-07 10:58:36 -08001351 log.info("Group AUDIT: Setting device {} initial AUDIT completed",
alshabibb0285992016-03-28 23:30:37 -07001352 deviceId);
Srikanth Vavilapalli23181912015-05-04 09:48:09 -07001353 deviceInitialAuditCompleted(deviceId, true);
1354 }
1355 }
1356
helenyrwu89470f12016-08-12 13:18:10 -07001357 @Override
1358 public void notifyOfFailovers(Collection<Group> failoverGroups) {
helenyrwu89470f12016-08-12 13:18:10 -07001359 failoverGroups.forEach(group -> {
1360 if (group.type() == Group.Type.FAILOVER) {
helenyrwua1c41152016-08-18 16:16:14 -07001361 groupTopic.publish(GroupStoreMessage.createGroupFailoverMsg(
1362 group.deviceId(), group));
helenyrwu89470f12016-08-12 13:18:10 -07001363 }
1364 });
helenyrwu89470f12016-08-12 13:18:10 -07001365 }
1366
alshabibb0285992016-03-28 23:30:37 -07001367 private void garbageCollect(DeviceId deviceId,
1368 Set<Group> southboundGroupEntries,
1369 Set<StoredGroupEntry> storedGroupEntries) {
1370 if (!garbageCollect) {
1371 return;
1372 }
1373
1374 Iterator<StoredGroupEntry> it = storedGroupEntries.iterator();
1375 while (it.hasNext()) {
1376 StoredGroupEntry group = it.next();
1377 if (group.state() != GroupState.PENDING_DELETE && checkGroupRefCount(group)) {
1378 log.debug("Garbage collecting group {} on {}", group, deviceId);
1379 deleteGroupDescription(deviceId, group.appCookie());
1380 southboundGroupEntries.remove(group);
1381 it.remove();
1382 }
1383 }
1384 }
1385
1386 private boolean checkGroupRefCount(Group group) {
1387 return (group.referenceCount() == 0 && group.age() >= gcThresh);
1388 }
1389
Srikanth Vavilapalli23181912015-05-04 09:48:09 -07001390 private void groupMissing(Group group) {
1391 switch (group.state()) {
1392 case PENDING_DELETE:
1393 log.debug("Group {} delete confirmation from device {}",
1394 group, group.deviceId());
1395 removeGroupEntry(group);
1396 break;
1397 case ADDED:
1398 case PENDING_ADD:
Srikanth Vavilapalli5428b6c2015-05-14 20:22:47 -07001399 case PENDING_ADD_RETRY:
Srikanth Vavilapalli23181912015-05-04 09:48:09 -07001400 case PENDING_UPDATE:
1401 log.debug("Group {} is in store but not on device {}",
1402 group, group.deviceId());
1403 StoredGroupEntry existing =
1404 getStoredGroupEntry(group.deviceId(), group.id());
Srikanth Vavilapalli5428b6c2015-05-14 20:22:47 -07001405 log.debug("groupMissing: group entry {} in device {} moving from {} to PENDING_ADD_RETRY",
alshabibb0285992016-03-28 23:30:37 -07001406 existing.id(),
1407 existing.deviceId(),
1408 existing.state());
Srikanth Vavilapalli5428b6c2015-05-14 20:22:47 -07001409 existing.setState(Group.GroupState.PENDING_ADD_RETRY);
Srikanth Vavilapalli23181912015-05-04 09:48:09 -07001410 //Re-PUT map entries to trigger map update events
1411 getGroupStoreKeyMap().
alshabibb0285992016-03-28 23:30:37 -07001412 put(new GroupStoreKeyMapKey(existing.deviceId(),
1413 existing.appCookie()), existing);
Srikanth Vavilapalli23181912015-05-04 09:48:09 -07001414 notifyDelegate(new GroupEvent(GroupEvent.Type.GROUP_ADD_REQUESTED,
1415 group));
1416 break;
1417 default:
1418 log.debug("Group {} has not been installed.", group);
1419 break;
1420 }
1421 }
1422
1423 private void extraneousGroup(Group group) {
Saurav Das0fd79d92016-03-07 10:58:36 -08001424 log.trace("Group {} is on device {} but not in store.",
Srikanth Vavilapalli23181912015-05-04 09:48:09 -07001425 group, group.deviceId());
1426 addOrUpdateExtraneousGroupEntry(group);
1427 }
1428
1429 private void groupAdded(Group group) {
1430 log.trace("Group {} Added or Updated in device {}",
1431 group, group.deviceId());
1432 addOrUpdateGroupEntry(group);
1433 }
alshabib10580802015-02-18 18:30:33 -08001434}