package com.qushida.util;
import java.beans.BeanInfo;
import java.beans.Introspector;
import java.beans.PropertyDescriptor;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;
import java.sql.Statement;
import java.text.DateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import javax.sql.DataSource;
import org.apache.log4j.Logger;
import com.mchange.v2.c3p0.ComboPooledDataSource;
/**
* 數據庫操作輔助類
*
* @version 3.0
* @author xiaocaiji
*/
public class DBUtil {
// 設置數據源(使用C3P0數據庫連接池)
private static DataSource dataSource = new ComboPooledDataSource("mysql-config");
private static Logger logger = Logger.getLogger("DBUtil");
private static ThreadLocal<Connection> tl = new ThreadLocal<Connection>();
public static DataSource getDataSource() {
return dataSource;
}
// private static Connection conn;
/**
* 該語句必須是 SQL INSERT、UPDATE 、DELETE 語句
*
* @param sql
* @return
* @throws Exception
*/
public int execute(String sql) throws Exception {
return execute(sql, new Object[] {});
}
/**
* insert語句使用，返回新增數據的主鍵。
*
* @param sql
* @return
*/
public Object execute(String sql, Object[] paramList, boolean falg) throws Exception {
Connection conn = null;
Object o = new Object();
try {
conn = getConnection();
o = this.execute(conn, sql, paramList, falg);
} catch (Exception e) {
logger.info(e.getMessage());
throw new Exception(e);
} finally {
closeConn(conn);
}
return o;
}
/**
* insert語句使用，返回新增數據的主鍵。
*
* @param sql
* @return
*/
public Object execute(Connection conn, String sql, Object[] paramList, boolean falg) throws Exception {
if (sql == null || sql.trim().equals("")) {
logger.info("parameter is valid!");
}
PreparedStatement pstmt = conn.prepareStatement(sql, Statement.RETURN_GENERATED_KEYS);
Object id = null;
try {
// 指定返回生成的主鍵
// 如果使用靜態的SQL，則不需要動態插入參數
setPreparedStatementParam(pstmt, paramList);
if (pstmt == null) {
return -1;
}
pstmt.executeUpdate();
// 檢索由於執行此 Statement 對象而創建的所有自動生成的鍵
ResultSet rs = pstmt.getGeneratedKeys();
if (rs.next()) {
id = rs.getObject(1);
System.out.println("數據主鍵地址：" + id);
}
} catch (Exception e) {
logger.info(e.getMessage());
throw new Exception(e);
} finally {
closeStatement(pstmt);
}
return id;
}
/**
* 該語句必須是 SQL INSERT、UPDATE 、DELETE 語句 insert into table values(?,?,?,?)
*
* @param sql
* @param paramList：參數，與SQL語句中的佔位符一
* @return
* @throws Exception
*/
public int execute(String sql, Object[] paramList) throws Exception {
if (sql == null || sql.trim().equals("")) {
logger.info("parameter is valid!");
}
Connection conn = null;
PreparedStatement pstmt = null;
int result = 0;
try {
conn = getConnection();
pstmt = DBUtil.getPreparedStatement(conn, sql);
setPreparedStatementParam(pstmt, paramList);
if (pstmt == null) {
return -1;
}
result = pstmt.executeUpdate();
} catch (Exception e) {
logger.info(e.getMessage());
throw new Exception(e);
} finally {
closeStatement(pstmt);
closeConn(conn);
}
return result;
}
/**
* 事物處理類
*
* @param connection
* @param sql
* @param paramList：參數，與SQL語句中的佔位符一
* @return
* @throws Exception
*/
public int execute(Connection conn, String sql, Object[] paramList) throws Exception {
if (sql == null || sql.trim().equals("")) {
logger.info("parameter is valid!");
}
PreparedStatement pstmt = null;
int result = 0;
try {
pstmt = DBUtil.getPreparedStatement(conn, sql);
setPreparedStatementParam(pstmt, paramList);
if (pstmt == null) {
return -1;
}
result = pstmt.executeUpdate();
} catch (Exception e) {
logger.info(e.getMessage());
throw new Exception(e);
} finally {
closeStatement(pstmt);
}
return result;
}
/**
* 獲取實體類型的方法，type爲實體類類型。
*
* @param type
* @param sql
* @param paramList
* @return
* @throws Exception
*/
public Object getObject(Class<?> type, String sql, Object[] paramList) throws Exception {
BeanInfo beanInfo = Introspector.getBeanInfo(type);
Object obj = type.newInstance();
PropertyDescriptor[] propertyDescriptors = beanInfo.getPropertyDescriptors();
Map map = getObject(sql, paramList);
if (map != null) {
for (int i = 0; i < propertyDescriptors.length; i++) {
PropertyDescriptor descriptor = propertyDescriptors[i];
String propertyName = descriptor.getName();
if (map != null && map.containsKey(propertyName)) {
Object value = map.get(propertyName);
Object[] args = new Object[1];
args[0] = value;
try {
descriptor.getWriteMethod().invoke(obj, args);
} catch (Exception e) {
logger.info("檢測一下Table列，和實體類屬性：" + propertyName + "" + "是否一致，並且是否是" + value.getClass() + "類型");
throw new Exception(
"檢測一下Table列，和實體類屬性：" + propertyName + "" + "是否一致，並且是否是" + value.getClass() + "類型");
}
}
}
} else {
obj = null;
}
return obj;
}
public List<Class<?>> getQueryList(Class<?> type, String sql, Object[] paramList) throws Exception {
BeanInfo beanInfo = Introspector.getBeanInfo(type);
PropertyDescriptor[] propertyDescriptors = beanInfo.getPropertyDescriptors();
List<Map<String, Object>> list = getQueryList(sql, paramList);
List beanList = new ArrayList();
for (Iterator iterator = list.iterator(); iterator.hasNext();) {
Map<String, Object> map = (Map<String, Object>) iterator.next();
Object obj = type.newInstance();
for (int i = 0; i < propertyDescriptors.length; i++) {
PropertyDescriptor descriptor = propertyDescriptors[i];
String propertyName = descriptor.getName();
if (map != null && map.containsKey(propertyName)) {
Object value = map.get(propertyName);
Object[] args = new Object[1];
args[0] = value;
try {
descriptor.getWriteMethod().invoke(obj, args);
} catch (Exception e) {
logger.info("檢測一下Table列，和實體類屬性：" + propertyName + "" + "是否一致，並且是否是" + value.getClass() + "類型");
throw new Exception(
"檢測一下Table列，和實體類屬性：" + propertyName + "" + "是否一致，並且是否是" + value.getClass() + "類型");
}
}
}
beanList.add(obj);
}
return beanList;
}
/**
* 將查詢數據庫獲得的結果集轉換爲Map對象
*
* @param sql：查詢
* @return
*/
public List<Map<String, Object>> getQueryList(String sql) throws Exception {
return getQueryList(sql, new Object[] {});
}
/**
* 將查詢數據庫獲得的結果集轉換爲Map對象
*
* @param sql：查詢
* @param paramList：參數
* @return
*/
public List<Map<String, Object>> getQueryList(String sql, Object[] paramList) throws Exception {
if (sql == null || sql.trim().equals("")) {
logger.info("parameter is valid!");
return null;
}
Connection conn = null;
PreparedStatement pstmt = null;
ResultSet rs = null;
List<Map<String, Object>> queryList = null;
try {
conn = getConnection();
pstmt = DBUtil.getPreparedStatement(conn, sql);
setPreparedStatementParam(pstmt, paramList);
if (pstmt == null) {
return null;
}
rs = getResultSet(pstmt);
queryList = getQueryList(rs);
} catch (RuntimeException e) {
logger.info(e.getMessage());
System.out.println("parameter is valid!");
throw new Exception(e);
} finally {
closeResultSet(rs);
closeStatement(pstmt);
closeConn(conn);
}
return queryList;
}
/**
* 分頁查詢
*
* @param sql
* @param params
* 查詢條件參數
* @param page
* 分頁信息
* @return
*/
public Page getQueryPage(Class<?> type, String sql, Object[] params, Page page) {
int totalPages = 0; // 頁數
Long rows = 0l;// 數據記錄數
// 分頁工具類
List<Class<?>> list = null;
Map countMap = null;
try {
list = this.getQueryList(type,
sql + " limit " + (page.getCurPage() - 1) * page.getPageNumber() + " , " + page.getPageNumber(),
params);
countMap = this.getObject(" " + "select count(*) c from (" + sql + ") as t ", params);
rows = (Long) countMap.get("c");
// 求餘數
if (rows % page.getPageNumber() == 0) {
totalPages = rows.intValue() / page.getPageNumber();
} else {
totalPages = rows.intValue() / page.getPageNumber() + 1;
}
page.setRows(rows.intValue());
page.setData(list);
page.setTotalPage(totalPages);
} catch (Exception e) {
e.printStackTrace();
}
return page;
}
/**
* 分頁查詢
*
* @param sql
* @param params
* 查詢條件參數
* @param page
* 分頁信息
* @return
*/
public Page getQueryPage(String sql, Object[] params, Page page) {
int totalPages = 0; // 頁數
Long rows = 0l;// 數據記錄數
// 分頁工具類
List<Map<String, Object>> list = null;
Map countMap = null;
try {
list = this.getQueryList(
sql + " limit " + (page.getCurPage() - 1) * page.getPageNumber() + " , " + page.getPageNumber(),
params);
countMap = this.getObject(" " + "select count(*) c from (" + sql + ") as t ", params);
rows = (Long) countMap.get("c");
// 求餘數
if (rows % page.getPageNumber() == 0) {
totalPages = rows.intValue() / page.getPageNumber();
} else {
totalPages = rows.intValue() / page.getPageNumber() + 1;
}
page.setRows(rows.intValue());
page.setData(list);
page.setTotalPage(totalPages);
} catch (Exception e) {
e.printStackTrace();
}
return page;
}
/**
* 將查詢數據庫獲得的結果集轉換爲Map對象
*
* @param sql：查詢
* @return
*/
public Map<String, Object> getObject(String sql) throws Exception {
return getObject(sql, new Object[] {});
}
/**
* 將查詢數據庫獲得的結果集轉換爲Map對象
*
* @param sql：查詢
* @param paramList：參數
* @return
*/
public Map<String, Object> getObject(String sql, Object[] paramList) throws Exception {
if (sql == null || sql.trim().equals("")) {
logger.info("parameter is valid!");
return null;
}
Connection conn = null;
PreparedStatement pstmt = null;
ResultSet rs = null;
Map map = new HashMap<String, Object>();
try {
conn = getConnection();
pstmt = DBUtil.getPreparedStatement(conn, sql);
setPreparedStatementParam(pstmt, paramList);
if (pstmt == null) {
return null;
}
rs = getResultSet(pstmt);
List list = getQueryList(rs);
if (list.isEmpty()) {
return null;
}
map = (HashMap) list.get(0);
} catch (RuntimeException e) {
logger.info(e.getMessage());
logger.info("parameter is valid!");
throw new Exception(e);
} finally {
closeResultSet(rs);
closeStatement(pstmt);
closeConn(conn);
}
return map;
}
private static PreparedStatement getPreparedStatement(Connection conn, String sql) throws Exception {
if (conn == null || sql == null || sql.trim().equals("")) {
return null;
}
PreparedStatement pstmt = conn.prepareStatement(sql.trim());
return pstmt;
}
private void setPreparedStatementParam(PreparedStatement pstmt, Object[] paramList) throws Exception {
if (pstmt == null || paramList == null) {
return;
}
DateFormat df = DateFormat.getDateTimeInstance();
for (int i = 0; i < paramList.length; i++) {
// -
if (paramList[i] instanceof Integer) {
int paramValue = ((Integer) paramList[i]).intValue();
pstmt.setInt(i + 1, paramValue);
} else if (paramList[i] instanceof Float) {
float paramValue = ((Float) paramList[i]).floatValue();
pstmt.setFloat(i + 1, paramValue);
} else if (paramList[i] instanceof Double) {
double paramValue = ((Double) paramList[i]).doubleValue();
pstmt.setDouble(i + 1, paramValue);
} else if (paramList[i] instanceof Date) {
pstmt.setString(i + 1, df.format((Date) paramList[i]));
} else if (paramList[i] instanceof Long) {
long paramValue = ((Long) paramList[i]).longValue();
pstmt.setLong(i + 1, paramValue);
} else if (paramList[i] instanceof String) {
pstmt.setString(i + 1, (String) paramList[i]);
}
// = pstmt.setObject(i + 1, paramList[i]);
}
return;
}
/**
* 獲得數據庫查詢結果集
*
* @param pstmt
* @return
* @throws Exception
*/
private ResultSet getResultSet(PreparedStatement pstmt) throws Exception {
if (pstmt == null) {
return null;
}
ResultSet rs = pstmt.executeQuery();
return rs;
}
/**
* @param rs
* @return
* @throws Exception
*/
private List<Map<String, Object>> getQueryList(ResultSet rs) throws Exception {
if (rs == null) {
return null;
}
ResultSetMetaData rsMetaData = rs.getMetaData();
int columnCount = rsMetaData.getColumnCount();
List<Map<String, Object>> dataList = new ArrayList<Map<String, Object>>();
while (rs.next()) {
Map<String, Object> dataMap = new HashMap<String, Object>();
for (int i = 0; i < columnCount; i++) {
dataMap.put(rsMetaData.getColumnLabel(i + 1), rs.getObject(i + 1));
}
dataList.add(dataMap);
}
return dataList;
}
/**
* 關閉數據庫
*
* @param conn
*/
private void closeConn(Connection conn) {
if (conn == null) {
return;
}
try {
conn.close();
} catch (SQLException e) {
logger.info(e.getMessage());
}
}
/**
* 關閉
*
* @param stmt
*/
private void closeStatement(Statement stmt) {
if (stmt == null) {
return;
}
try {
stmt.close();
} catch (SQLException e) {
logger.info(e.getMessage());
}
}
/**
* 關閉
*
* @param rs
*/
private void closeResultSet(ResultSet rs) {
if (rs == null) {
return;
}
try {
rs.close();
} catch (SQLException e) {
logger.info(e.getMessage());
}
}
/**
* 可以選擇三個不同的數據庫連接
*
* @param JDBC
* ,JNDI（依賴web容器 DBCP
* @return
* @throws Exception
*/
public static Connection getConnection() throws Exception {
Connection conn = tl.get();
if (conn == null) {
conn = dataSource.getConnection();
}
return conn;
}
/*********** 事務處理方法 ************/
/**
* 開啓事務
*/
public static void beginTranscation() throws Exception {
Connection conn = tl.get();
if (conn != null) {
logger.info("事務已經開始！");
throw new SQLException("事務已經開始！");
}
conn = dataSource.getConnection();
conn.setAutoCommit(false);
tl.set(conn);
}
/**
* 結束事務
*
* @throws SQLException
*/
public static void endTranscation() throws SQLException {
Connection conn = tl.get();
if (conn == null) {
logger.info("當前沒有事務！");
throw new SQLException("當前沒有事務！");
}
conn.commit();
}
/**
* 回滾
*
* @throws SQLException
*/
public static void rollback() throws SQLException {
Connection conn = tl.get();
if (conn == null) {
logger.info("當前沒有事務,不能回滾！");
throw new SQLException("當前沒有事務,不能回滾！");
}
conn.rollback();
}
/**
* 事務處理，關閉資源
*
* @throws SQLException
*/
public static void closeConn() throws SQLException {
Connection conn = tl.get();
if (conn == null) {
logger.info("當前沒有連接，不需要關閉Connection。");
throw new SQLException("當前沒有連接，不需要關閉Connection。");
}
conn.close();
tl.remove();
}
}
