|
此版本仍在开发中,尚未被视为稳定版。如需最新稳定版本,请使用 Spring Integration 7.0.4! |
测试连接
在某些场景下,当首次建立连接时发送某种健康检查请求可能非常有用。 一个这样的场景可能是使用 TCP 故障转移客户端连接工厂,以便在选定的服务器允许建立连接但报告其不健康时进行故障转移。
为了支持此功能,请向客户端连接工厂添加一个 connectionTest。
/**
* 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 故障转移客户端连接工厂 一起使用时,这将触发尝试下一个服务器。
| 只有服务器的第一条回复会发送到测试监听器。 |
在以下示例中,如果我们发送 PING,服务器回复 PONG 时,则认为服务器是健康的。
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();
});