Spring Boot項目——ScriptUtils.executeSqlScript執行sql腳本文件

背景

• 項目測試時需要執行一些sql腳本文件，將數據初始化到數據庫中

代碼案例

/**
 * @author ********
 * @Description:
 * @date 2020/4/2
 */
public class ConnUtil  {
    private static final Logger logger = LoggerFactory.getLogger( ConnUtil.class );
    private Connection connection;

    public Connection executeSql( Connection connection,String sqlFileName ){

        try {
            String driverClassName = "com.mysql.cj.jdbc.Driver";
            String url = "jdbc:mysql://localhost:3306/test?useUnicode=true&CharSet=utf8mb4&characterEncoding=utf8&characterSetResults=utf8&autoReconnect=true&autoReconnectForPools=true&rewriteBatchedStatements=true&allowMultiQueries=true&useSSL=false&serverTimezone=GMT";
            String username = "test";
            String password = "test";

            Class.forName(driverClassName);
            connection = DriverManager.getConnection(url, username, password);
        } catch (Exception e) {
            e.printStackTrace();
        }

        ClassPathResource rc = new ClassPathResource(sqlFileName);
        EncodedResource er = new EncodedResource(rc, "utf-8");
        ScriptUtils.executeSqlScript(connection, er);

        return connection;
    }
//手動釋放連接
    void closeAll(Connection conn,PreparedStatement ps,ResultSet rs){
        try {
            if(rs != null){
                rs.close();
            }
            if(ps != null){
                ps.close();
            }
            if(conn != null){
                conn.close();
            }
        }catch (SQLException e){
            e.getMessage();
        }
    }

　　//測試
　　public void testConn(){
　　　　executeSql("testsql/test_inset_user.sql");//test_inset_user.sql存儲在resources的testsql目錄下
　　}


}

注意

• 使用ScriptUtils.executeSqlScript執行sql腳本文件必須手動釋放連接
• sql腳本文件需要在resources目錄下

executeSql1( "testSql/test_superviser_record.sql");

executeSql2( "testSql/test_deldata.sql");
executeSql2( "testSql/test_miners.sql");
executeSql2( "testSql/test_user.sql");