Parse redis database from url if present

Fix GH-43810

Signed-off-by: Yanming Zhou <[email protected]>

Author: quaff

spring-boot-project/spring-boot-autoconfigure/src/main/java/org/springframework/boot/autoconfigure/data/redis/PropertiesRedisConnectionDetails.java

@@ -27,6 +27,7 @@
  * @author Andy Wilkinson
  * @author Phillip Webb
  * @author Scott Frederick
+ * @author Yanming Zhou
  */
 class PropertiesRedisConnectionDetails implements RedisConnectionDetails {
 
@@ -59,7 +60,8 @@ public Standalone getStandalone() {
 		if (this.properties.getUrl() != null) {
 			ConnectionInfo connectionInfo = ConnectionInfo.of(this.properties.getUrl());
 			return Standalone.of(connectionInfo.getUri().getHost(), connectionInfo.getUri().getPort(),
-					this.properties.getDatabase());
+					(connectionInfo.getDatabase() != null) ? connectionInfo.getDatabase()
+							: this.properties.getDatabase());
 		}
 		return Standalone.of(this.properties.getHost(), this.properties.getPort(), this.properties.getDatabase());
 	}

spring-boot-project/spring-boot-autoconfigure/src/main/java/org/springframework/boot/autoconfigure/data/redis/RedisConnectionConfiguration.java

@@ -33,6 +33,7 @@
 import org.springframework.data.redis.connection.RedisSentinelConfiguration;
 import org.springframework.data.redis.connection.RedisStandaloneConfiguration;
 import org.springframework.util.ClassUtils;
+import org.springframework.util.StringUtils;
 
 /**
  * Base Redis connection configuration.
@@ -45,6 +46,7 @@
  * @author Moritz Halbritter
  * @author Andy Wilkinson
  * @author Phillip Webb
+ * @author Yanming Zhou
  */
 abstract class RedisConnectionConfiguration {
 
@@ -189,11 +191,14 @@ static final class ConnectionInfo {
 
 		private final String password;
 
-		private ConnectionInfo(URI uri, boolean useSsl, String username, String password) {
+		private final Integer database;
+
+		private ConnectionInfo(URI uri, boolean useSsl, String username, String password, Integer database) {
 			this.uri = uri;
 			this.useSsl = useSsl;
 			this.username = username;
 			this.password = password;
+			this.database = database;
 		}
 
 		URI getUri() {
@@ -212,6 +217,10 @@ String getPassword() {
 			return this.password;
 		}
 
+		Integer getDatabase() {
+			return this.database;
+		}
+
 		static ConnectionInfo of(String url) {
 			try {
 				URI uri = new URI(url);
@@ -233,7 +242,14 @@ static ConnectionInfo of(String url) {
 						password = candidate;
 					}
 				}
-				return new ConnectionInfo(uri, useSsl, username, password);
+				Integer database = null;
+				if (StringUtils.hasText(uri.getPath())) {
+					String[] pathSplit = uri.getPath().split("/", 2);
+					if (pathSplit.length > 1 && !pathSplit[1].isEmpty()) {
+						database = Integer.parseInt(pathSplit[1]);
+					}
+				}
+				return new ConnectionInfo(uri, useSsl, username, password, database);
 			}
 			catch (URISyntaxException ex) {
 				throw new RedisUrlSyntaxException(url, ex);

spring-boot-project/spring-boot-autoconfigure/src/main/java/org/springframework/boot/autoconfigure/data/redis/RedisProperties.java

@@ -31,6 +31,7 @@
  * @author Mark Paluch
  * @author Stephane Nicoll
  * @author Scott Frederick
+ * @author Yanming Zhou
  * @since 1.0.0
  */
 @ConfigurationProperties(prefix = "spring.data.redis")
@@ -42,8 +43,8 @@ public class RedisProperties {
 	private int database = 0;
 
 	/**
-	 * Connection URL. Overrides host, port, username, and password. Example:
-	 * redis://user:[email protected]:6379
+	 * Connection URL. Overrides host, port, username, password, and database. Example:
+	 * redis://user:[email protected]:6379/8
 	 */
 	private String url;
 

spring-boot-project/spring-boot-autoconfigure/src/test/java/org/springframework/boot/autoconfigure/data/redis/PropertiesRedisConnectionDetailsTests.java

@@ -103,6 +103,17 @@ void standaloneIsConfiguredFromUrl() {
 		RedisConnectionDetails.Standalone standalone = connectionDetails.getStandalone();
 		assertThat(standalone.getHost()).isEqualTo("example.com");
 		assertThat(standalone.getPort()).isEqualTo(1234);
+		assertThat(standalone.getDatabase()).isEqualTo(9999);
+	}
+
+	@Test
+	void standaloneIsConfiguredFromUrlWithoutDatabase() {
+		this.properties.setUrl("redis://example.com:1234");
+		this.properties.setDatabase(5);
+		PropertiesRedisConnectionDetails connectionDetails = new PropertiesRedisConnectionDetails(this.properties);
+		RedisConnectionDetails.Standalone standalone = connectionDetails.getStandalone();
+		assertThat(standalone.getHost()).isEqualTo("example.com");
+		assertThat(standalone.getPort()).isEqualTo(1234);
 		assertThat(standalone.getDatabase()).isEqualTo(5);
 	}