複製代碼代碼如下:
package sns.team6.util;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
/**
* 連結資料庫的工具類
*
* @author 徐銳
*
*/
public class DBHelper {
// mysql驅動路徑
private static final String driver = "com.mysql.jdbc.Driver";
// 資料庫的連線路徑
private static final String url = "jdbc:mysql://localhost:3306/snsteam6";
/**
* 連接資料庫的對象
*
* @return
*/
public static Connection getConnection() {
Connection conn = null;
try {
Class.forName(driver);
conn = DriverManager.getConnection(url, "root", "root");
} catch (Exception e) {
e.printStackTrace();
}
return conn;
}
/**
* 關閉開啟的資源
*
* @param conn
* @param pst
* @param rst
*/
public static void closeInfo(Connection conn, PreparedStatement pst,
ResultSet rst) {
try {
if (rst != null) {
rst.close();
rst = null;
}
if (pst != null) {
pst.close();
pst = null;
}
if (conn != null) {
conn.close();
conn = null;
}
} catch (Exception e) {
e.printStackTrace();
}
}
/**
* 取得結果集對象
*
* @param sql
* @param params
* @return
*/
public static ResultSet resultSet(String sql, Object[] params) {
// 資料庫的連結對象
Connection conn = null;
// 資料庫的操作對象
PreparedStatement pst = null;
// 結果對象
ResultSet rst = null;
try {
// 資料庫的連結對象
conn = DBHelper.getConnection();
// 資料庫的操作對象
pst = conn.prepareStatement(sql);
// 判斷是否有參數
if (params != null && params.length > 0) {
for (int i = 0; i < params.length; i++) {
// 給操作物件賦值
pst.setObject(i + 1, params[i]);
}
}
// 取得結果對象
rst = pst.executeQuery();
} catch (SQLException e) {
rst = null;
e.printStackTrace();
}
return rst;
}
/**
* 取得新增的結果對象
*
* @param sql
* @param params
* @return
*/
public static boolean result(String sql, Object[] params) {
boolean flag = false;
// 資料庫的連結對象
Connection conn = null;
// 資料庫的操作對象
PreparedStatement pst = null;
try {
// 資料庫的連結對象
conn = DBHelper.getConnection();
// 資料庫的操作對象
pst = conn.prepareStatement(sql);
// 判斷是否有參數
if (params != null && params.length > 0) {
for (int i = 0; i < params.length; i++) {
// 給操作物件賦值
pst.setObject(i + 1, params[i]);
}
}
// 取得結果物件,是int型，表示執行成功的行數
int row = pst.executeUpdate();
if (row > 0) {
// 如果大於0，表示執行成功
flag = true;
}
} catch (SQLException e) {
flag = false;
e.printStackTrace();
}
return flag;
}
}