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

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 '记录更新时间';


版权声明:本文为qq_38360823原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。