Polishing.

Add author and since tags. Wrap connection allocation with fromSupplier(…). Move connection proxy decoration into doInConnection(…) as connection decoration isn't directly related to connection creation.

See #2145
Original pull request: #2162.

Author: mp911de

Diff for: src/main/java/org/springframework/data/redis/core/ReactiveRedisTemplate.java

@@ -58,6 +58,7 @@
  *
  * @author Mark Paluch
  * @author Christoph Strobl
+ * @author Petromir Dzhunev
  * @since 2.0
  * @param <K> the Redis key type against which the template works (usually a String)
  * @param <V> the Redis value type against which the template works
@@ -192,7 +193,14 @@ private <T> Publisher<T> doInConnection(ReactiveRedisCallback<T> action, boolean
 
 		Assert.notNull(action, "Callback object must not be null");
 
-		return Flux.usingWhen(getConnection(exposeConnection), conn -> {
+		Mono<ReactiveRedisConnection> connection = getConnection();
+
+		if (!exposeConnection) {
+			connection = connection.map(this::createRedisConnectionProxy);
+		}
+
+		return Flux.usingWhen(connection, conn -> {
+
 			Publisher<T> result = action.doInRedis(conn);
 
 			return postProcessResult(result, conn, false);
@@ -201,18 +209,16 @@ private <T> Publisher<T> doInConnection(ReactiveRedisCallback<T> action, boolean
 	}
 
 	/**
-	 * Creates a Mono which generates a new connection. The successors of {@link ReactiveRedisTemplate} might override
-	 * the default behaviour.
+	 * Creates a {@link Mono} which emits a new {@link ReactiveRedisConnection}. Can be overridden in subclasses to
+	 * provide a different mechanism for connection allocation for the given method.
 	 *
-	 * @param exposeConnection whether to enforce exposure of the native Redis Connection to callback code
-	 * return a {@link Mono} wrapping the {@link ReactiveRedisConnection}.
+	 * @since 2.5.5
 	 */
-	protected Mono<ReactiveRedisConnection> getConnection(boolean exposeConnection) {
+	protected Mono<ReactiveRedisConnection> getConnection() {
+
 		ReactiveRedisConnectionFactory factory = getConnectionFactory();
-		ReactiveRedisConnection conn = factory.getReactiveConnection();
-		ReactiveRedisConnection connToUse = preProcessConnection(conn, false);
 
-		return Mono.just(exposeConnection ? connToUse : createRedisConnectionProxy(connToUse));
+		return Mono.fromSupplier(() -> preProcessConnection(factory.getReactiveConnection(), false));
 	}
 
 	/*