Change SqlStageExecution.completeSources to concurrent set

Completed sources is accessed from synchronized and unsynchronized contexts
prestodb · Apr 29, 2015 · a32f88b · a32f88b
1 parent 8a97b14
commit a32f88b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/presto-main/src/main/java/com/facebook/presto/execution/SqlStageExecution.java b/presto-main/src/main/java/com/facebook/presto/execution/SqlStageExecution.java
@@ -90,6 +90,7 @@
 import static com.google.common.collect.Iterables.all;
 import static com.google.common.collect.Iterables.any;
 import static com.google.common.collect.Iterables.transform;
+import static com.google.common.collect.Sets.newConcurrentHashSet;
 import static io.airlift.http.client.HttpUriBuilder.uriBuilderFrom;
 import static io.airlift.units.DataSize.Unit.BYTE;
 import static java.util.concurrent.TimeUnit.NANOSECONDS;
@@ -126,7 +127,7 @@ public final class SqlStageExecution
 
     private final LinkedBlockingQueue<Throwable> failureCauses = new LinkedBlockingQueue<>();
 
-    private final Set<PlanNodeId> completeSources = new HashSet<>();
+    private final Set<PlanNodeId> completeSources = newConcurrentHashSet();
 
     @GuardedBy("this")
     private OutputBuffers currentOutputBuffers = INITIAL_EMPTY_OUTPUT_BUFFERS;