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

Extract connection creation into ReactiveRedisTemplate.getConnection() method.

Browse Source 
A new method, called getConnection, is introduced to allow overriding of connection creation.

Closes #2145
Original pull request: #2162.
This commit is contained in:
Petromir Dzhunev
2021-09-10 17:54:20 +03:00
committed by Mark Paluch
parent cb20f8a652
commit aaf7818427
1 changed files with 16 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
src/main/java/org/springframework/data/redis/core/ReactiveRedisTemplate.java
View File

@@ -232,14 +232,7 @@ public class ReactiveRedisTemplate<K, V> implements ReactiveRedisOperations<K, V
Assert.notNull(action, "Callback object must not be null");
return Flux.usingWhen(Mono.fromSupplier(() -> {
ReactiveRedisConnectionFactory factory = getConnectionFactory();
ReactiveRedisConnection conn = factory.getReactiveConnection();
ReactiveRedisConnection connToUse = preProcessConnection(conn, false);
return (exposeConnection ? connToUse : createRedisConnectionProxy(connToUse));
}), conn -> {
return Flux.usingWhen(getConnection(exposeConnection), conn -> {
Publisher<T> result = action.doInRedis(conn);
return postProcessResult(result, conn, false);
@@ -247,6 +240,21 @@ public class ReactiveRedisTemplate<K, V> implements ReactiveRedisOperations<K, V
}, ReactiveRedisConnection::closeLater);
}
/**
* Creates a Mono which generates a new connection. The successors of {@link ReactiveRedisTemplate} might override
* the default behaviour.
*
* @param exposeConnection whether to enforce exposure of the native Redis Connection to callback code
* return a {@link Mono} wrapping the {@link ReactiveRedisConnection}.
*/
protected Mono<ReactiveRedisConnection> getConnection(boolean exposeConnection) {
ReactiveRedisConnectionFactory factory = getConnectionFactory();
ReactiveRedisConnection conn = factory.getReactiveConnection();
ReactiveRedisConnection connToUse = preProcessConnection(conn, false);
return Mono.just(exposeConnection ? connToUse : createRedisConnectionProxy(connToUse));
}
/*
* (non-Javadoc)
* @see org.springframework.data.redis.core.ReactiveRedisOperations#convertAndSend(java.lang.String, java.lang.Object)