import akka.actor.PoisonPill;
import com.google.common.collect.HashMultimap;
import com.google.common.collect.Multimap;
-import java.util.Arrays;
import java.util.Collection;
+import java.util.Collections;
+import java.util.HashSet;
import java.util.IdentityHashMap;
import java.util.Map;
-import org.opendaylight.controller.cluster.datastore.entityownership.messages.EntityOwnershipChanged;
+import java.util.Set;
import org.opendaylight.controller.md.sal.common.api.clustering.Entity;
+import org.opendaylight.controller.md.sal.common.api.clustering.EntityOwnershipChange;
import org.opendaylight.controller.md.sal.common.api.clustering.EntityOwnershipListener;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
class EntityOwnershipListenerSupport {
private static final Logger LOG = LoggerFactory.getLogger(EntityOwnershipListenerSupport.class);
+ private final String logId;
private final ActorContext actorContext;
private final Map<EntityOwnershipListener, ListenerActorRefEntry> listenerActorMap = new IdentityHashMap<>();
- private final Multimap<Entity, EntityOwnershipListener> entityListenerMap = HashMultimap.create();
+ private final Set<Entity> entitiesWithCandidateSet = new HashSet<>();
private final Multimap<String, EntityOwnershipListener> entityTypeListenerMap = HashMultimap.create();
+ private volatile boolean inJeopardy = false;
- EntityOwnershipListenerSupport(ActorContext actorContext) {
+ EntityOwnershipListenerSupport(ActorContext actorContext, String logId) {
this.actorContext = actorContext;
+ this.logId = logId;
}
- void addEntityOwnershipListener(Entity entity, EntityOwnershipListener listener) {
- LOG.debug("Adding EntityOwnershipListener {} for {}", listener, entity);
+ String getLogId() {
+ return logId;
+ }
- addListener(listener, entity, entityListenerMap);
+ /**
+ * Set the in-jeopardy flag and indicate its previous state.
+ *
+ * @param inJeopardy new value of the in-jeopardy flag
+ * @return Previous value of the flag.
+ */
+ boolean setInJeopardy(final boolean inJeopardy) {
+ final boolean wasInJeopardy = this.inJeopardy;
+ this.inJeopardy = inJeopardy;
+ return wasInJeopardy;
}
- void addEntityOwnershipListener(String entityType, EntityOwnershipListener listener) {
- LOG.debug("Adding EntityOwnershipListener {} for entity type {}", listener, entityType);
+ boolean hasCandidateForEntity(Entity entity) {
+ return entitiesWithCandidateSet.contains(entity);
+ }
- addListener(listener, entityType, entityTypeListenerMap);
+ void setHasCandidateForEntity(Entity entity) {
+ entitiesWithCandidateSet.add(entity);
}
- void removeEntityOwnershipListener(Entity entity, EntityOwnershipListener listener) {
- LOG.debug("Removing EntityOwnershipListener {} for {}", listener, entity);
+ void unsetHasCandidateForEntity(Entity entity) {
+ entitiesWithCandidateSet.remove(entity);
+ }
+
+ void addEntityOwnershipListener(String entityType, EntityOwnershipListener listener) {
+ LOG.debug("{}: Adding EntityOwnershipListener {} for entity type {}", logId, listener, entityType);
- removeListener(listener, entity, entityListenerMap);
+ addListener(listener, entityType);
}
void removeEntityOwnershipListener(String entityType, EntityOwnershipListener listener) {
- LOG.debug("Removing EntityOwnershipListener {} for entity type {}", listener, entityType);
+ LOG.debug("{}: Removing EntityOwnershipListener {} for entity type {}", logId, listener, entityType);
- removeListener(listener, entityType, entityTypeListenerMap);
+ removeListener(listener, entityType);
}
- void notifyEntityOwnershipListeners(Entity entity, boolean wasOwner, boolean isOwner) {
- notifyListeners(entity, entity, wasOwner, isOwner, entityListenerMap);
- notifyListeners(entity, entity.getType(), wasOwner, isOwner, entityTypeListenerMap);
+ void notifyEntityOwnershipListeners(Entity entity, boolean wasOwner, boolean isOwner, boolean hasOwner) {
+ notifyListeners(entity, entity.getType(), wasOwner, isOwner, hasOwner);
}
- void notifyEntityOwnershipListener(Entity entity, boolean wasOwner, boolean isOwner,
+ void notifyEntityOwnershipListener(Entity entity, boolean wasOwner, boolean isOwner, boolean hasOwner,
EntityOwnershipListener listener) {
- notifyListeners(entity, wasOwner, isOwner, Arrays.asList(listener));
+ notifyListeners(entity, wasOwner, isOwner, hasOwner, Collections.singleton(listener));
}
- private <T> void notifyListeners(Entity entity, T mapKey, boolean wasOwner, boolean isOwner,
- Multimap<T, EntityOwnershipListener> listenerMap) {
- Collection<EntityOwnershipListener> listeners = listenerMap.get(mapKey);
+ private void notifyListeners(Entity entity, String mapKey, boolean wasOwner, boolean isOwner, boolean hasOwner) {
+ Collection<EntityOwnershipListener> listeners = entityTypeListenerMap.get(mapKey);
if(!listeners.isEmpty()) {
- notifyListeners(entity, wasOwner, isOwner, listeners);
+ notifyListeners(entity, wasOwner, isOwner, hasOwner, listeners);
}
}
- private void notifyListeners(Entity entity, boolean wasOwner, boolean isOwner,
+ private void notifyListeners(Entity entity, boolean wasOwner, boolean isOwner, boolean hasOwner,
Collection<EntityOwnershipListener> listeners) {
- EntityOwnershipChanged changed = new EntityOwnershipChanged(entity, wasOwner, isOwner);
+ EntityOwnershipChange changed = new EntityOwnershipChange(entity, wasOwner, isOwner, hasOwner, inJeopardy);
for(EntityOwnershipListener listener: listeners) {
ActorRef listenerActor = listenerActorFor(listener);
- LOG.debug("Notifying EntityOwnershipListenerActor {} with {}", listenerActor, changed);
+ LOG.debug("{}: Notifying EntityOwnershipListenerActor {} with {}", logId, listenerActor, changed);
listenerActor.tell(changed, ActorRef.noSender());
}
}
- private <T> void addListener(EntityOwnershipListener listener, T mapKey,
- Multimap<T, EntityOwnershipListener> toListenerMap) {
- if(toListenerMap.put(mapKey, listener)) {
+ private void addListener(EntityOwnershipListener listener, String mapKey) {
+ if (entityTypeListenerMap.put(mapKey, listener)) {
ListenerActorRefEntry listenerEntry = listenerActorMap.get(listener);
if(listenerEntry == null) {
listenerActorMap.put(listener, new ListenerActorRefEntry());
}
}
- private <T> void removeListener(EntityOwnershipListener listener, T mapKey,
- Multimap<T, EntityOwnershipListener> fromListenerMap) {
- if(fromListenerMap.remove(mapKey, listener)) {
+ private void removeListener(EntityOwnershipListener listener, String mapKey) {
+ if (entityTypeListenerMap.remove(mapKey, listener)) {
ListenerActorRefEntry listenerEntry = listenerActorMap.get(listener);
- LOG.debug("Found {}", listenerEntry);
+ LOG.debug("{}: Found {}", logId, listenerEntry);
listenerEntry.referenceCount--;
if(listenerEntry.referenceCount <= 0) {
if(actorRef == null) {
actorRef = actorContext.actorOf(EntityOwnershipListenerActor.props(listener));
- LOG.debug("Created EntityOwnershipListenerActor {} for listener {}", actorRef, listener);
+ LOG.debug("{}: Created EntityOwnershipListenerActor {} for listener {}", logId, actorRef, listener);
}
return actorRef;