日期:2014-05-16 浏览次数:20503 次
package baor.src.dbunit;
import java.sql.ResultSet;
import java.sql.SQLException;
import baor.test.sql.ConnPropInfo;
import baor.test.sql.ConnUtil;
import java.sql.Statement;
public interface CRUDDemo {
public abstract ResultSet selectAllDemo(Statement stmt, String SQLString)
throws SQLException;
public abstract Statement setSTMT(ConnUtil connUtil, ConnPropInfo demo)
throws SQLException;
}package baor.src.dbunit;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import baor.test.sql.ConnPropInfo;
import baor.test.sql.ConnUtil;
public class SelectAllDemo implements CRUDDemo {
@Override
public ResultSet selectAllDemo(Statement stmt, String SQLString)
throws SQLException {
// TODO Auto-generated method stub
// 创建数据集对象
ResultSet rs = stmt.executeQuery(SQLString);
// System.out.println("TestNo" + "\t" + "Description" + "\t" +
// "Author");
return rs;
}
@Override
public Statement setSTMT(ConnUtil connUtil, ConnPropInfo demo)
throws SQLException {
// TODO Auto-generated method stub
// 创建stmt对象
Statement stmt = connUtil.getConnection(demo).createStatement();
return stmt;
}
}
package baor.test.sql;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
public class ConnUtil {
Connection connection = null;
public Connection getConnection(ConnPropInfo prop) {
if (connection == null) {
try {
// 调用Class.forName()方法加载驱动程序
Class.forName(prop.getDriver());
System.out.println("成功加载MySQL驱动!");
connection = DriverManager.getConnection(prop.getUrl(),
prop.getUser(), prop.getPwd());
System.out.print("成功连接到数据库!");
} catch (ClassNotFoundException e1) {
System.out.println("找不到MySQL驱动!");
e1.printStackTrace();
} catch (SQLException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
return connection;
}
public void closeConnection(Connection conn) {
try {
if (conn != null) {
conn.close();
conn = null;
System.out.println("连接关闭!!");
}
} catch (SQLException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
}
package baor.src.dbunit;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.sql.Connection;
import java.sql.SQLException;
import org.dbunit.DatabaseUnitException;
import org.dbunit.database.DatabaseConnection;
import org.dbunit.database.IDatabaseConnection;
import org.dbunit.database.QueryDataSet;
import org.dbunit.dataset.IDataSet;
import org.dbunit.dataset.xml.FlatXmlDataSet;
public class DataUtil {
public void simgleTableExport(Connection conn, String tableName,
String xmlFile) {
IDatabaseConnection connection = null;
try {
connection = new DatabaseConnection(conn);
QueryDataSet dataSet = new QueryDataSet(connection);
dataSet.addTable(tableName);
FlatXmlDataSet.write(dataSet, new FileOutputStream(xmlFile));
} catch (DatabaseUnitException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (FileNotFoundException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
public void allTableExport(Connection conn, String fileName) {
IDatabaseConnection connection = null;
try {
connection = new DatabaseConne