评论

收藏

[Java] Spring执行sql脚本文件的方法

编程语言 编程语言 发布于:2021-09-18 16:17 | 阅读数:343 | 评论:0

这篇文章主要介绍了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);
  }
参考:https://github.com/spring-projects/spring-boot/issues/9048
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持CodeAE代码之家
原文链接:https://segmentfault.com/a/1190000018344940

关注下面的标签,发现更多相似文章