根據MySQL表資料匯出INSERT INTO語句的方法
阿新 • • 發佈：2019-02-06
因專案中有需要需根據MySQL表資料生成INSERT INTO語句，在網上找了些現成的程式碼，原作者是誰就不知道了，但是發現有BUG，不能適用，遂對他人程式碼進行了修改。修改後能較好的匯出INSERT INTO語句。程式碼如下：
package dwz.interaction; import java.io.*; import java.sql.*; import java.util.ArrayList; import java.util.List; /** * Update by internetroot on 2014-09-06. */ public class SqliteSQLGenerator { private static Connection conn = null; private static Statement sm = null; private static String insert = "INSERT INTO";//插入sql private static String values = "VALUES";//values關鍵字 private static List<String> tableList = new ArrayList<String>();//全域性存放表名列表 private static List<String> insertList = new ArrayList<String>();//全域性存放insertsql檔案的資料 private static String filePath = "E://insertSQL.txt";//絕對路徑 匯出資料的檔案 public static String generateTableDataSQL(String sql, String[] params) { return null; } public static String executeSelectSQLFile(String file, String[] params) throws Exception { List<String> listSQL = new ArrayList<String>(); connectSQL("com.mysql.jdbc.Driver", "jdbc:mysql://127.0.0.1:3308/htedu?useUnicode=true&characterEncoding=UTF-8", "root", "");//連線資料庫 listSQL = createSQL(file);//建立查詢語句 executeSQL(conn, sm, listSQL, tableList);//執行sql並拼裝 createFile();//建立檔案 return null; } /** * 拼裝查詢語句 * * @return 返回select集合 */ private static List<String> createSQL(String file) throws Exception { List<String> listSQL = new ArrayList<String>(); BufferedReader br = null; InputStreamReader fr = null; InputStream is = null; int i;//表名的第一個字元位置 int k;//表名單最後一個字元的位置 String tableName; try { is = SqliteSQLGenerator.class.getResourceAsStream(file); fr = new InputStreamReader(is); br = new BufferedReader(fr); String rec = null;//一行 while ((rec = br.readLine()) != null) { rec = rec.toLowerCase(); i = rec.indexOf("from ", 1) + 5; k = rec.indexOf(" ", i); if (k == -1) { k = rec.length(); } ; tableName = rec.substring(i, k); tableList.add(tableName); //獲取所有查詢語句 listSQL.add(rec.toString()); } } finally { if (br != null) { br.close(); } if (fr != null) { fr.close(); } if (is != null) { is.close(); } } return listSQL; } /** * 建立insertsql.txt並匯出資料 */ private static void createFile() { File file = new File(filePath); if (!file.exists()) { try { file.createNewFile(); } catch (IOException e) { System.out.println("建立檔名失敗！！"); e.printStackTrace(); } } FileWriter fw = null; BufferedWriter bw = null; try { fw = new FileWriter(file); bw = new BufferedWriter(fw); if (insertList.size() > 0) { for (int i = 0; i < insertList.size(); i++) { bw.append(insertList.get(i)); bw.append("\n"); } } } catch (IOException e) { e.printStackTrace(); } finally { try { bw.close(); fw.close(); } catch (IOException e) { e.printStackTrace(); } } } /** * 連線資料庫 建立statement物件 * * @param driver * @param url * @param UserName * @param Password */ public static void connectSQL(String driver, String url, String UserName, String Password) { try { Class.forName(driver).newInstance(); conn = DriverManager.getConnection(url, UserName, Password); sm = conn.createStatement(); } catch (Exception e) { e.printStackTrace(); } } /** * 執行sql並返回插入sql * * @param conn * @param sm * @param listSQL * @throws java.sql.SQLException */ public static void executeSQL(Connection conn, Statement sm, List listSQL, List listTable) throws SQLException { List<String> insertSQL = new ArrayList<String>(); ResultSet rs = null; try { rs = getColumnNameAndColumeValue(sm, listSQL, listTable, rs); } catch (SQLException e) { e.printStackTrace(); } finally { rs.close(); sm.close(); conn.close(); } } /** * 獲取列名和列值 * * @param sm * @param listSQL * @param rs * @return * @throws java.sql.SQLException */ private static ResultSet getColumnNameAndColumeValue(Statement sm, List listSQL, List ListTable, ResultSet rs) throws SQLException { for (int j = 0; j < listSQL.size(); j++) { String sql = String.valueOf(listSQL.get(j)); rs = sm.executeQuery(sql); ResultSetMetaData rsmd = rs.getMetaData(); int columnCount = rsmd.getColumnCount(); while (rs.next()) { StringBuffer ColumnName = new StringBuffer(); StringBuffer ColumnValue = new StringBuffer(); for (int i = 1; i <= columnCount; i++) { String value = rs.getString(i); if (i == columnCount) { ColumnName.append(rsmd.getColumnName(i)); if (Types.CHAR == rsmd.getColumnType(i) || Types.VARCHAR == rsmd.getColumnType(i) || Types.LONGVARCHAR == rsmd.getColumnType(i)) { if (value == null) { ColumnValue.append("null"); } else { ColumnValue.append("'").append(value).append("'"); } } else if (Types.SMALLINT == rsmd.getColumnType(i) || Types.INTEGER == rsmd.getColumnType(i) || Types.BIGINT == rsmd.getColumnType(i) || Types.FLOAT == rsmd.getColumnType(i) || Types.DOUBLE == rsmd.getColumnType(i) || Types.NUMERIC == rsmd.getColumnType(i) || Types.DECIMAL == rsmd.getColumnType(i)) { if (value == null) { ColumnValue.append("null"); } else { ColumnValue.append(value); } } else if (Types.DATE == rsmd.getColumnType(i) || Types.TIME == rsmd.getColumnType(i) || Types.TIMESTAMP == rsmd.getColumnType(i)) { if (value == null) { ColumnValue.append("null"); } else { ColumnValue.append("timestamp'").append(value).append("'"); } } else { if (value == null) { ColumnValue.append("null"); } else { ColumnValue.append(value); } } } else { ColumnName.append(rsmd.getColumnName(i) + ","); if (Types.CHAR == rsmd.getColumnType(i) || Types.VARCHAR == rsmd.getColumnType(i) || Types.LONGVARCHAR == rsmd.getColumnType(i)) { if (value == null) { ColumnValue.append("null,"); } else { ColumnValue.append("'").append(value).append("',"); } } else if (Types.SMALLINT == rsmd.getColumnType(i) || Types.INTEGER == rsmd.getColumnType(i) || Types.BIGINT == rsmd.getColumnType(i) || Types.FLOAT == rsmd.getColumnType(i) || Types.DOUBLE == rsmd.getColumnType(i) || Types.NUMERIC == rsmd.getColumnType(i) || Types.DECIMAL == rsmd.getColumnType(i)) { if (value == null) { ColumnValue.append("null,"); } else { ColumnValue.append(value).append(","); } } else if (Types.DATE == rsmd.getColumnType(i) || Types.TIME == rsmd.getColumnType(i) || Types.TIMESTAMP == rsmd.getColumnType(i)) { if (value == null) { ColumnValue.append("null,"); } else { ColumnValue.append("timestamp'").append(value).append("',"); } } else { if (value == null) { ColumnValue.append("null,"); } else { ColumnValue.append(value).append(","); } } } } //System.out.println(ColumnName.toString()); //System.out.println(ColumnValue.toString()); insertSQL(ListTable.get(j).toString(), ColumnName, ColumnValue); } } return rs; } /** * 拼裝insertsql 放到全域性list裡面 * * @param ColumnName * @param ColumnValue */ private static void insertSQL(String TableName, StringBuffer ColumnName, StringBuffer ColumnValue) { StringBuffer insertSQL = new StringBuffer(); insertSQL.append(insert).append(" ").append(TableName).append("(").append(ColumnName.toString()) .append(")").append(values).append("(").append(ColumnValue.toString()).append(");"); insertList.add(insertSQL.toString()); System.out.println(insertSQL.toString()); } public static void main(String[] args) throws Exception { //String file1 = "/config/export_sqlite_data_clear.cfg"; //executeSelectSQLFile(file1, null); String file2 = "/config/export_sqlite_data_select.cfg"; executeSelectSQLFile(file2, null); } }
程式碼中應用的cfg檔案形式：
select * from t_chapter
select * from t_question_type使用中目前未發現什麼問題，如有BUG，請各位大神改進。