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;
 }
}