import org.testng.Assert;
import org.testng.annotations.Test;
import java.sql.*;
public class DButil {
public static Connection getConnection(String urlP,String userName,String pwd,String driverName){
Connection conn=null;
try {
String url = urlP;
String user= userName;
String password= pwd;
Class.forName(driverName);//加载数据驱动
conn = DriverManager.getConnection(url, user, password);// 连接数据库
} catch (ClassNotFoundException e) {
Assert.assertTrue(false,"加载数据库驱动失败:" + e.getMessage());
e.printStackTrace();
System.out.println("加载数据库驱动失败");
}catch(Exception e){
Assert.assertTrue(false,"连接数据库失败:" + e.getMessage());
e.printStackTrace();
System.out.println("连接数据库失败");
}
return conn;
}
public static void close(Connection conn, PreparedStatement ps, ResultSet rs){
try {
if(rs!=null){
rs.close();
}
} catch (SQLException e) {
Assert.assertTrue(false,"数据库关闭失败:" + e.getMessage());
e.printStackTrace();
}
try {
if(ps!=null){
ps.close();
}
} catch (SQLException e) {
Assert.assertTrue(false,"数据库关闭失败:" + e.getMessage());
e.printStackTrace();
}
try {
if(conn!=null){
conn.close();
}
} catch (SQLException e) {
Assert.assertTrue(false,"数据库关闭失败:" + e.getMessage());
e.printStackTrace();
}
}
public static void closeState(PreparedStatement preparedStatement ) {
try {
if (preparedStatement != null) {
preparedStatement.close();
}
} catch (SQLException e) {
Assert.assertTrue(false,"数据库关闭失败:" + e.getMessage());
e.printStackTrace();
}
}
public static void closeResult(ResultSet resultSet ) {
try {
if (resultSet != null) {
resultSet.close();
}