You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@ratis.apache.org by ru...@apache.org on 2020/11/02 01:40:01 UTC
[incubator-ratis] branch master updated: RATIS-1125. Fix
TestDataStream (#247)
This is an automated email from the ASF dual-hosted git repository.
runzhiwang pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-ratis.git
The following commit(s) were added to refs/heads/master by this push:
new e0958c9 RATIS-1125. Fix TestDataStream (#247)
e0958c9 is described below
commit e0958c9842f0236fa7b996efcc8121d1047274e8
Author: Rui Wang <am...@users.noreply.github.com>
AuthorDate: Sun Nov 1 17:39:52 2020 -0800
RATIS-1125. Fix TestDataStream (#247)
* RATIS.1125. Fix TestDataStream
* fixup! add comment back
---
.../java/org/apache/ratis/datastream/DataStreamBaseTest.java | 11 +++--------
1 file changed, 3 insertions(+), 8 deletions(-)
diff --git a/ratis-test/src/test/java/org/apache/ratis/datastream/DataStreamBaseTest.java b/ratis-test/src/test/java/org/apache/ratis/datastream/DataStreamBaseTest.java
index dd8505f..7646651 100644
--- a/ratis-test/src/test/java/org/apache/ratis/datastream/DataStreamBaseTest.java
+++ b/ratis-test/src/test/java/org/apache/ratis/datastream/DataStreamBaseTest.java
@@ -22,7 +22,6 @@ import org.apache.ratis.MiniRaftCluster;
import org.apache.ratis.client.impl.DataStreamClientImpl;
import org.apache.ratis.client.impl.DataStreamClientImpl.DataStreamOutputImpl;
import org.apache.ratis.conf.RaftProperties;
-import org.apache.ratis.netty.NettyConfigKeys;
import org.apache.ratis.proto.RaftProtos.*;
import org.apache.ratis.protocol.DataStreamReply;
import org.apache.ratis.protocol.GroupInfoReply;
@@ -151,11 +150,9 @@ abstract class DataStreamBaseTest extends BaseTest {
private List<DataStreamServerImpl> servers;
private List<RaftPeer> peers;
- private List<MultiDataStreamStateMachine> stateMachines;
+ private ConcurrentMap<RaftGroupId, MultiDataStreamStateMachine> stateMachines;
protected RaftServer newRaftServer(RaftPeer peer, RaftProperties properties) {
- final ConcurrentMap<RaftGroupId, StateMachine> stateMachines = new ConcurrentHashMap<>();
-
return new RaftServer() {
@Override
public RaftPeerId getId() {
@@ -285,11 +282,9 @@ abstract class DataStreamBaseTest extends BaseTest {
.map(id -> new RaftPeer(id, NetUtils.createLocalServerAddress()))
.collect(Collectors.toList());
servers = new ArrayList<>(peers.size());
- stateMachines = new ArrayList<>(peers.size());
+ stateMachines = new ConcurrentHashMap<>();
// start stream servers on raft peers.
for (int i = 0; i < peers.size(); i++) {
- final MultiDataStreamStateMachine stateMachine = new MultiDataStreamStateMachine();
- stateMachines.add(stateMachine);
final RaftPeer peer = peers.get(i);
final RaftServer server = newRaftServer(peer, properties);
final DataStreamServerImpl streamServer = new DataStreamServerImpl(server, properties, null);
@@ -378,7 +373,7 @@ abstract class DataStreamBaseTest extends BaseTest {
}
final RaftClientRequest header = out.getHeader();
- for (MultiDataStreamStateMachine s : stateMachines) {
+ for (MultiDataStreamStateMachine s : stateMachines.values()) {
final SingleDataStream stream = s.getSingleDataStream(header.getCallId());
if (stream == null) {
continue;