Skip to content

Commit 08b0099

Browse files
snicollsnicoll
committed
Merge pull request #23693 from somayaj
* pr/23693: Polish "Fix detection logic for embedded databases" Fix detection logic for embedded databases Closes gh-23693
2 parents 23073d9 + c0b267f commit 08b0099

File tree

5 files changed

+161
-18
lines changed

5 files changed

+161
-18
lines changed

‎spring-boot-project/spring-boot-autoconfigure/src/main/java/org/springframework/boot/autoconfigure/jdbc/DataSourceProperties.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -325,7 +325,7 @@ public String determineUsername() {
325325
if (StringUtils.hasText(this.username)) {
326326
return this.username;
327327
}
328-
if (EmbeddedDatabaseConnection.isEmbedded(determineDriverClassName())) {
328+
if (EmbeddedDatabaseConnection.isEmbedded(determineDriverClassName(), determineUrl())) {
329329
return "sa";
330330
}
331331
return null;
@@ -353,7 +353,7 @@ public String determinePassword() {
353353
if (StringUtils.hasText(this.password)) {
354354
return this.password;
355355
}
356-
if (EmbeddedDatabaseConnection.isEmbedded(determineDriverClassName())) {
356+
if (EmbeddedDatabaseConnection.isEmbedded(determineDriverClassName(), determineUrl())) {
357357
return "";
358358
}
359359
return null;

‎spring-boot-project/spring-boot-autoconfigure/src/test/java/org/springframework/boot/autoconfigure/jdbc/DataSourcePropertiesTests.java

Lines changed: 36 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -98,6 +98,24 @@ void determineUsername() throws Exception {
9898
assertThat(properties.determineUsername()).isEqualTo("sa");
9999
}
100100

101+
@Test
102+
void determineUsernameWhenEmpty() throws Exception {
103+
DataSourceProperties properties = new DataSourceProperties();
104+
properties.setUsername("");
105+
properties.afterPropertiesSet();
106+
assertThat(properties.getUsername());
107+
assertThat(properties.determineUsername()).isEqualTo("sa");
108+
}
109+
110+
@Test
111+
void determineUsernameWhenNull() throws Exception {
112+
DataSourceProperties properties = new DataSourceProperties();
113+
properties.setUsername(null);
114+
properties.afterPropertiesSet();
115+
assertThat(properties.getUsername());
116+
assertThat(properties.determineUsername()).isEqualTo("sa");
117+
}
118+
101119
@Test
102120
void determineUsernameWithExplicitConfig() throws Exception {
103121
DataSourceProperties properties = new DataSourceProperties();
@@ -107,6 +125,15 @@ void determineUsernameWithExplicitConfig() throws Exception {
107125
assertThat(properties.determineUsername()).isEqualTo("foo");
108126
}
109127

128+
@Test
129+
void determineUsernameWithNonEmbeddedUrl() throws Exception {
130+
DataSourceProperties properties = new DataSourceProperties();
131+
properties.setUrl("jdbc:h2:~/test");
132+
properties.afterPropertiesSet();
133+
assertThat(properties.getPassword()).isNull();
134+
assertThat(properties.determineUsername()).isNull();
135+
}
136+
110137
@Test
111138
void determinePassword() throws Exception {
112139
DataSourceProperties properties = new DataSourceProperties();
@@ -124,6 +151,15 @@ void determinePasswordWithExplicitConfig() throws Exception {
124151
assertThat(properties.determinePassword()).isEqualTo("bar");
125152
}
126153

154+
@Test
155+
void determinePasswordWithNonEmbeddedUrl() throws Exception {
156+
DataSourceProperties properties = new DataSourceProperties();
157+
properties.setUrl("jdbc:h2:~/test");
158+
properties.afterPropertiesSet();
159+
assertThat(properties.getPassword()).isNull();
160+
assertThat(properties.determinePassword()).isNull();
161+
}
162+
127163
@Test
128164
void determineCredentialsForSchemaScripts() {
129165
DataSourceProperties properties = new DataSourceProperties();

‎spring-boot-project/spring-boot-docs/src/docs/asciidoc/howto.adoc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1982,8 +1982,8 @@ This is controlled through two external properties:
19821982
You can set `spring.jpa.hibernate.ddl-auto` explicitly and the standard Hibernate property values are `none`, `validate`, `update`, `create`, and `create-drop`.
19831983
Spring Boot chooses a default value for you based on whether it thinks your database is embedded.
19841984
It defaults to `create-drop` if no schema manager has been detected or `none` in all other cases.
1985-
An embedded database is detected by looking at the `Connection` type.
1986-
`hsqldb`, `h2`, and `derby` are embedded, and others are not.
1985+
An embedded database is detected by looking at the `Connection` type and JDBC url.
1986+
`hsqldb`, `h2`, and `derby` are candidates, and others are not.
19871987
Be careful when switching from in-memory to a '`real`' database that you do not make assumptions about the existence of the tables and data in the new platform.
19881988
You either have to set `ddl-auto` explicitly or use one of the other mechanisms to initialize the database.
19891989

‎spring-boot-project/spring-boot/src/main/java/org/springframework/boot/jdbc/EmbeddedDatabaseConnection.java

Lines changed: 46 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,11 @@
1717
package org.springframework.boot.jdbc;
1818

1919
import java.sql.Connection;
20+
import java.sql.DatabaseMetaData;
2021
import java.sql.SQLException;
2122
import java.util.Locale;
23+
import java.util.function.Predicate;
24+
import java.util.stream.Stream;
2225

2326
import javax.sql.DataSource;
2427

@@ -44,32 +47,33 @@ public enum EmbeddedDatabaseConnection {
4447
/**
4548
* No Connection.
4649
*/
47-
NONE(null, null, null),
50+
NONE(null, null, null, (url) -> false),
4851

4952
/**
5053
* H2 Database Connection.
5154
*/
5255
H2(EmbeddedDatabaseType.H2, DatabaseDriver.H2.getDriverClassName(),
53-
"jdbc:h2:mem:%s;DB_CLOSE_DELAY=-1;DB_CLOSE_ON_EXIT=FALSE"),
56+
"jdbc:h2:mem:%s;DB_CLOSE_DELAY=-1;DB_CLOSE_ON_EXIT=FALSE", (url) -> url.contains(":h2:mem")),
5457

5558
/**
5659
* Derby Database Connection.
5760
*/
58-
DERBY(EmbeddedDatabaseType.DERBY, DatabaseDriver.DERBY.getDriverClassName(), "jdbc:derby:memory:%s;create=true"),
61+
DERBY(EmbeddedDatabaseType.DERBY, DatabaseDriver.DERBY.getDriverClassName(), "jdbc:derby:memory:%s;create=true",
62+
(url) -> true),
5963

6064
/**
6165
* HSQL Database Connection.
6266
* @deprecated since 2.4.0 in favor of {@link EmbeddedDatabaseConnection#HSQLDB}.
6367
*/
6468
@Deprecated
6569
HSQL(EmbeddedDatabaseType.HSQL, DatabaseDriver.HSQLDB.getDriverClassName(), "org.hsqldb.jdbcDriver",
66-
"jdbc:hsqldb:mem:%s"),
70+
"jdbc:hsqldb:mem:%s", (url) -> url.contains(":hsqldb:mem:")),
6771

6872
/**
6973
* HSQL Database Connection.
7074
*/
7175
HSQLDB(EmbeddedDatabaseType.HSQL, DatabaseDriver.HSQLDB.getDriverClassName(), "org.hsqldb.jdbcDriver",
72-
"jdbc:hsqldb:mem:%s");
76+
"jdbc:hsqldb:mem:%s", (url) -> url.contains(":hsqldb:mem:"));
7377

7478
private final EmbeddedDatabaseType type;
7579

@@ -79,15 +83,20 @@ public enum EmbeddedDatabaseConnection {
7983

8084
private final String url;
8185

82-
EmbeddedDatabaseConnection(EmbeddedDatabaseType type, String driverClass, String url) {
83-
this(type, driverClass, null, url);
86+
private final Predicate<String> embeddedUrl;
87+
88+
EmbeddedDatabaseConnection(EmbeddedDatabaseType type, String driverClass, String url,
89+
Predicate<String> embeddedUrl) {
90+
this(type, driverClass, null, url, embeddedUrl);
8491
}
8592

86-
EmbeddedDatabaseConnection(EmbeddedDatabaseType type, String driverClass, String fallbackDriverClass, String url) {
93+
EmbeddedDatabaseConnection(EmbeddedDatabaseType type, String driverClass, String fallbackDriverClass, String url,
94+
Predicate<String> embeddedUrl) {
8795
this.type = type;
8896
this.driverClass = driverClass;
8997
this.alternativeDriverClass = fallbackDriverClass;
9098
this.url = url;
99+
this.embeddedUrl = embeddedUrl;
91100
}
92101

93102
/**
@@ -116,19 +125,41 @@ public String getUrl(String databaseName) {
116125
return (this.url != null) ? String.format(this.url, databaseName) : null;
117126
}
118127

128+
boolean isEmbeddedUrl(String url) {
129+
return this.embeddedUrl.test(url);
130+
}
131+
132+
boolean isDriverCompatible(String driverClass) {
133+
return (driverClass != null
134+
&& (driverClass.equals(this.driverClass) || driverClass.equals(this.alternativeDriverClass)));
135+
}
136+
119137
/**
120138
* Convenience method to determine if a given driver class name represents an embedded
121139
* database type.
122140
* @param driverClass the driver class
123141
* @return true if the driver class is one of the embedded types
142+
* @deprecated since 2.4.0 in favor of {@link #isEmbedded(String, String)}
124143
*/
144+
@Deprecated
125145
public static boolean isEmbedded(String driverClass) {
126-
return driverClass != null && (matches(HSQL, driverClass) || matches(H2, driverClass)
127-
|| matches(DERBY, driverClass) || matches(HSQLDB, driverClass));
146+
return isEmbedded(driverClass, null);
147+
}
148+
149+
/**
150+
* Convenience method to determine if a given driver class name and url represent an
151+
* embedded database type.
152+
* @param driverClass the driver class
153+
* @param url the jdbc url (can be {@code null)}
154+
* @return true if the driver class and url refer to an embedded database
155+
*/
156+
public static boolean isEmbedded(String driverClass, String url) {
157+
return driverClass != null && getEmbeddedDatabaseConnection(driverClass).isEmbeddedUrl(url);
128158
}
129159

130-
private static boolean matches(EmbeddedDatabaseConnection candidate, String driverClass) {
131-
return driverClass.equals(candidate.driverClass) || driverClass.equals(candidate.alternativeDriverClass);
160+
private static EmbeddedDatabaseConnection getEmbeddedDatabaseConnection(String driverClass) {
161+
return Stream.of(H2, HSQLDB, DERBY).filter((connection) -> connection.isDriverCompatible(driverClass))
162+
.findFirst().orElse(NONE);
132163
}
133164

134165
/**
@@ -169,15 +200,16 @@ private static class IsEmbedded implements ConnectionCallback<Boolean> {
169200

170201
@Override
171202
public Boolean doInConnection(Connection connection) throws SQLException, DataAccessException {
172-
String productName = connection.getMetaData().getDatabaseProductName();
203+
DatabaseMetaData metaData = connection.getMetaData();
204+
String productName = metaData.getDatabaseProductName();
173205
if (productName == null) {
174206
return false;
175207
}
176208
productName = productName.toUpperCase(Locale.ENGLISH);
177209
EmbeddedDatabaseConnection[] candidates = EmbeddedDatabaseConnection.values();
178210
for (EmbeddedDatabaseConnection candidate : candidates) {
179211
if (candidate != NONE && productName.contains(candidate.name())) {
180-
return true;
212+
return candidate.isEmbeddedUrl(metaData.getURL());
181213
}
182214
}
183215
return false;

‎spring-boot-project/spring-boot/src/test/java/org/springframework/boot/jdbc/EmbeddedDatabaseConnectionTests.java

Lines changed: 75 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,10 +16,24 @@
1616

1717
package org.springframework.boot.jdbc;
1818

19+
import java.sql.Connection;
20+
import java.sql.DatabaseMetaData;
21+
import java.sql.SQLException;
22+
23+
import javax.sql.DataSource;
24+
1925
import org.junit.jupiter.api.Test;
26+
import org.junit.jupiter.params.ParameterizedTest;
27+
import org.junit.jupiter.params.provider.MethodSource;
28+
29+
import org.springframework.jdbc.datasource.embedded.EmbeddedDatabase;
30+
import org.springframework.jdbc.datasource.embedded.EmbeddedDatabaseBuilder;
31+
import org.springframework.jdbc.datasource.embedded.EmbeddedDatabaseType;
2032

2133
import static org.assertj.core.api.Assertions.assertThat;
2234
import static org.assertj.core.api.Assertions.assertThatIllegalArgumentException;
35+
import static org.mockito.BDDMockito.given;
36+
import static org.mockito.Mockito.mock;
2337

2438
/**
2539
* Tests for {@link EmbeddedDatabaseConnection}.
@@ -78,4 +92,65 @@ void getUrlWithEmptyDatabaseNameForHsqldb() {
7892
.withMessageContaining("DatabaseName must not be empty");
7993
}
8094

95+
@ParameterizedTest(name = "{1}")
96+
@MethodSource("embeddedDriverAndUrlParameters")
97+
void isEmbeddedWithDriverAndUrl(EmbeddedDatabaseConnection connection, String url, boolean embedded) {
98+
assertThat(EmbeddedDatabaseConnection.isEmbedded(connection.getDriverClassName(), url)).isEqualTo(embedded);
99+
}
100+
101+
static Object[] embeddedDriverAndUrlParameters() {
102+
return new Object[] { new Object[] { EmbeddedDatabaseConnection.H2, "jdbc:h2:~/test", false },
103+
new Object[] { EmbeddedDatabaseConnection.H2, "jdbc:h2:mem:test;DB_CLOSE_DELAY=-1", true },
104+
new Object[] { EmbeddedDatabaseConnection.HSQLDB, "jdbc:hsqldb:hsql://localhost", false },
105+
new Object[] { EmbeddedDatabaseConnection.HSQLDB, "jdbc:hsqldb:mem:test", true },
106+
new Object[] { EmbeddedDatabaseConnection.DERBY, "jdbc:derby:memory:test", true } };
107+
}
108+
109+
@Test
110+
void isEmbeddedWithH2DataSource() {
111+
testEmbeddedDatabase(new EmbeddedDatabaseBuilder().setType(EmbeddedDatabaseType.H2).build());
112+
}
113+
114+
@Test
115+
void isEmbeddedWithHsqlDataSource() {
116+
testEmbeddedDatabase(new EmbeddedDatabaseBuilder().setType(EmbeddedDatabaseType.HSQL).build());
117+
}
118+
119+
@Test
120+
void isEmbeddedWithDerbyDataSource() {
121+
testEmbeddedDatabase(new EmbeddedDatabaseBuilder().setType(EmbeddedDatabaseType.DERBY).build());
122+
}
123+
124+
void testEmbeddedDatabase(EmbeddedDatabase database) {
125+
try {
126+
assertThat(EmbeddedDatabaseConnection.isEmbedded(database)).isTrue();
127+
}
128+
finally {
129+
database.shutdown();
130+
}
131+
}
132+
133+
@Test
134+
void isEmbeddedWithUnknownDataSource() throws SQLException {
135+
assertThat(EmbeddedDatabaseConnection.isEmbedded(mockDataSource("unknown-db", null))).isFalse();
136+
}
137+
138+
@Test
139+
void isEmbeddedWithH2File() throws SQLException {
140+
assertThat(EmbeddedDatabaseConnection
141+
.isEmbedded(mockDataSource(EmbeddedDatabaseConnection.H2.getDriverClassName(), "jdbc:h2:~/test")))
142+
.isFalse();
143+
}
144+
145+
DataSource mockDataSource(String productName, String connectionUrl) throws SQLException {
146+
DatabaseMetaData metaData = mock(DatabaseMetaData.class);
147+
given(metaData.getDatabaseProductName()).willReturn(productName);
148+
given(metaData.getURL()).willReturn(connectionUrl);
149+
Connection connection = mock(Connection.class);
150+
given(connection.getMetaData()).willReturn(metaData);
151+
DataSource dataSource = mock(DataSource.class);
152+
given(dataSource.getConnection()).willReturn(connection);
153+
return dataSource;
154+
}
155+
81156
}

0 commit comments

Comments
 (0)