因项目中有需要需根据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 tableList = new ArrayList();//全局存放表名列表
private static List insertList = new ArrayList();//全局存放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 listSQL = new ArrayList();
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 createSQL(String file) throws Exception {
List listSQL = new ArrayList();
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 insertSQL = new ArrayList();
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);
}
}
select * from t_chapter
select * from t_question_type
使用中目前未发现什么问题,如有BUG,请各位大神改进。