日期:2014-05-16 浏览次数:20444 次
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