Remove Netty 5 support
Closes gh-34345
This commit is contained in:
@@ -1,5 +1,5 @@
|
||||
/*
|
||||
* Copyright 2002-2024 the original author or authors.
|
||||
* Copyright 2002-2025 the original author or authors.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
@@ -22,31 +22,17 @@ import org.jspecify.annotations.Nullable;
|
||||
|
||||
import org.springframework.messaging.simp.SimpLogging;
|
||||
import org.springframework.messaging.tcp.TcpOperations;
|
||||
import org.springframework.messaging.tcp.reactor.ReactorNetty2TcpClient;
|
||||
import org.springframework.messaging.tcp.reactor.ReactorNettyTcpClient;
|
||||
import org.springframework.util.Assert;
|
||||
import org.springframework.util.ClassUtils;
|
||||
|
||||
/**
|
||||
* A STOMP over TCP client, configurable with either
|
||||
* {@link ReactorNettyTcpClient} or {@link ReactorNetty2TcpClient}.
|
||||
* A STOMP over TCP client, configurable with {@link ReactorNettyTcpClient}.
|
||||
*
|
||||
* @author Rossen Stoyanchev
|
||||
* @since 5.0
|
||||
*/
|
||||
public class ReactorNettyTcpStompClient extends StompClientSupport {
|
||||
|
||||
private static final boolean reactorNettyClientPresent;
|
||||
|
||||
private static final boolean reactorNetty2ClientPresent;
|
||||
|
||||
static {
|
||||
ClassLoader classLoader = StompBrokerRelayMessageHandler.class.getClassLoader();
|
||||
reactorNettyClientPresent = ClassUtils.isPresent("reactor.netty.http.client.HttpClient", classLoader);
|
||||
reactorNetty2ClientPresent = ClassUtils.isPresent("reactor.netty5.http.client.HttpClient", classLoader);
|
||||
}
|
||||
|
||||
|
||||
private final TcpOperations<byte[]> tcpClient;
|
||||
|
||||
|
||||
@@ -76,17 +62,9 @@ public class ReactorNettyTcpStompClient extends StompClientSupport {
|
||||
}
|
||||
|
||||
private static TcpOperations<byte[]> initTcpClient(String host, int port) {
|
||||
if (reactorNettyClientPresent) {
|
||||
ReactorNettyTcpClient<byte[]> client = new ReactorNettyTcpClient<>(host, port, new StompReactorNettyCodec());
|
||||
client.setLogger(SimpLogging.forLog(client.getLogger()));
|
||||
return client;
|
||||
}
|
||||
else if (reactorNetty2ClientPresent) {
|
||||
ReactorNetty2TcpClient<byte[]> client = new ReactorNetty2TcpClient<>(host, port, new StompTcpMessageCodec());
|
||||
client.setLogger(SimpLogging.forLog(client.getLogger()));
|
||||
return client;
|
||||
}
|
||||
throw new IllegalStateException("No compatible version of Reactor Netty");
|
||||
ReactorNettyTcpClient<byte[]> client = new ReactorNettyTcpClient<>(host, port, new StompReactorNettyCodec());
|
||||
client.setLogger(SimpLogging.forLog(client.getLogger()));
|
||||
return client;
|
||||
}
|
||||
|
||||
|
||||
|
||||
@@ -1,5 +1,5 @@
|
||||
/*
|
||||
* Copyright 2002-2023 the original author or authors.
|
||||
* Copyright 2002-2025 the original author or authors.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
@@ -44,13 +44,10 @@ import org.springframework.messaging.support.MessageHeaderInitializer;
|
||||
import org.springframework.messaging.tcp.FixedIntervalReconnectStrategy;
|
||||
import org.springframework.messaging.tcp.TcpConnection;
|
||||
import org.springframework.messaging.tcp.TcpOperations;
|
||||
import org.springframework.messaging.tcp.reactor.ReactorNetty2TcpClient;
|
||||
import org.springframework.messaging.tcp.reactor.ReactorNettyCodec;
|
||||
import org.springframework.messaging.tcp.reactor.ReactorNettyTcpClient;
|
||||
import org.springframework.messaging.tcp.reactor.TcpMessageCodec;
|
||||
import org.springframework.scheduling.TaskScheduler;
|
||||
import org.springframework.util.Assert;
|
||||
import org.springframework.util.ClassUtils;
|
||||
|
||||
/**
|
||||
* A {@link org.springframework.messaging.MessageHandler} that handles messages by
|
||||
@@ -105,18 +102,10 @@ public class StompBrokerRelayMessageHandler extends AbstractBrokerMessageHandler
|
||||
|
||||
private static final Message<byte[]> HEARTBEAT_MESSAGE;
|
||||
|
||||
private static final boolean reactorNettyClientPresent;
|
||||
|
||||
private static final boolean reactorNetty2ClientPresent;
|
||||
|
||||
static {
|
||||
HEART_BEAT_ACCESSOR = StompHeaderAccessor.createForHeartbeat();
|
||||
HEARTBEAT_MESSAGE = MessageBuilder.createMessage(
|
||||
StompDecoder.HEARTBEAT_PAYLOAD, HEART_BEAT_ACCESSOR.getMessageHeaders());
|
||||
|
||||
ClassLoader classLoader = StompBrokerRelayMessageHandler.class.getClassLoader();
|
||||
reactorNettyClientPresent = ClassUtils.isPresent("reactor.netty.http.client.HttpClient", classLoader);
|
||||
reactorNetty2ClientPresent = ClassUtils.isPresent("reactor.netty5.http.client.HttpClient", classLoader);
|
||||
}
|
||||
|
||||
|
||||
@@ -352,8 +341,7 @@ public class StompBrokerRelayMessageHandler extends AbstractBrokerMessageHandler
|
||||
|
||||
/**
|
||||
* Configure a TCP client for managing TCP connections to the STOMP broker.
|
||||
* <p>By default {@link ReactorNettyTcpClient} or
|
||||
* {@link ReactorNetty2TcpClient} is used.
|
||||
* <p>By default {@link ReactorNettyTcpClient} is used.
|
||||
* <p><strong>Note:</strong> when this property is used, any
|
||||
* {@link #setRelayHost(String) host} or {@link #setRelayPort(int) port}
|
||||
* specified are effectively ignored.
|
||||
@@ -468,19 +456,10 @@ public class StompBrokerRelayMessageHandler extends AbstractBrokerMessageHandler
|
||||
if (this.headerInitializer != null) {
|
||||
decoder.setHeaderInitializer(this.headerInitializer);
|
||||
}
|
||||
if (reactorNettyClientPresent) {
|
||||
ReactorNettyCodec<byte[]> codec = new StompReactorNettyCodec(decoder);
|
||||
ReactorNettyTcpClient<byte[]> client = new ReactorNettyTcpClient<>(this.relayHost, this.relayPort, codec);
|
||||
client.setLogger(SimpLogging.forLog(client.getLogger()));
|
||||
return client;
|
||||
}
|
||||
else if (reactorNetty2ClientPresent) {
|
||||
TcpMessageCodec<byte[]> codec = new StompTcpMessageCodec(decoder);
|
||||
ReactorNetty2TcpClient<byte[]> client = new ReactorNetty2TcpClient<>(this.relayHost, this.relayPort, codec);
|
||||
client.setLogger(SimpLogging.forLog(client.getLogger()));
|
||||
return client;
|
||||
}
|
||||
throw new IllegalStateException("No compatible version of Reactor Netty");
|
||||
ReactorNettyCodec<byte[]> codec = new StompReactorNettyCodec(decoder);
|
||||
ReactorNettyTcpClient<byte[]> client = new ReactorNettyTcpClient<>(this.relayHost, this.relayPort, codec);
|
||||
client.setLogger(SimpLogging.forLog(client.getLogger()));
|
||||
return client;
|
||||
}
|
||||
|
||||
@Override
|
||||
|
||||
@@ -1,349 +0,0 @@
|
||||
/*
|
||||
* Copyright 2002-2023 the original author or authors.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package org.springframework.messaging.tcp.reactor;
|
||||
|
||||
import java.nio.ByteBuffer;
|
||||
import java.time.Duration;
|
||||
import java.util.List;
|
||||
import java.util.concurrent.CompletableFuture;
|
||||
import java.util.function.BiFunction;
|
||||
import java.util.function.Function;
|
||||
|
||||
import io.netty5.buffer.Buffer;
|
||||
import io.netty5.channel.ChannelHandlerContext;
|
||||
import io.netty5.channel.group.ChannelGroup;
|
||||
import io.netty5.channel.group.DefaultChannelGroup;
|
||||
import io.netty5.handler.codec.ByteToMessageDecoder;
|
||||
import io.netty5.util.concurrent.ImmediateEventExecutor;
|
||||
import org.apache.commons.logging.Log;
|
||||
import org.apache.commons.logging.LogFactory;
|
||||
import org.jspecify.annotations.Nullable;
|
||||
import org.reactivestreams.Publisher;
|
||||
import reactor.core.publisher.Mono;
|
||||
import reactor.core.publisher.Sinks;
|
||||
import reactor.core.scheduler.Scheduler;
|
||||
import reactor.core.scheduler.Schedulers;
|
||||
import reactor.netty5.Connection;
|
||||
import reactor.netty5.NettyInbound;
|
||||
import reactor.netty5.NettyOutbound;
|
||||
import reactor.netty5.resources.ConnectionProvider;
|
||||
import reactor.netty5.resources.LoopResources;
|
||||
import reactor.netty5.tcp.TcpClient;
|
||||
import reactor.util.retry.Retry;
|
||||
|
||||
import org.springframework.messaging.Message;
|
||||
import org.springframework.messaging.tcp.ReconnectStrategy;
|
||||
import org.springframework.messaging.tcp.TcpConnection;
|
||||
import org.springframework.messaging.tcp.TcpConnectionHandler;
|
||||
import org.springframework.messaging.tcp.TcpOperations;
|
||||
import org.springframework.util.Assert;
|
||||
|
||||
/**
|
||||
* Reactor Netty based implementation of {@link TcpOperations}.
|
||||
*
|
||||
* <p>This class is based on {@link ReactorNettyTcpClient}.
|
||||
*
|
||||
* @author Rossen Stoyanchev
|
||||
* @since 6.0
|
||||
* @param <P> the type of payload for in and outbound messages
|
||||
*/
|
||||
public class ReactorNetty2TcpClient<P> implements TcpOperations<P> {
|
||||
|
||||
private static final int PUBLISH_ON_BUFFER_SIZE = 16;
|
||||
|
||||
|
||||
private final TcpClient tcpClient;
|
||||
|
||||
private final TcpMessageCodec<P> codec;
|
||||
|
||||
private final @Nullable ChannelGroup channelGroup;
|
||||
|
||||
private final @Nullable LoopResources loopResources;
|
||||
|
||||
private final @Nullable ConnectionProvider poolResources;
|
||||
|
||||
private final Scheduler scheduler = Schedulers.newParallel("tcp-client-scheduler");
|
||||
|
||||
private Log logger = LogFactory.getLog(ReactorNetty2TcpClient.class);
|
||||
|
||||
private volatile boolean stopping;
|
||||
|
||||
|
||||
/**
|
||||
* Simple constructor with the host and port to use to connect to.
|
||||
* <p>This constructor manages the lifecycle of the {@link TcpClient} and
|
||||
* underlying resources such as {@link ConnectionProvider},
|
||||
* {@link LoopResources}, and {@link ChannelGroup}.
|
||||
* <p>For full control over the initialization and lifecycle of the
|
||||
* TcpClient, use {@link #ReactorNetty2TcpClient(TcpClient, TcpMessageCodec)}.
|
||||
* @param host the host to connect to
|
||||
* @param port the port to connect to
|
||||
* @param codec for encoding and decoding the input/output byte streams
|
||||
* @see org.springframework.messaging.simp.stomp.StompReactorNettyCodec
|
||||
*/
|
||||
public ReactorNetty2TcpClient(String host, int port, TcpMessageCodec<P> codec) {
|
||||
Assert.notNull(host, "host is required");
|
||||
Assert.notNull(codec, "ReactorNettyCodec is required");
|
||||
|
||||
this.channelGroup = new DefaultChannelGroup(ImmediateEventExecutor.INSTANCE);
|
||||
this.loopResources = LoopResources.create("tcp-client-loop");
|
||||
this.poolResources = ConnectionProvider.create("tcp-client-pool", 10000);
|
||||
this.codec = codec;
|
||||
|
||||
this.tcpClient = TcpClient.create(this.poolResources)
|
||||
.host(host).port(port)
|
||||
.runOn(this.loopResources, false)
|
||||
.doOnConnected(conn -> this.channelGroup.add(conn.channel()));
|
||||
}
|
||||
|
||||
/**
|
||||
* A variant of {@link #ReactorNetty2TcpClient(String, int, TcpMessageCodec)}
|
||||
* that still manages the lifecycle of the {@link TcpClient} and underlying
|
||||
* resources, but allows for direct configuration of other properties of the
|
||||
* client through a {@code Function<TcpClient, TcpClient>}.
|
||||
* @param clientConfigurer the configurer function
|
||||
* @param codec for encoding and decoding the input/output byte streams
|
||||
* @since 5.1.3
|
||||
* @see org.springframework.messaging.simp.stomp.StompReactorNettyCodec
|
||||
*/
|
||||
public ReactorNetty2TcpClient(Function<TcpClient, TcpClient> clientConfigurer, TcpMessageCodec<P> codec) {
|
||||
Assert.notNull(codec, "ReactorNettyCodec is required");
|
||||
|
||||
this.channelGroup = new DefaultChannelGroup(ImmediateEventExecutor.INSTANCE);
|
||||
this.loopResources = LoopResources.create("tcp-client-loop");
|
||||
this.poolResources = ConnectionProvider.create("tcp-client-pool", 10000);
|
||||
this.codec = codec;
|
||||
|
||||
this.tcpClient = clientConfigurer.apply(TcpClient
|
||||
.create(this.poolResources)
|
||||
.runOn(this.loopResources, false)
|
||||
.doOnConnected(conn -> this.channelGroup.add(conn.channel())));
|
||||
}
|
||||
|
||||
/**
|
||||
* Constructor with an externally created {@link TcpClient} instance whose
|
||||
* lifecycle is expected to be managed externally.
|
||||
* @param tcpClient the TcpClient instance to use
|
||||
* @param codec for encoding and decoding the input/output byte streams
|
||||
* @see org.springframework.messaging.simp.stomp.StompReactorNettyCodec
|
||||
*/
|
||||
public ReactorNetty2TcpClient(TcpClient tcpClient, TcpMessageCodec<P> codec) {
|
||||
Assert.notNull(tcpClient, "TcpClient is required");
|
||||
Assert.notNull(codec, "ReactorNettyCodec is required");
|
||||
this.tcpClient = tcpClient;
|
||||
this.codec = codec;
|
||||
|
||||
this.channelGroup = null;
|
||||
this.loopResources = null;
|
||||
this.poolResources = null;
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* Set an alternative logger to use than the one based on the class name.
|
||||
* @param logger the logger to use
|
||||
* @since 5.1
|
||||
*/
|
||||
public void setLogger(Log logger) {
|
||||
this.logger = logger;
|
||||
}
|
||||
|
||||
/**
|
||||
* Return the currently configured Logger.
|
||||
* @since 5.1
|
||||
*/
|
||||
public Log getLogger() {
|
||||
return logger;
|
||||
}
|
||||
|
||||
|
||||
@Override
|
||||
public CompletableFuture<Void> connectAsync(TcpConnectionHandler<P> handler) {
|
||||
Assert.notNull(handler, "TcpConnectionHandler is required");
|
||||
|
||||
if (this.stopping) {
|
||||
return handleShuttingDownConnectFailure(handler);
|
||||
}
|
||||
|
||||
return extendTcpClient(this.tcpClient, handler)
|
||||
.handle(new ReactorNettyHandler(handler))
|
||||
.connect()
|
||||
.doOnError(handler::afterConnectFailure)
|
||||
.then().toFuture();
|
||||
}
|
||||
|
||||
/**
|
||||
* Provides an opportunity to initialize the {@link TcpClient} for the given
|
||||
* {@link TcpConnectionHandler} which may implement sub-interfaces such as
|
||||
* {@link org.springframework.messaging.simp.stomp.StompTcpConnectionHandler}
|
||||
* that expose further information.
|
||||
* @param tcpClient the candidate TcpClient
|
||||
* @param handler the handler for the TCP connection
|
||||
* @return the same handler or an updated instance
|
||||
*/
|
||||
protected TcpClient extendTcpClient(TcpClient tcpClient, TcpConnectionHandler<P> handler) {
|
||||
return tcpClient;
|
||||
}
|
||||
|
||||
@Override
|
||||
public CompletableFuture<Void> connectAsync(TcpConnectionHandler<P> handler, ReconnectStrategy strategy) {
|
||||
Assert.notNull(handler, "TcpConnectionHandler is required");
|
||||
Assert.notNull(strategy, "ReconnectStrategy is required");
|
||||
|
||||
if (this.stopping) {
|
||||
return handleShuttingDownConnectFailure(handler);
|
||||
}
|
||||
|
||||
// Report first connect to the ListenableFuture
|
||||
CompletableFuture<Void> connectFuture = new CompletableFuture<>();
|
||||
|
||||
extendTcpClient(this.tcpClient, handler)
|
||||
.handle(new ReactorNettyHandler(handler))
|
||||
.connect()
|
||||
.doOnNext(conn -> connectFuture.complete(null))
|
||||
.doOnError(connectFuture::completeExceptionally)
|
||||
.doOnError(handler::afterConnectFailure) // report all connect failures to the handler
|
||||
.flatMap(Connection::onDispose) // post-connect issues
|
||||
.retryWhen(Retry.from(signals -> signals
|
||||
.map(retrySignal -> (int) retrySignal.totalRetriesInARow())
|
||||
.flatMap(attempt -> reconnect(attempt, strategy))))
|
||||
.repeatWhen(flux -> flux
|
||||
.scan(1, (count, element) -> count++)
|
||||
.flatMap(attempt -> reconnect(attempt, strategy)))
|
||||
.subscribe();
|
||||
return connectFuture;
|
||||
}
|
||||
|
||||
private CompletableFuture<Void> handleShuttingDownConnectFailure(TcpConnectionHandler<P> handler) {
|
||||
IllegalStateException ex = new IllegalStateException("Shutting down.");
|
||||
handler.afterConnectFailure(ex);
|
||||
return Mono.<Void>error(ex).toFuture();
|
||||
}
|
||||
|
||||
private Publisher<? extends Long> reconnect(Integer attempt, ReconnectStrategy reconnectStrategy) {
|
||||
Long time = reconnectStrategy.getTimeToNextAttempt(attempt);
|
||||
return (time != null ? Mono.delay(Duration.ofMillis(time), this.scheduler) : Mono.empty());
|
||||
}
|
||||
|
||||
@Override
|
||||
public CompletableFuture<Void> shutdownAsync() {
|
||||
if (this.stopping) {
|
||||
return CompletableFuture.completedFuture(null);
|
||||
}
|
||||
|
||||
this.stopping = true;
|
||||
|
||||
Mono<Void> result;
|
||||
if (this.channelGroup != null) {
|
||||
Sinks.Empty<Void> channnelGroupCloseSink = Sinks.empty();
|
||||
this.channelGroup.close().addListener(future -> channnelGroupCloseSink.tryEmitEmpty());
|
||||
result = channnelGroupCloseSink.asMono();
|
||||
if (this.loopResources != null) {
|
||||
result = result.onErrorComplete().then(this.loopResources.disposeLater());
|
||||
}
|
||||
if (this.poolResources != null) {
|
||||
result = result.onErrorComplete().then(this.poolResources.disposeLater());
|
||||
}
|
||||
result = result.onErrorComplete().then(stopScheduler());
|
||||
}
|
||||
else {
|
||||
result = stopScheduler();
|
||||
}
|
||||
|
||||
return result.toFuture();
|
||||
}
|
||||
|
||||
private Mono<Void> stopScheduler() {
|
||||
return Mono.fromRunnable(() -> {
|
||||
this.scheduler.dispose();
|
||||
for (int i = 0; i < 20; i++) {
|
||||
if (this.scheduler.isDisposed()) {
|
||||
break;
|
||||
}
|
||||
try {
|
||||
Thread.sleep(100);
|
||||
}
|
||||
catch (Throwable ex) {
|
||||
break;
|
||||
}
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
@Override
|
||||
public String toString() {
|
||||
return "ReactorNetty2TcpClient[" + this.tcpClient + "]";
|
||||
}
|
||||
|
||||
|
||||
private class ReactorNettyHandler implements BiFunction<NettyInbound, NettyOutbound, Publisher<Void>> {
|
||||
|
||||
private final TcpConnectionHandler<P> connectionHandler;
|
||||
|
||||
ReactorNettyHandler(TcpConnectionHandler<P> handler) {
|
||||
this.connectionHandler = handler;
|
||||
}
|
||||
|
||||
@Override
|
||||
@SuppressWarnings("unchecked")
|
||||
public Publisher<Void> apply(NettyInbound inbound, NettyOutbound outbound) {
|
||||
inbound.withConnection(conn -> {
|
||||
if (logger.isDebugEnabled()) {
|
||||
logger.debug("Connected to " + conn.address());
|
||||
}
|
||||
});
|
||||
Sinks.Empty<Void> completionSink = Sinks.empty();
|
||||
TcpConnection<P> connection = new ReactorNetty2TcpConnection<>(inbound, outbound, codec, completionSink);
|
||||
scheduler.schedule(() -> this.connectionHandler.afterConnected(connection));
|
||||
|
||||
inbound.withConnection(conn -> conn.addHandlerFirst(new StompMessageDecoder<>(codec)));
|
||||
|
||||
inbound.receiveObject()
|
||||
.cast(Message.class)
|
||||
.publishOn(scheduler, PUBLISH_ON_BUFFER_SIZE)
|
||||
.subscribe(
|
||||
this.connectionHandler::handleMessage,
|
||||
this.connectionHandler::handleFailure,
|
||||
this.connectionHandler::afterConnectionClosed);
|
||||
|
||||
return completionSink.asMono();
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
private static class StompMessageDecoder<P> extends ByteToMessageDecoder {
|
||||
|
||||
private final TcpMessageCodec<P> codec;
|
||||
|
||||
StompMessageDecoder(TcpMessageCodec<P> codec) {
|
||||
this.codec = codec;
|
||||
}
|
||||
|
||||
@Override
|
||||
protected void decode(ChannelHandlerContext ctx, Buffer buffer) throws Exception {
|
||||
ByteBuffer byteBuffer = ByteBuffer.allocate(buffer.readableBytes());
|
||||
buffer.readBytes(byteBuffer);
|
||||
byteBuffer.flip();
|
||||
List<Message<P>> messages = this.codec.decode(byteBuffer);
|
||||
for (Message<P> message : messages) {
|
||||
ctx.fireChannelRead(message);
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
@@ -1,88 +0,0 @@
|
||||
/*
|
||||
* Copyright 2002-2022 the original author or authors.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package org.springframework.messaging.tcp.reactor;
|
||||
|
||||
import java.nio.ByteBuffer;
|
||||
import java.util.concurrent.CompletableFuture;
|
||||
|
||||
import io.netty5.buffer.Buffer;
|
||||
import reactor.core.publisher.Mono;
|
||||
import reactor.core.publisher.Sinks;
|
||||
import reactor.netty5.NettyInbound;
|
||||
import reactor.netty5.NettyOutbound;
|
||||
|
||||
import org.springframework.messaging.Message;
|
||||
import org.springframework.messaging.tcp.TcpConnection;
|
||||
|
||||
/**
|
||||
* Reactor Netty based implementation of {@link TcpConnection}.
|
||||
*
|
||||
* <p>This class is based on {@link ReactorNettyTcpConnection}.
|
||||
*
|
||||
* @author Rossen Stoyanchev
|
||||
* @since 6.0
|
||||
* @param <P> the type of payload for outbound messages
|
||||
*/
|
||||
public class ReactorNetty2TcpConnection<P> implements TcpConnection<P> {
|
||||
|
||||
private final NettyInbound inbound;
|
||||
|
||||
private final NettyOutbound outbound;
|
||||
|
||||
private final TcpMessageCodec<P> codec;
|
||||
|
||||
private final Sinks.Empty<Void> completionSink;
|
||||
|
||||
|
||||
public ReactorNetty2TcpConnection(NettyInbound inbound, NettyOutbound outbound,
|
||||
TcpMessageCodec<P> codec, Sinks.Empty<Void> completionSink) {
|
||||
|
||||
this.inbound = inbound;
|
||||
this.outbound = outbound;
|
||||
this.codec = codec;
|
||||
this.completionSink = completionSink;
|
||||
}
|
||||
|
||||
|
||||
@Override
|
||||
public CompletableFuture<Void> sendAsync(Message<P> message) {
|
||||
ByteBuffer byteBuffer = this.codec.encode(message);
|
||||
Buffer buffer = this.outbound.alloc().copyOf(byteBuffer);
|
||||
return this.outbound.send(Mono.just(buffer)).then().toFuture();
|
||||
}
|
||||
|
||||
@Override
|
||||
public void onReadInactivity(Runnable runnable, long inactivityDuration) {
|
||||
this.inbound.withConnection(conn -> conn.onReadIdle(inactivityDuration, runnable));
|
||||
}
|
||||
|
||||
@Override
|
||||
public void onWriteInactivity(Runnable runnable, long inactivityDuration) {
|
||||
this.inbound.withConnection(conn -> conn.onWriteIdle(inactivityDuration, runnable));
|
||||
}
|
||||
|
||||
@Override
|
||||
public void close() {
|
||||
// Ignore result: concurrent attempts to complete are ok
|
||||
this.completionSink.tryEmitEmpty();
|
||||
}
|
||||
|
||||
@Override
|
||||
public String toString() {
|
||||
return "ReactorNetty2TcpConnection[inbound=" + this.inbound + "]";
|
||||
}
|
||||
}
|
||||
@@ -1,38 +0,0 @@
|
||||
/*
|
||||
* Copyright 2002-2023 the original author or authors.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package org.springframework.messaging.simp.stomp;
|
||||
|
||||
import org.junit.jupiter.api.Disabled;
|
||||
|
||||
import org.springframework.messaging.tcp.TcpOperations;
|
||||
import org.springframework.messaging.tcp.reactor.ReactorNetty2TcpClient;
|
||||
|
||||
/**
|
||||
* Integration tests for {@link StompBrokerRelayMessageHandler} running against
|
||||
* ActiveMQ with {@link ReactorNetty2TcpClient}.
|
||||
*
|
||||
* @author Rossen Stoyanchev
|
||||
*/
|
||||
@Disabled("gh-29287 :: Disabled because they fail too frequently")
|
||||
public class ReactorNetty2StompBrokerRelayIntegrationTests extends AbstractStompBrokerRelayIntegrationTests {
|
||||
|
||||
@Override
|
||||
protected TcpOperations<byte[]> initTcpClient(int port) {
|
||||
return new ReactorNetty2TcpClient<>("127.0.0.1", port, new StompTcpMessageCodec());
|
||||
}
|
||||
|
||||
}
|
||||
@@ -1,5 +1,5 @@
|
||||
/*
|
||||
* Copyright 2002-2024 the original author or authors.
|
||||
* Copyright 2002-2025 the original author or authors.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
@@ -37,7 +37,6 @@ import org.junit.jupiter.api.TestInfo;
|
||||
|
||||
import org.springframework.messaging.converter.StringMessageConverter;
|
||||
import org.springframework.messaging.simp.stomp.StompSession.Subscription;
|
||||
import org.springframework.messaging.tcp.reactor.ReactorNetty2TcpClient;
|
||||
import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler;
|
||||
import org.springframework.util.Assert;
|
||||
|
||||
@@ -58,8 +57,6 @@ class ReactorNettyTcpStompClientTests {
|
||||
|
||||
private ReactorNettyTcpStompClient client;
|
||||
|
||||
private ReactorNettyTcpStompClient client2;
|
||||
|
||||
|
||||
@BeforeEach
|
||||
void setup(TestInfo testInfo) throws Exception {
|
||||
@@ -84,10 +81,6 @@ class ReactorNettyTcpStompClientTests {
|
||||
this.client = new ReactorNettyTcpStompClient(host, port);
|
||||
this.client.setMessageConverter(new StringMessageConverter());
|
||||
this.client.setTaskScheduler(taskScheduler);
|
||||
|
||||
this.client2 = new ReactorNettyTcpStompClient(new ReactorNetty2TcpClient<>(host, port, new StompTcpMessageCodec()));
|
||||
this.client2.setMessageConverter(new StringMessageConverter());
|
||||
this.client2.setTaskScheduler(taskScheduler);
|
||||
}
|
||||
|
||||
private TransportConnector createStompConnector() throws Exception {
|
||||
@@ -100,7 +93,6 @@ class ReactorNettyTcpStompClientTests {
|
||||
void shutdown() throws Exception {
|
||||
try {
|
||||
this.client.shutdown();
|
||||
this.client2.shutdown();
|
||||
}
|
||||
catch (Throwable ex) {
|
||||
logger.error("Failed to shut client", ex);
|
||||
@@ -120,11 +112,6 @@ class ReactorNettyTcpStompClientTests {
|
||||
testPublishSubscribe(this.client);
|
||||
}
|
||||
|
||||
@Test
|
||||
void publishSubscribeOnReactorNetty2() throws Exception {
|
||||
testPublishSubscribe(this.client2);
|
||||
}
|
||||
|
||||
private void testPublishSubscribe(ReactorNettyTcpStompClient clientToUse) throws Exception {
|
||||
String destination = "/topic/foo";
|
||||
ConsumingHandler consumingHandler1 = new ConsumingHandler(destination);
|
||||
|
||||
Reference in New Issue
Block a user