Allow configuration to specify randomly generated database name

See gh-7004
This commit is contained in:
Rob Fletcher 2016-09-22 22:12:05 -07:00 committed by Stephane Nicoll
parent 0f97ccf2fa
commit 7fcb197092
3 changed files with 56 additions and 1 deletions

View File

@ -57,6 +57,11 @@ public class DataSourceProperties
*/
private String name = "testdb";
/**
* If <code>true</code> the database name is randomly generated.
*/
private boolean generateName = false;
/**
* Fully qualified name of the connection pool implementation to use. By default, it
* is auto-detected from the classpath.
@ -183,6 +188,14 @@ public class DataSourceProperties
this.name = name;
}
public boolean isGenerateName() {
return this.generateName;
}
public void setGenerateName(boolean generateName) {
this.generateName = generateName;
}
public Class<? extends DataSource> getType() {
return this.type;
}

View File

@ -55,7 +55,13 @@ public class EmbeddedDataSourceConfiguration implements BeanClassLoaderAware {
public EmbeddedDatabase dataSource() {
EmbeddedDatabaseBuilder builder = new EmbeddedDatabaseBuilder()
.setType(EmbeddedDatabaseConnection.get(this.classLoader).getType());
this.database = builder.setName(this.properties.getName()).build();
if (this.properties.isGenerateName()) {
builder.generateUniqueName(true);
}
else {
builder.setName(this.properties.getName());
}
this.database = builder.build();
return this.database;
}

View File

@ -16,11 +16,17 @@
package org.springframework.boot.autoconfigure.jdbc;
import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.Properties;
import javax.sql.DataSource;
import org.junit.Test;
import org.springframework.context.annotation.AnnotationConfigApplicationContext;
import org.springframework.core.env.PropertiesPropertySource;
import static org.assertj.core.api.Assertions.assertThat;
@ -42,4 +48,34 @@ public class EmbeddedDataSourceConfigurationTests {
this.context.close();
}
@Test
public void generatesUniqueDatabaseName() throws Exception {
Properties myProps = new Properties();
myProps.setProperty("spring.datasource.generate-name", "true");
this.context = new AnnotationConfigApplicationContext();
this.context.register(EmbeddedDataSourceConfiguration.class);
this.context.getEnvironment().getPropertySources().addFirst(new PropertiesPropertySource("whatever", myProps));
this.context.refresh();
DataSource dataSource = this.context.getBean(DataSource.class);
assertThat(getDatabaseName(dataSource)).isNotEqualToIgnoringCase("testdb");
this.context.close();
}
private String getDatabaseName(DataSource dataSource) throws SQLException {
Connection connection = dataSource.getConnection();
try {
ResultSet catalogs = connection.getMetaData().getCatalogs();
if (catalogs.next()) {
return catalogs.getString(1);
}
else {
throw new IllegalStateException("Unable to get database name");
}
}
finally {
connection.close();
}
}
}