栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Java

springboot 容器加载完成后执行sql脚本

Java 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

springboot 容器加载完成后执行sql脚本

import org.apache.commons.lang3.StringUtils;
import org.apache.ibatis.io.Resources;
import org.apache.ibatis.jdbc.ScriptRunner;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.ApplicationArguments;
import org.springframework.boot.ApplicationRunner;
import org.springframework.core.annotation.Order;
import org.springframework.jdbc.datasource.DataSourceUtils;
import org.springframework.stereotype.Component;

import javax.sql.DataSource;
import java.io.IOException;
import java.io.PrintWriter;
import java.io.Reader;
import java.io.StringWriter;
import java.nio.charset.StandardCharsets;
import java.sql.Connection;
import java.sql.SQLException;


@Component
@Order(value = 1)
public class SysMgrSlqScriptRunner implements ApplicationRunner {

	private static final Logger logger = LoggerFactory.getLogger(SysMgrSlqScriptRunner.class);

	@Autowired
	private DataSource dataSource;

	@Override
	public void run(ApplicationArguments args) throws Exception {
		String scriptPath = "db/ca.sql";
		execSqlScript(dataSource, scriptPath);
		}

	}

	
	public static void execSqlScript(DataSource dataSource, String sqlFilePath) throws IOException {
		// 通过数据源获取数据库连接
		Connection connection = DataSourceUtils.getConnection(dataSource);
		// 创建脚本执行器
		ScriptRunner scriptRunner = new ScriptRunner(connection);
		// 设置执行器日志输出
		StringWriter writer = new StringWriter();
		PrintWriter printWriter = new PrintWriter(writer);
		scriptRunner.setLogWriter(printWriter);
		// 设置执行器错误日志输出
		StringWriter errorWriter = new StringWriter();
		PrintWriter errorPrintWriter = new PrintWriter(errorWriter);
		scriptRunner.setErrorLogWriter(errorPrintWriter);
		// 设置读取资源文件的字符集
		Resources.setCharset(StandardCharsets.UTF_8);

		try (Reader reader = Resources.getResourceAsReader(sqlFilePath)) {
			scriptRunner.runScript(reader);
		} finally {
			try {
				connection.close();
			} catch (SQLException e) {
				logger.error("to close the connection of database is failure", e);
			}
			if (StringUtils.isNotEmpty(writer.toString())) {
				logger.info(writer.toString());
			}
			if (StringUtils.isNotEmpty(errorWriter.toString())) {
				logger.error(errorWriter.toString());
			}
		}
	}
}

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/999679.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号