Spring执行sql脚本文件的方法

Spring执行sql脚本文件的方法

本篇解决 Spring 执行SQL脚本(文件)的问题。

场景描述可以不看。

场景描述：

我在运行单测的时候，也就是 Spring 工程启动的时候，Spring 会去执行 classpath:schema.sql（后面会解释），我想利用这一点，解决一个问题：

一次运行多个测试文件，每个文件先后独立运行，而上一个文件创建的数据，会对下一个文件运行时造成影响，所以我要在每个文件执行完成之后，重置数据库，不单单是把数据删掉，而 schema.sql 里面有 drop table 和create table。

解决方法：

//Schema 处理器

@Component

public class SchemaHandler {

private final String SCHEMA_SQL = "classpath:schema.sql";

@Autowired

private DataSource datasource;

@Autowired

private SpringContextGetter springContextGetter;

public void execute() throws Exception {

Resource resource = springContextGetter.getApplicationContext().getResource(SCHEMA_SQL);

ScriptUtils.executeSqlScript(datasource.getConnection(), resource);

}

}

// 获取 ApplicationContext

@Component

public class SpringContextGetter implements ApplicationContextAware {

private ApplicationContext applicationContext;

public ApplicationContext getApplicationContext() {

return applicationContext;

}

@Override

public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {

this.applicationContext = applicationContext;

}

}

备注：

关于为何 Spring 会去执行 classpath:schema.sql，可以参考源码

org.springframework.boot.autoconfigure.jdbc.DataSourceInitializer#runSchemaScripts

private void runSchemaScripts() {

List scripts = getScripts("spring.datasource.schema",

this.properties.getSchema(), "schema");

if (!scripts.isEmpty()) {

String username = this.properties.getSchemaUsername();

String password = this.properties.getSchemaPassword();

runScripts(scripts, username, password);

http:// try {

this.applicationContext

.publishEvent(new DataSourceInitializedEvent(this.dataSource));

// The listener might not be registered yet, so don't rely on it.

if (!this.initialized) {

runDataScripts();

this.initialized = true;

}

}

catch (IllegalStateException ex) {

logger.warhttp://n("Could not send event to complete DataSource initialization ("

+ ex.getMessage() + ")");

}

}

}

/**

* 默认拿 classpath*:schema-all.sql 和 classpath*:schema.sql

*/

private List getScripts(String http://propertyName, List resources,

String fallback) {

if (resources != null) {

return getResources(propertyName, resources, true);

}

String platform = this.properties.getPlatform();

http:// List fallbackResources = new ArrayList();

fallbackResources.add("classpath*:" + fallback + "-" + platform + ".sql");

fallbackResources.add("classpath*:" + fallback + ".sql");

return getResources(propertyName, fallbackResources, false);

}

参考：https://github.com/spring-projects/spring-boot/issues/9048

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。