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