Java调用SQL脚本执行的方案

时间:2024-01-23 09:07:29

在Java中调用SQL脚本的方式有多种,在这里只记录一种自己常用的方式,个人觉得挺实用方便的。

运用ScriptRunner这个类。

 1 import org.apache.ibatis.io.Resources;
 2 import org.apache.ibatis.jdbc.ScriptRunner;
 3 
 4 import java.io.IOException;
 5 import java.sql.Connection;
 6 import java.sql.DriverManager;
 7 import java.sql.SQLException;
 8 import java.util.Properties;
 9 
10 /**
11  * 运行Sql脚本
12  * sql脚本放在resources下的sql文件夹下
13  */
14 public final class RunSqlScript {
15     /**
16      * <p>运行指定的sql脚本
17      * @param sqlFileName 需要执行的sql脚本的名字
18      */
19     public static void run(String sqlFileName) {
20         try {
21             // 获取数据库相关配置信息
22             Properties props = Resources.getResourceAsProperties("db.properties");
23 
24             // jdbc 连接信息: 注: 现在版本的JDBC不需要配置driver,因为不需要Class.forName手动加载驱动
25             String url = props.getProperty("jdbc.url");
26             String username = props.getProperty("jdbc.username");
27             String password = props.getProperty("jdbc.password");
28 
29             // 建立连接
30             Connection conn = DriverManager.getConnection(url, username, password);
31 
32             // 创建ScriptRunner,用于执行SQL脚本
33             ScriptRunner runner = new ScriptRunner(conn);
34             runner.setErrorLogWriter(null);
35             runner.setLogWriter(null);
36             // 执行SQL脚本
37             runner.runScript(Resources.getResourceAsReader("sql/" + sqlFileName + ".sql"));
38 
39             // 关闭连接
40             conn.close();
41 
42             // 若成功,打印提示信息
43             System.out.println("====== SUCCESS ======");
44         } catch (IOException | SQLException e) {
45             e.printStackTrace();
46         }
47     }
48 }

ps下:在现在版本的JDBC驱动中,已经不需要像以前一样加载driver和手动class.forName(driver)来加载驱动了,详情可以自行点进DriverManager中看,有一个static {} 静态代码块,里面已经为我们加载初始化了驱动。