Spring执行sql脚本文件的方法_Java教程

Spring执行sql脚本文件的方法

2021-07-17 13:17志哥的成长笔记 Java教程

这篇文章主要介绍了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<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);

									  }