數據庫導入導出工具類

只在MySQL中測試過，可將數據庫中的數據導出到指定文件，也可從包含insert語句的文件中導入數據到數據庫。

 

去掉導出的方法exportDatas()中的註釋可將表結構等也導出。

 

package cn.com.kingkit.rmdbs.comm.util; import java.io.BufferedReader; import java.io.FileOutputStream; import java.io.IOException; import java.io.OutputStreamWriter; import java.sql.Connection; import java.sql.DatabaseMetaData; import java.sql.DriverManager; import java.sql.ResultSet; import java.sql.ResultSetMetaData; import java.sql.SQLException; import java.sql.Statement; import java.util.ArrayList; import org.apache.commons.lang.StringUtils; /** * 數據庫導入導出工具 * * @author 黃慶勇 * @version 1.0 * @date 2009-12-10 * */ public class MysqlTool { /** * 將數據導出到文件 * * @param backupFile * 導出文件的絕對路徑 * @param driver * 連接數據庫驅動名稱，如："com.mysql.jdbc.Driver" * @param url * 連接數據庫地址，如："jdbc:mysql://localhost:3306/rmdbs" * @param username * 登錄數據庫的用戶名 * @param password * 登錄數據庫的密碼 */ public static void exportDatas(String backupFile, String driver, String url, String username, String password) { Connection conn = null; ResultSet rs = null; Statement stmt = null; OutputStreamWriter osw = null; try { Class.forName(driver); conn = DriverManager.getConnection(url, username, password); // Get tables DatabaseMetaData dmd = conn.getMetaData(); rs = dmd.getTables(null, null, "%", null); ArrayList<String> tables = new ArrayList<String>(); while (rs.next()) { tables.add(rs.getString(3)); } rs.close(); String resultString = ""; // resultString += "/*/n"; // resultString += "MySQL Data Transfer/n"; // resultString += "url: " + url + "/n"; // resultString += "Date: " + new Date() + "/n"; // resultString += "*//n/n"; resultString += "SET FOREIGN_KEY_CHECKS=0;/n/n"; // ResultSetMetaData rsmd = null; stmt = conn.createStatement(); // // for (String table : tables) { // resultString += "-- ----------------------------/n"; // resultString += "-- Table structure for " + table + "/n"; // resultString += "-- ----------------------------/n"; // resultString += "DROP TABLE " + table + ";/n"; // rs = stmt.executeQuery("SHOW CREATE TABLE " + table); // rsmd = rs.getMetaData(); // while (rs.next()) { // // JDBC is 1-based, Java is not !? // resultString += rs.getString(2) + ";/n/n"; // } // rs.close(); // } // // resultString += "-- ----------------------------/n"; // resultString += "-- Records/n"; // resultString += "-- ----------------------------/n"; for (String table : tables) { // System.out.println("Dumping data for table " + table // + "..."); resultString += "DELETE FROM " + table + ";/n"; rs = stmt.executeQuery("SELECT * FROM " + table); rsmd = rs.getMetaData(); while (rs.next()) { resultString += "INSERT INTO " + table + " VALUES("; // JDBC is 1-based, Java is not ! for (int col = 1; col <= rsmd.getColumnCount(); col++) { resultString += "'"; if (rs.getString(col) == null) resultString += ""; else resultString += rs.getString(col).replaceAll( "[//r]?[//n]", "////r////n"); if (col == rsmd.getColumnCount()) resultString += "'"; else resultString += "', "; } resultString += ");/n"; } resultString += "/n"; rs.close(); } stmt.close(); osw = new OutputStreamWriter(new FileOutputStream(backupFile), "UTF-8"); osw.append(resultString); osw.flush(); // System.out.println("backup is complete!!!"); } catch (Exception e) { e.printStackTrace(); } finally { try { osw.close(); } catch (IOException e1) { e1.printStackTrace(); } try { rs.close(); } catch (SQLException e) { e.printStackTrace(); } try { stmt.close(); } catch (SQLException e) { e.printStackTrace(); } try { conn.close(); } catch (SQLException e) { e.printStackTrace(); } } } /** * 導入數據到數據庫 * <p> * <b>注意：該方法是針對本工具類中MysqlTool.exportDatas()方法導出的文件進行恢復數據而開發，不保證通用與正確性！</b> * </p> * * @param input * 數據所在文件的輸入流 * @param driver * 連接數據庫驅動名稱，如："com.mysql.jdbc.Driver" * @param url * 連接數據庫地址，如："jdbc:mysql://localhost:3306/rmdbs" * @param username * 登錄數據庫的用戶名 * @param password * 登錄數據庫的密碼 */ public static void importDatas(BufferedReader input, String driver, String url, String username, String password) { Connection conn = null; Statement stmt = null; try { Class.forName(driver); conn = DriverManager.getConnection(url, username, password); stmt = conn.createStatement(); String sql = ""; do { sql = input.readLine(); if (StringUtils.isBlank(sql)) { sql = input.readLine(); } if (StringUtils.isBlank(sql)) { sql = input.readLine(); } if (StringUtils.isBlank(sql)) { sql = input.readLine(); } else { System.out.println(sql); stmt.execute(sql.trim()); } } while (!StringUtils.isBlank(sql)); stmt.close(); } catch (Exception e) { e.printStackTrace(); } finally { try { stmt.close(); } catch (SQLException e) { e.printStackTrace(); } try { conn.close(); } catch (SQLException e) { e.printStackTrace(); } } } /** * @param args */ public static void main(String[] args) { String backupFile = "d://rmdb.sql"; String driver = "com.mysql.jdbc.Driver"; String url = "jdbc:mysql://localhost:3306/rmdbs"; String username = "root"; String password = "kingkit"; MysqlTool.exportDatas(backupFile, driver, url, username, password); } }