import org.apache.tajo.rpc.test.impl.DummyProtocolAsyncImpl; //导入依赖的package包/类
@BeforeClass
public static void setupClass() throws Exception {
  final DummyProtocolAsyncImpl service = new DummyProtocolAsyncImpl();
  server = new AsyncRpcServer(DummyProtocol.class,
      service, new InetSocketAddress("127.0.0.1", 0), 10);
  server.start();

}
 

import org.apache.tajo.rpc.test.impl.DummyProtocolAsyncImpl; //导入依赖的package包/类
@Before
public void setUp() throws Exception {
  service = new DummyProtocolAsyncImpl();
  server = new AsyncRpcServer(DummyProtocol.class,
      service, new InetSocketAddress("127.0.0.1", 0), 2);
  server.start();
  client = new AsyncRpcClient(DummyProtocol.class,
      NetUtils.getConnectAddress(server.getListenAddress()));
  stub = client.getStub();
}
 

import org.apache.tajo.rpc.test.impl.DummyProtocolAsyncImpl; //导入依赖的package包/类
public void setUpRpcServer() throws Exception {
  service = new DummyProtocolAsyncImpl();
  server = new AsyncRpcServer(DummyProtocol.class,
      service, new InetSocketAddress("127.0.0.1", 0), 3);
  server.start();
}
 

import org.apache.tajo.rpc.test.impl.DummyProtocolAsyncImpl; //导入依赖的package包/类
@Test
@SetupRpcConnection(setupRpcServer = false, setupRpcClient = false)
public void testClientRetryOnStartup() throws Exception {
  retries = 10;
  ServerSocket serverSocket = new ServerSocket(0);
  final InetSocketAddress address = new InetSocketAddress("127.0.0.1", serverSocket.getLocalPort());
  serverSocket.close();
  service = new DummyProtocolAsyncImpl();

  EchoMessage echoMessage = EchoMessage.newBuilder()
      .setMessage(MESSAGE).build();
  CallFuture<EchoMessage> future = new CallFuture<>();

  //lazy startup
  Thread serverThread = new Thread(new Runnable() {
    @Override
    public void run() {
      try {
        Thread.sleep(1000);
        server = new AsyncRpcServer(DummyProtocol.class,
            service, address, 2);
      } catch (Exception e) {
        fail(e.getMessage());
      }
      server.start();
    }
  });
  serverThread.start();

  RpcConnectionKey rpcConnectionKey =
      new RpcConnectionKey(address, DummyProtocol.class, true);

  Properties connParams = new Properties();
  connParams.setProperty(RpcConstants.CLIENT_RETRY_NUM, String.valueOf(retries));

  AsyncRpcClient client = manager.newClient(rpcConnectionKey, connParams);
  assertTrue(client.isConnected());

  Interface stub = client.getStub();
  stub.echo(future.getController(), echoMessage, future);

  assertFalse(future.isDone());
  assertEquals(echoMessage, future.get());
  assertTrue(future.isDone());
  client.close();
  server.shutdown(true);
}