This commit is contained in:
Keith Donald
2009-05-06 09:14:25 +00:00
parent 0320445316
commit 4bfcd16130
2 changed files with 16 additions and 4 deletions

View File

@@ -21,6 +21,7 @@ import java.sql.Connection;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.LinkedList;
import java.util.List;
@@ -52,6 +53,14 @@ public class ResourceDatabasePopulator implements DatabasePopulator {
scripts.add(script);
}
/**
* Set the scripts to execute to populate the database.
* @param scripts the scripts to execute
*/
public void setScripts(Resource[] scripts) {
this.scripts = Arrays.asList(scripts);
}
/**
* 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}.
@@ -60,7 +69,7 @@ public class ResourceDatabasePopulator implements DatabasePopulator {
public void setSqlScriptEncoding(String sqlScriptEncoding) {
this.sqlScriptEncoding = sqlScriptEncoding;
}
public void populate(Connection connection) throws SQLException {
for (Resource script : scripts) {
executeSqlScript(connection, applyEncodingIfNecessary(script), false);