root
/
spring-framework
mirror of https://github.com/spring-projects/spring-framework.git
1
0
Fork 0
Code Issues Actions 2 Packages Projects Releases Wiki Activity

polish

This commit is contained in:
Keith Donald 2009-05-06 08:38:47 +00:00
parent c25760f5db
commit 6771a22988
1 changed files with 16 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
org.springframework.jdbc/src/main/java/org/springframework/jdbc/datasource/embedded/ResourceDatabasePopulator.java
View File

@ -31,22 +31,19 @@ import org.springframework.core.io.support.EncodedResource;
import org.springframework.util.StringUtils;
/**
* Populates a database from schema and test-data SQL defined in external resources. By default, looks for a schema.sql
* file and test-data.sql resource in the root of the classpath.
* Populates a database from SQL scripts defined in external resources.
* <p>
* May be configured:<br>
* Call {@link #setSchemaLocation(Resource)} to configure the location of the database schema file.<br>
* Call {@link #setTestDataLocation(Resource)} to configure the location of the test data file.<br>
* Call {@link #setSqlScriptEncoding(String)} to set the encoding for the schema and test data SQL.<br>
* Call {@link #addScript(Resource)} to add a SQL script location.<br>
* Call {@link #setSqlScriptEncoding(String)} to set the encoding for all added scripts.<br>
*/
public class ResourceDatabasePopulator implements DatabasePopulator {
private static final Log logger = LogFactory.getLog(ResourceDatabasePopulator.class);
private String sqlScriptEncoding;
private List<Resource> scripts = new ArrayList<Resource>();
private String sqlScriptEncoding;
/**
* Add a script to execute to populate the database.
* @param script the path to a SQL script
@ -57,6 +54,8 @@ public class ResourceDatabasePopulator implements DatabasePopulator {
/**
* Specify the encoding for SQL scripts, if different from the platform encoding.
* Note setting this property has no effect on added scripts that are already {@link EncodedResource encoded resources}.
* @see #addScript(Resource)
*/
public void setSqlScriptEncoding(String sqlScriptEncoding) {
this.sqlScriptEncoding = sqlScriptEncoding;
@ -64,10 +63,18 @@ public class ResourceDatabasePopulator implements DatabasePopulator {
public void populate(Connection connection) throws SQLException {
for (Resource script : scripts) {
executeSqlScript(connection, new EncodedResource(script, sqlScriptEncoding), false);
executeSqlScript(connection, applyEncodingIfNecessary(script), false);
}
}
private EncodedResource applyEncodingIfNecessary(Resource script) {
if (script instanceof EncodedResource) {
return (EncodedResource) script;
} else {
return new EncodedResource(script, sqlScriptEncoding);
}
}
/**
* Execute the given SQL script. <p>The script will normally be loaded by classpath. There should be one statement
* per line. Any semicolons will be removed. <b>Do not use this method to execute DDL if you expect rollback.</b>