springboot启动运行.sql文件自动建表

springboot启动运行.sql文件自动建表

import org.apache.ibatis.jdbc.ScriptRunner; import org.springframework.beans.factory.annotation.Value; import org.springframework.context.annotation.Configuration; import javax.annotation.PostConstruct; import javax.annotation.Resource; import java.io.File; import java.io.FileInputStream; import java.io.IOException; import java.io.InputStreamReader; import java.nio.charset.StandardCharsets; import java.sql.Connection; import java.sql.DriverManager; import java.sql.SQLException; import java.util.List; /** * springboot启动运行.sql文件自动建表 */ @Configuration public class InitSqlTableClass { // 这里从.yml配置文件读取数据库连接配置 @Value("${spring.datasource.name}") private String name; @Value("${spring.datasource.url}") private String url; @Value("${spring.datasource.username}") private String username; @Value("${spring.datasource.password}") private String password; @PostConstruct public void test() throws IOException, SQLException { // 放在项目中.sql文件的路径 String path = "src/main/resources/sql"; // 连接数据库 Connection conn = DriverManager.getConnection(url, username, password); // 创建ScriptRunner，用于执行SQL脚本 ScriptRunner runner = new ScriptRunner(conn); runner.setErrorLogWriter(null); runner.setLogWriter(null); // 获取项目中的所有要执行的sql文件 File[] files = new File(path).listFiles(); assert files != null; for (File file : files) { // 执行SQL脚本 runner.runScript(new InputStreamReader(new FileInputStream(file), StandardCharsets.UTF_8)); } } conn.close(); } }