import org.apache.ibatis.io.Resources;
import org.apache.ibatis.jdbc.ScriptRunner;

import java.io.IOException;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.util.Properties;

/**
 * 運行Sql腳本
 * sql腳本放在resources下的sql文件夾下
 */
public final class RunSqlScript {
 /**
 * <p>運行指定的sql腳本
 * @param sqlFileName 需要執(zhí)行的sql腳本的名字
 */
 public static void run(String sqlFileName) {
 try {
  // 獲取數(shù)據(jù)庫相關配置信息
  Properties props = Resources.getResourceAsProperties("db.properties");

  // jdbc 連接信息: 注: 現(xiàn)在版本的JDBC不需要配置driver，因為不需要Class.forName手動加載驅動
  String url = props.getProperty("jdbc.url");
  String username = props.getProperty("jdbc.username");
  String password = props.getProperty("jdbc.password");

  // 建立連接
  Connection conn = DriverManager.getConnection(url, username, password);

  // 創(chuàng)建ScriptRunner，用于執(zhí)行SQL腳本
  ScriptRunner runner = new ScriptRunner(conn);
  runner.setErrorLogWriter(null);
  runner.setLogWriter(null);
  // 執(zhí)行SQL腳本
  runner.runScript(Resources.getResourceAsReader("sql/" + sqlFileName + ".sql"));

  // 關閉連接
  conn.close();

  // 若成功，打印提示信息
  System.out.println("====== SUCCESS ======");
 } catch (IOException | SQLException e) {
  e.printStackTrace();
 }
 }
}