Spring/spring-integration
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

GH-9061: renew connection in PostgresChannelMessageTableSubscriber

Browse Source 
Fixes: #9061

`PostgresChannelMessageTableSubscriber` never renews the connection.
This causes problems on DB failover.
With this change the connection is renewed when notifications are not received for a certain time.

(cherry picked from commit 642278dad1)
This commit is contained in:
Johannes Edmeier
2024-04-01 19:05:56 +02:00
committed by Spring Builds
parent 226bca3181
commit fa6494f28f
2 changed files with 77 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
spring-integration-jdbc/src/test/java/org/springframework/integration/jdbc/channel/PostgresChannelMessageTableSubscriberTests.java
View File

@@ -17,6 +17,8 @@
package org.springframework.integration.jdbc.channel;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.time.Duration;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.CountDownLatch;
@@ -62,6 +64,7 @@ import static org.assertj.core.api.Assertions.assertThat;
* @author Artem Bilan
* @author Igor Lovich
* @author Adama Sorho
* @author Johannes Edmeier
*
* @since 6.0
*/
@@ -102,15 +105,14 @@ public class PostgresChannelMessageTableSubscriberTests implements PostgresConta
private String groupId;
private ConnectionSupplier connectionSupplier;
@BeforeEach
void setUp(TestInfo testInfo) {
// Not initiated as a bean to allow for registrations prior and post the life cycle
this.postgresChannelMessageTableSubscriber =
new PostgresChannelMessageTableSubscriber(() ->
DriverManager.getConnection(POSTGRES_CONTAINER.getJdbcUrl(),
POSTGRES_CONTAINER.getUsername(),
POSTGRES_CONTAINER.getPassword())
.unwrap(PgConnection.class));
this.connectionSupplier = new ConnectionSupplier();
this.postgresChannelMessageTableSubscriber = new PostgresChannelMessageTableSubscriber(connectionSupplier);
this.postgresChannelMessageTableSubscriber.setNotificationTimeout(Duration.ofSeconds(5));
this.taskExecutor = new ThreadPoolTaskExecutor();
@@ -263,6 +265,26 @@ public class PostgresChannelMessageTableSubscriberTests implements PostgresConta
assertThat(payloads).containsExactly("1");
}
@Test
public void testRenewConnection() throws Exception {
CountDownLatch latch = new CountDownLatch(2);
List<Object> payloads = new ArrayList<>();
CountDownLatch connectionLatch = new CountDownLatch(2);
connectionSupplier.onGetConnection = connectionLatch::countDown;
postgresChannelMessageTableSubscriber.start();
postgresSubscribableChannel.subscribe(message -> {
payloads.add(message.getPayload());
latch.countDown();
});
assertThat(connectionLatch.await(10, TimeUnit.SECONDS)).isTrue();
messageStore.addMessageToGroup(groupId, new GenericMessage<>("1"));
messageStore.addMessageToGroup(groupId, new GenericMessage<>("2"));
assertThat(latch.await(10, TimeUnit.SECONDS)).isTrue();
assertThat(payloads).containsExactlyInAnyOrder("1", "2");
}
@Configuration
@EnableIntegration
public static class Config {
@@ -302,4 +324,21 @@ public class PostgresChannelMessageTableSubscriberTests implements PostgresConta
}
private static class ConnectionSupplier implements PgConnectionSupplier {
Runnable onGetConnection;
@Override
public PgConnection get() throws SQLException {
var conn = DriverManager.getConnection(POSTGRES_CONTAINER.getJdbcUrl(),
POSTGRES_CONTAINER.getUsername(),
POSTGRES_CONTAINER.getPassword())
.unwrap(PgConnection.class);
if (this.onGetConnection != null) {
this.onGetConnection.run();
}
return conn;
}
}
}