crabapples
/
Sentinel
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Make ClusterNodeTest.testGetOrCreateOriginNodeMultiThread stable

This commit is contained in:
jason 2019-04-12 10:23:21 +08:00 committed by Eric Zhao
parent 16082a875c
commit 73f166e258
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
sentinel-core/src/test/java/com/alibaba/csp/sentinel/node/ClusterNodeTest.java
View File

@ -23,7 +23,6 @@ import java.util.ArrayList;
import java.util.Arrays; import java.util.Arrays;
import java.util.Collections; import java.util.Collections;
import java.util.List; import java.util.List;
import java.util.Random;
import java.util.Set; import java.util.Set;
import java.util.concurrent.Callable; import java.util.concurrent.Callable;
import java.util.concurrent.ConcurrentHashMap; import java.util.concurrent.ConcurrentHashMap;
@ -81,8 +80,6 @@ public class ClusterNodeTest {
// Here we need a thread-safe concurrent set (created from ConcurrentHashMap). // Here we need a thread-safe concurrent set (created from ConcurrentHashMap).
final Set<Node> createdNodes = Collections.newSetFromMap(new ConcurrentHashMap<Node, Boolean>()); final Set<Node> createdNodes = Collections.newSetFromMap(new ConcurrentHashMap<Node, Boolean>());
final Random random = new Random();
// 10 threads, 3 origins, 20 tasks (calling 20 times of ClusterNode#getOrCreateOriginNode concurrently) // 10 threads, 3 origins, 20 tasks (calling 20 times of ClusterNode#getOrCreateOriginNode concurrently)
final ExecutorService es = Executors.newFixedThreadPool(10); final ExecutorService es = Executors.newFixedThreadPool(10);
final List<String> origins = Arrays.asList("origin1", "origin2", "origin3"); final List<String> origins = Arrays.asList("origin1", "origin2", "origin3");
@ -91,11 +88,12 @@ public class ClusterNodeTest {
List<Callable<Object>> tasks = new ArrayList<Callable<Object>>(taskCount); List<Callable<Object>> tasks = new ArrayList<Callable<Object>>(taskCount);
for (int i = 0; i < taskCount; i++) { for (int i = 0; i < taskCount; i++) {
final int index = i % origins.size();
tasks.add(new Callable<Object>() { tasks.add(new Callable<Object>() {
@Override @Override
public Object call() throws Exception { public Object call() throws Exception {
// one task call one times of ClusterNode#getOrCreateOriginNode // one task call one times of ClusterNode#getOrCreateOriginNode
Node node = clusterNode.getOrCreateOriginNode(origins.get(random.nextInt(origins.size()))); Node node = clusterNode.getOrCreateOriginNode(origins.get(index));
// add the result node to the createdNodes set // add the result node to the createdNodes set
createdNodes.add(node); createdNodes.add(node);
latch.countDown(); latch.countDown();