Java 在springboot中使用 ScriptUtils.executeSqlScript手动控制执行sql脚本

最新推荐文章于 2024-09-08 15:28:15 发布

杨树叶子0526

最新推荐文章于 2024-09-08 15:28:15 发布

阅读量1.6k

点赞数

分类专栏： java开发过程记录文章标签： java 开发语言 mysql Powered by 金山文档

本文链接：https://blog.csdn.net/qq_38360823/article/details/128964784

版权

java开发过程记录专栏收录该内容

19 篇文章 0 订阅

订阅专栏

该文章演示了如何在SpringBoot应用中利用ScriptUtils的executeSqlScript方法，在Oracle数据库中执行SQL脚本，特别是在定时任务中的用法。通过PathMatchingResourcePatternResolver加载多个.sql文件，然后遍历执行，用于创建或更新数据库表。

摘要由CSDN通过智能技术生成

springboot 使用 ScriptUtils.executeSqlScript手动控制执行sql脚本（可使用在定时器等地方）

例子：在定时任务中(代码仅供参考，其中省略了部分数据判断，可自行添加)，当前数据库为oracle数据库

import org.springframework.beans.factory.annotation.Autowired;
import org.apache.commons.io.IOUtils;
import org.springframework.core.io.Resource;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
import org.springframework.jdbc.datasource.init.ScriptUtils;
import org.springframework.core.io.InputStreamResource;
import org.springframework.stereotype.Component;

import javax.sql.DataSource;
import java.io.InputStream;
import java.io.ByteArrayInputStream;

@Component
public class TableDataTask {
    @Autowired
    DataSource dataSource;

    public void createTable() {
        try {
            //db文件夹下的脚本可能有多个，所以，这里要使用PathMatchingResourcePatternResolver，读取多个.sql的文件；
            PathMatchingResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
            //获取所有sql脚本
            Resource[] resources = resolver.getResources("classpath:dbsql/oracle/*.sql");
            if (resources.length == 0) {
                log.error("没有需要生成表的sql脚本");
                return;
            }
            /*可以在中间加如对文件名信息的判断， 当前代码中省略*/
            //循环创建
            for (Resource resource : resourceList) {
                //获取创建脚本内容,并替换文件信息
                 String sql = getSqlData(resource.getInputStream());
                /*可对字符串内容处理*/
                //创建表，执行sql, 将sql字符串以流的形式传入
                 ScriptUtils.executeSqlScript(dataSource.getConnection(), new InputStreamResource(new ByteArrayInputStream(sql.getBytes())));
            }
        } catch (Exception e) {
            e.printStackTrace();
            log.error("定时创建表操作失败：" +e.getMessage());
        }
    }

    /**
     * 获取sql脚本内容
     * @param is 文件流信息
     * @return 文件流内容
     * @throws IOException 异常
     */
    private String getSqlData(InputStream is) throws IOException {
        return IOUtils.toString(is, StandardCharsets.UTF_8);
    }
}

sql脚本路径在：resources.dbsql.oracle目录下；

sql脚本例子

CREATE TABLE TEST_TABLESPACE.TEST
(
    TABLE_NAME  VARCHAR2(30) NOT NULL
        CONSTRAINT TEST_PK
            PRIMARY KEY,
    UPDATE_TIME VARCHAR2(20) NOT NULL
);

COMMENT ON COLUMN TEST_TABLESPACE.TEST.TABLE_NAME IS '名称';

COMMENT ON COLUMN TEST_TABLESPACE.TEST.UPDATE_TIME IS '记录更新时间';