Spring执行sql脚本文件的方法
程序员文章站
2024-02-17 23:34:34
本篇解决 spring 执行sql脚本(文件)的问题。
场景描述可以不看。
场景描述:
我在运行单测的时候,也就是 spring 工程启动的时候,spring 会去执...
本篇解决 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<resource> 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); 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.warn("could not send event to complete datasource initialization (" + ex.getmessage() + ")"); } } } /** * 默认拿 classpath*:schema-all.sql 和 classpath*:schema.sql */ private list<resource> getscripts(string propertyname, list<string> resources, string fallback) { if (resources != null) { return getresources(propertyname, resources, true); } string platform = this.properties.getplatform(); list<string> fallbackresources = new arraylist<string>(); fallbackresources.add("classpath*:" + fallback + "-" + platform + ".sql"); fallbackresources.add("classpath*:" + fallback + ".sql"); return getresources(propertyname, fallbackresources, false); }
参考:
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。