使用java执行sql脚本的方法
解析sql脚本,删除不必要的注释和空行
将语句按分号拆开
并将最终的语句放入batch里面 最后进行执行
package test;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.InputStreamReader;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
/* * 使用java以及jdbc执行sql脚本的工具示例代码 */
public class SqlHelper {
public static void main(String[] args){
String path = "文件地址字符串";
String sql = getText(path);
List<String> sqlarr = getSql(sql);
for(int i=0; i<10; i++){
System.out.println(i+":"+sqlarr.get(i));
}
try{
SqlHelper.execute(getConn(),sqlarr);
}catch(Exception e){
e.printStackTrace();
}
}
private static Connection getConn() {
String driver = "com.mysql.jdbc.Driver";
String url = "数据库连接";
String username = "账号";
String password = "密码";
Connection conn = null;
try {
Class.forName(driver); //classLoader,加载对应驱动
conn = (Connection) DriverManager.getConnection(url, username, password);
} catch (ClassNotFoundException e) {
e.printStackTrace();
} catch (SQLException e) {
e.printStackTrace();
}
return conn;
}
public static void execute(Connection conn, List<String> sqlFile) throws Exception {
Statement stmt = null;
stmt = conn.createStatement();
for (String sql : sqlFile) {
sql = sql.trim();
if(sql!=null&&!sql.equals(""))
stmt.addBatch(sql);
}
int[] rows = stmt.executeBatch();
System.out.println("Row count:" + Arrays.toString(rows));
conn.close();
}
/* * getText方法吧path路径里面的文件按行读数来放入一个大的String里面去 * 并在换行的时候加入\r\n */
public static String getText(String path){
File file = new File(path);
if(!file.exists()||file.isDirectory()){
return null;
}
StringBuilder sb = new StringBuilder();
try{
FileInputStream fis = new FileInputStream(path);
InputStreamReader isr = new InputStreamReader(fis,"UTF-8");
BufferedReader br = new BufferedReader(isr);
String temp = null;
temp = br.readLine();
while(temp!=null){
if(temp.length()>=2){
String str1 = temp.substring(0, 1);
String str2 = temp.substring(0, 2);
if(str1.equals("#")||str2.equals("--")||str2.equals("/*")||str2.equals("//")){
temp = br.readLine();
continue;
}
sb.append(temp+"\r\n");
}
temp = br.readLine();
}
br.close();
}catch(Exception e){
e.printStackTrace();
}
return sb.toString();
}
/* * getSqlArray方法 * 从文件的sql字符串中分析出能够独立执行的sql语句并返回 */
public static List<String> getSql(String sql){
String s = sql;
s = s.replaceAll("\r\n", "\r");
s = s.replaceAll("\r", "\n");
List<String> ret = new ArrayList<String>();
String[] sqlarry = s.split(";"); //用;把所有的语句都分开成一个个单独的句子
sqlarry = filter(sqlarry);
ret = Arrays.asList(sqlarry);
return ret;
}
public static String[] filter(String[] ss){
List<String> strs = new ArrayList<String>();
for(String s : ss){
if(s!=null&&!s.equals("")){
strs.add(s);
}
}
String[] result = new String[strs.size()];
for(int i=0; i<strs.size(); i++){
result[i] = strs.get(i).toString();
}
return result;
}
}