测试连接

在某些场景下，当首次建立连接时发送某种健康检查请求可能非常有用。一个这样的场景可能是使用 TCP 故障转移客户端连接工厂，以便在选定的服务器允许建立连接但报告其不健康时进行故障转移。spring-doc.cadn.net.cn

为了支持此功能，请向客户端连接工厂添加一个 connectionTest。spring-doc.cadn.net.cn

/**
 * Set a {@link Predicate} that will be invoked to test a new connection; return true
 * to accept the connection, false the reject.
 * @param connectionTest the predicate.
 * @since 5.3
 */
public void setConnectionTest(@Nullable Predicate<TcpConnectionSupport> connectionTest) {
    this.connectionTest = connectionTest;
}

为了测试连接，请在测试中为连接附加一个临时监听器。如果测试失败，连接将被关闭并抛出异常。当与 TCP 故障转移客户端连接工厂一起使用时，这将触发尝试下一个服务器。spring-doc.cadn.net.cn

只有服务器的第一条回复会发送到测试监听器。

在以下示例中，如果我们发送 PING，服务器回复 PONG 时，则认为服务器是健康的。spring-doc.cadn.net.cn

Message<String> ping = new GenericMessage<>("PING");
byte[] pong = "PONG".getBytes();
clientFactory.setConnectionTest(conn -> {
    CountDownLatch latch = new CountDownLatch(1);
    AtomicBoolean result = new AtomicBoolean();
    conn.registerTestListener(msg -> {
        if (Arrays.equals(pong, (byte[]) msg.getPayload())) {
            result.set(true);
        }
        latch.countDown();
        return false;
    });
    conn.send(ping);
    try {
        latch.await(10, TimeUnit.SECONDS);
    }
    catch (InterruptedException e) {
        Thread.currentThread().interrupt();
    }
    return result.get();
});