import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;

/**
 * 鏈接數(shù)據(jù)庫(kù)的工具類
 *
 * @author 徐銳
 *
 */
public class DBHelper {

 // mysql驅(qū)動(dòng)路徑
 private static final String driver = "com.mysql.jdbc.Driver";
 // 數(shù)據(jù)庫(kù)的連接路徑
 private static final String url = "jdbc:mysql://localhost:3306/snsteam6";

 /**
  * 連接數(shù)據(jù)庫(kù)的對(duì)象
  *
  * @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;
 }

 /**
  * 關(guān)閉打開的資源
  *
  * @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();
  }
 }

 /**
  * 獲取結(jié)果集對(duì)象
  *
  * @param sql
  * @param params
  * @return
  */
 public static ResultSet resultSet(String sql, Object[] params) {

  // 數(shù)據(jù)庫(kù)的鏈接對(duì)象
  Connection conn = null;
  // 數(shù)據(jù)庫(kù)的操作對(duì)象
  PreparedStatement pst = null;
  // 結(jié)果對(duì)象
  ResultSet rst = null;
  try {

   // 數(shù)據(jù)庫(kù)的鏈接對(duì)象
   conn = DBHelper.getConnection();
   // 數(shù)據(jù)庫(kù)的操作對(duì)象
   pst = conn.prepareStatement(sql);
   // 判斷是否有參數(shù)
   if (params != null && params.length > 0) {
    for (int i = 0; i < params.length; i++) {
     // 給操作對(duì)象賦值
     pst.setObject(i + 1, params[i]);
    }
   }
   // 獲取結(jié)果對(duì)象
   rst = pst.executeQuery();
  } catch (SQLException e) {
   rst = null;
   e.printStackTrace();
  }
  return rst;
 }

 /**
  * 獲取添加的結(jié)果對(duì)象
  *
  * @param sql
  * @param params
  * @return
  */
 public static boolean result(String sql, Object[] params) {

  boolean flag = false;
  // 數(shù)據(jù)庫(kù)的鏈接對(duì)象
  Connection conn = null;
  // 數(shù)據(jù)庫(kù)的操作對(duì)象
  PreparedStatement pst = null;
  try {
   // 數(shù)據(jù)庫(kù)的鏈接對(duì)象
   conn = DBHelper.getConnection();
   // 數(shù)據(jù)庫(kù)的操作對(duì)象
   pst = conn.prepareStatement(sql);
   // 判斷是否有參數(shù)
   if (params != null && params.length > 0) {
    for (int i = 0; i < params.length; i++) {
     // 給操作對(duì)象賦值
     pst.setObject(i + 1, params[i]);
    }
   }
   // 獲取結(jié)果對(duì)象,是int類型，表示執(zhí)行成功的行數(shù)
   int row = pst.executeUpdate();
   if (row > 0) {
    // 如果大于0，表示執(zhí)行成功
    flag = true;
   }
  } catch (SQLException e) {
   flag = false;
   e.printStackTrace();
  }
  return flag;
 }
}