Java Ö´ÐÐ SQL ½Å±¾Îļþ

ÊÇ¿½±´µÄ±ðÈ˵ģ¬ÒÔ±¸Ñ§Ï°
package com.unmi.db;
import java.io.FileInputStream;
import java.io.InputStream;
import java.sql.Connection;
import java.sql.Statement;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
/**
* ¶ÁÈ¡ SQL ½Å±¾²¢Ö´ÐÐ
* @author Unmi
*/
public class SqlFileExecutor {
/**
* ¶ÁÈ¡ SQL Îļþ£¬»ñÈ¡ SQL Óï¾ä
* @param sqlFile SQL ½Å±¾Îļþ
* @return List<sql> ·µ»ØËùÓÐ SQL Óï¾äµÄ List
* @throws Exception
*/
private List<String> loadSql(String sqlFile) throws Exception {
List<String> sqlList = new ArrayList<String>();
try {
InputStream sqlFileIn = new FileInputStream(sqlFile);
StringBuffer sqlSb = new StringBuffer();
byte[] buff = new byte[1024];
int byteRead = 0;
while ((byteRead = sqlFileIn.read(buff)) != -1) {
sqlSb.append(new String(buff, 0, byteRead));
}
// Windows Ï»»ÐÐÊÇ \r\n, Linux ÏÂÊÇ \n
String[] sqlArr = sqlSb.toString().split("(;\\s*\\r\\n)|(;\\s*\\n)");
for (int i = 0; i < sqlArr.length; i++) {
String sql = sqlArr[i].replaceAll("--.*", "").trim();
if (!sql.equals("")) {
sqlList.add(sql);
}
}
return sqlList;
} catch (Exception ex) {
throw new Exception(ex.getMessage());
}
}
/**
* ´«ÈëÁ¬½ÓÀ´Ö´ÐÐ SQL ½Å±¾Îļþ£¬ÕâÑù¿ÉÓëÆäÍâµÄÊý¾Ý¿â²Ù×÷ͬ´¦Ò»¸öÊÂÎïÖÐ
* @param conn ´«ÈëÊý¾Ý¿âÁ¬½Ó
* @param sqlFile SQL ½Å±¾Îļþ
* @throws Exception
*/
public void execute(Connection conn, String sqlFile) throws Exception {
Statement stmt = null;
List<String> sqlList = loadSql(sqlFile);
stmt = conn.createStatement();
for (String sql : sqlList) {
stmt.addBatch(sql);
}
int[] rows = stmt.executeBatch();
System.out.println("Row count:" + Arrays.toString(rows));
}
/**
* ×Ô½¨Á¬½Ó£¬¶ÀÁ¢ÊÂÎïÖÐÖ´ÐÐ SQL Îļþ
* @param sqlFile SQL ½Å±¾Îļþ
* @throws Exception
*/
public void execute(String sqlFile) throws Exception {
Connection conn = DBCente