Provide pendingDevices explicitly
Change-Id: I2cc29d5672d11aa176c7fa774e9d66f32b2d300f
diff --git a/core/net/src/main/java/org/onosproject/net/flow/impl/FlowRuleManager.java b/core/net/src/main/java/org/onosproject/net/flow/impl/FlowRuleManager.java
index 9e0db6a..f1de6fa 100644
--- a/core/net/src/main/java/org/onosproject/net/flow/impl/FlowRuleManager.java
+++ b/core/net/src/main/java/org/onosproject/net/flow/impl/FlowRuleManager.java
@@ -16,6 +16,7 @@
package org.onosproject.net.flow.impl;
import com.google.common.collect.ArrayListMultimap;
+import com.google.common.collect.ImmutableSet;
import com.google.common.collect.Iterables;
import com.google.common.collect.Lists;
import com.google.common.collect.Maps;
@@ -600,22 +601,22 @@
private class FlowOperationsProcessor implements Runnable {
// Immutable
private final FlowRuleOperations fops;
+ private final ImmutableSet<DeviceId> pendingDevices;
// Mutable
private final List<Set<FlowRuleOperation>> stages;
- private final Set<DeviceId> pendingDevices;
private boolean hasFailed = false;
FlowOperationsProcessor(FlowRuleOperations ops) {
this.stages = Lists.newArrayList(ops.stages());
this.fops = ops;
- this.pendingDevices = new HashSet<>();
+ this.pendingDevices = ImmutableSet.of();
}
- FlowOperationsProcessor(FlowOperationsProcessor src, boolean hasFailed) {
+ FlowOperationsProcessor(FlowOperationsProcessor src, boolean hasFailed, Set<DeviceId> pendingDevices) {
this.fops = src.fops;
this.stages = Lists.newArrayList(src.stages);
- this.pendingDevices = new HashSet<>(src.pendingDevices);
+ this.pendingDevices = ImmutableSet.copyOf(pendingDevices);
this.hasFailed = hasFailed;
}
@@ -635,28 +636,33 @@
perDeviceBatches.put(op.rule().deviceId(),
new FlowRuleBatchEntry(mapOperationType(op.type()), op.rule()));
}
- pendingDevices.addAll(perDeviceBatches.keySet());
+ ImmutableSet<DeviceId> newPendingDevices = ImmutableSet.<DeviceId>builder()
+ .addAll(pendingDevices)
+ .addAll(perDeviceBatches.keySet())
+ .build();
for (DeviceId deviceId : perDeviceBatches.keySet()) {
long id = idGenerator.getNewId();
final FlowRuleBatchOperation b = new FlowRuleBatchOperation(perDeviceBatches.get(deviceId),
deviceId, id);
- pendingFlowOperations.put(id, this);
+ pendingFlowOperations.put(id, new FlowOperationsProcessor(this, hasFailed, newPendingDevices));
deviceInstallers.execute(() -> store.storeBatch(b));
}
}
synchronized void satisfy(DeviceId devId) {
- pendingDevices.remove(devId);
- if (pendingDevices.isEmpty()) {
- operationsService.execute(new FlowOperationsProcessor(this, hasFailed));
+ Set<DeviceId> newPendingDevices = new HashSet<>(pendingDevices);
+ newPendingDevices.remove(devId);
+ if (newPendingDevices.isEmpty()) {
+ operationsService.execute(new FlowOperationsProcessor(this, hasFailed, newPendingDevices));
}
}
synchronized void fail(DeviceId devId, Set<? extends FlowRule> failures) {
- pendingDevices.remove(devId);
- if (pendingDevices.isEmpty()) {
- operationsService.execute(new FlowOperationsProcessor(this, true));
+ Set<DeviceId> newPendingDevices = new HashSet<>(pendingDevices);
+ newPendingDevices.remove(devId);
+ if (newPendingDevices.isEmpty()) {
+ operationsService.execute(new FlowOperationsProcessor(this, true, newPendingDevices));
}
FlowRuleOperations.Builder failedOpsBuilder = FlowRuleOperations.builder();