|
59 | 59 | import static org.junit.jupiter.api.Assertions.fail; |
60 | 60 |
|
61 | 61 | public class RpcClientRetryTest extends ShuffleReadWriteBase { |
62 | | - private static List<ShuffleServerInfo> grpcShuffleServerInfoList = Lists.newArrayList(); |
| 62 | + private static List<ShuffleServerInfo> grpcShuffleServerInfoList; |
63 | 63 | private static MockedShuffleWriteClientImpl shuffleWriteClientImpl; |
64 | 64 |
|
65 | 65 | private ShuffleClientFactory.ReadClientBuilder baseReadBuilder(StorageType storageType) { |
@@ -94,13 +94,13 @@ public static MockedShuffleServer createMockedShuffleServer( |
94 | 94 | public void initCluster(@TempDir File tmpDir) throws Exception { |
95 | 95 | CoordinatorConf coordinatorConf = getCoordinatorConf(); |
96 | 96 | createCoordinatorServer(coordinatorConf); |
97 | | - for (int i = 0; i < 1; i++) { |
| 97 | + for (int i = 0; i < 3; i++) { |
98 | 98 | grpcShuffleServers.add(createMockedShuffleServer(i, tmpDir, ServerType.GRPC)); |
99 | 99 | } |
100 | 100 |
|
101 | 101 | startServers(); |
102 | | - |
103 | | - for (int i = 0; i < 1; i++) { |
| 102 | + grpcShuffleServerInfoList = Lists.newArrayList(); |
| 103 | + for (int i = 0; i < 3; i++) { |
104 | 104 | grpcShuffleServerInfoList.add( |
105 | 105 | new ShuffleServerInfo( |
106 | 106 | String.format("127.0.0.1-%s", grpcShuffleServers.get(i).getGrpcPort()), |
@@ -285,6 +285,8 @@ private void registerShuffleServer( |
285 | 285 | .unregisterTimeSec(10) |
286 | 286 | .unregisterRequestTimeSec(10)); |
287 | 287 |
|
| 288 | + System.out.println("============"); |
| 289 | + System.out.println(grpcShuffleServerInfoList); |
288 | 290 | for (int i = 0; i < replica; i++) { |
289 | 291 | shuffleWriteClientImpl.registerShuffle( |
290 | 292 | grpcShuffleServerInfoList.get(i), |
|
0 commit comments