1. 程式人生 > >Java常用工具類封裝——連線資料庫的工具類

Java常用工具類封裝——連線資料庫的工具類

資料庫操作工具類,供大家參考。

package com.itjh.javaUtil;
 
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
 
import org.apache.commons.dbcp.ConnectionFactory;
import org.apache.commons.dbcp.DriverManagerConnectionFactory;
import org.apache.commons.dbcp.PoolableConnectionFactory;
import org.apache.commons.dbcp.PoolingDriver;
import org.apache.commons.dbutils.DbUtils;
import org.apache.commons.dbutils.QueryRunner;
import org.apache.commons.dbutils.handlers.MapListHandler;
import org.apache.commons.pool.ObjectPool;
import org.apache.commons.pool.impl.GenericObjectPool;
 
/**
* 連線資料庫的綜合類。</br>
* 依賴jar包:commons.dbcp-1.4,commons.dbutils-1.3,commons.pool-1.5.4包。
*
* @author jiangshuai
* @date 2016年10月03日
*/
 
public class DBUtil {
 
    private String dri = null;
    private String url = null;
    private String username = null;
    private String password = null;
    private String poolName = null; // 連線池名稱
    private ObjectPool connectionPool = null; // 連線池
    // 對應的定時查詢類
    private QueryThread queryThread = null;
 
    /**
     * 功能:建構函式
     *
     * @author jiangshuai
     * @date 2016年10月03日
     * @param dri
     *            驅動全類名,例如:com.mysql.jdbc.Driver。
     * @param url
     *            資料庫url連線,例如:
     *            "jdbc:mysql://127.0.0.1:3306/test?autoReconnect=true&useUnicode=true&characterEncoding=UTF-8"
     * @param userName
     *            資料庫使用者名稱,例如:root
     * @param password
     *            資料庫密碼,例如:abc
     * @param poolName
     *            建立的資料庫連線池的名稱,例如mypool,注意一個web容器此名稱不能重複。
     */
    public DBUtil(String dri, String url, String userName, String password,
            String poolName) {
        this.dri = dri;
        this.url = url;
        this.username = userName;
        this.password = password;
        this.poolName = poolName;
    }
 
    /**
     * 執行sql。
     *
     * @param conn
     *            連線
     * @param pstm
     *            PreparedStatement
     * @return int 執行sql對應的影響行。
     * @throws SQLException
     * @author jiangshuai
     * @date 2016年10月03日
     */
    public int execute(Connection conn, PreparedStatement pstm)
            throws SQLException {
        try {
            return pstm.executeUpdate();
        } finally {
            Close(conn);
        }
    }
 
    /**
     * 查詢sql。
     *
     * @param conn
     *            連線
     * @param pstm
     *            PreparedStatement
     * @return List<Map<String,Object>> 查詢的結果集
     * @throws SQLException
     * @author jiangshuai
     * @date 2016年10月03日
     */
    public List<Map<String, Object>> query(Connection conn,
            PreparedStatement pstm) throws SQLException {
        try {
            return resultSetToList(pstm.executeQuery());
        } finally {
            Close(conn);
        }
    }
 
    /**
     * 功能:ResultSet 轉為List<Map<String,Object>>
     *
     *
     * @param rs
     *            ResultSet 原始資料集
     * @return List<Map<String,Object>>
     * @throws java.sql.SQLException
     * @author jiangshuai
     * @date 2016年10月03日
     */
    private List<Map<String, Object>> resultSetToList(ResultSet rs)
            throws java.sql.SQLException {
        if (rs == null)
            return Collections.EMPTY_LIST;
 
        ResultSetMetaData md = rs.getMetaData(); // 得到結果集(rs)的結構資訊,比如欄位數、欄位名等
        int columnCount = md.getColumnCount(); // 返回此 ResultSet 物件中的列數
        List<Map<String, Object>> list = new ArrayList<Map<String, Object>>();
        Map<String, Object> rowData = new HashMap<String, Object>();
        while (rs.next()) {
            rowData = new HashMap<String, Object>(columnCount);
            for (int i = 1; i <= columnCount; i++) {
                rowData.put(md.getColumnName(i), rs.getObject(i));
            }
            list.add(rowData);
        }
        return list;
    }
 
    /**
     * 查詢sql語句。
     *
     * @param sql
     *            被執行的sql語句
     * @return List<Map<String,Object>>
     * @throws SQLException
     * @author jiangshuai
     * @date 2016年10月03日
     */
    public List<Map<String, Object>> query(String sql) throws SQLException {
        List<Map<String, Object>> results = null;
        Connection conn = null;
        try {
            conn = getConnection();
            QueryRunner qr = new QueryRunner();
            results = qr.query(conn, sql, new MapListHandler());
        } finally {
            Close(conn);
        }
        return results;
    }
 
    /**
     * 根據引數查詢sql語句
     *
     * @param sql
     *            sql語句
     * @param param
     *            引數
     * @return List<Map<String,Object>>
     * @throws SQLException
     * @author jiangshuai
     * @date 2016年10月03日
     */
    public List<Map<String, Object>> query(String sql, Object param)
            throws SQLException {
        List<Map<String, Object>> results = null;
        Connection conn = null;
        try {
            conn = getConnection();
            QueryRunner qr = new QueryRunner();
            results = (List<Map<String, Object>>) qr.query(conn, sql, param,
                    new MapListHandler());
        } catch (SQLException e) {
            e.printStackTrace();
        } finally {
            Close(conn);
        }
        return results;
    }
 
    /**
     * 執行sql語句
     *
     * @param sql
     *            被執行的sql語句
     * @return 受影響的行
     * @throws Exception
     * @author jiangshuai
     * @date 2016年10月03日
     */
    public int execute(String sql) throws Exception {
        Connection conn = getConnection();
        int rows = 0;
        try {
            QueryRunner qr = new QueryRunner();
            rows = qr.update(conn, sql);
        } finally {
            Close(conn);
        }
        return rows;
    }
 
    /**
     * 執行含引數的sql語句
     *
     * @param sql
     *            被執行的sql語句
     * @param params
     *            引數
     * @return 返回受影響的行
     * @throws Exception
     * @author jiangshuai
     * @date 2016年10月03日
     */
    public int execute(String sql, Object[] params) throws Exception {
        Connection conn = getConnection();
        int rows = 0;
        try {
            QueryRunner qr = new QueryRunner();
            rows = qr.update(conn, sql, params);
        } finally {
            Close(conn);
        }
        return rows;
    }
 
    /**
     * 關閉連線
     *
     * @param conn
     * @throws SQLException
     * @author jiangshuai
     * @date 2016年10月03日
     */
    public void Close(Connection conn) throws SQLException {
        if (conn != null) {
            conn.close();
        }
        DbUtils.closeQuietly(conn);
    }
 
    /**
     * 啟動連線池
     *
     * @author jiangshuai
     * @date 2016年10月03日
     */
    private void StartPool() {
        try {
            Class.forName(dri);
        } catch (ClassNotFoundException e1) {
            e1.printStackTrace();
        }
        if (connectionPool != null) {
            ShutdownPool();
        }
        try {
            connectionPool = new GenericObjectPool(null);
            ConnectionFactory connectionFactory = new DriverManagerConnectionFactory(
                    url, username, password);
            PoolableConnectionFactory poolableConnectionFactory = new PoolableConnectionFactory(
                    connectionFactory, connectionPool, null, "SELECT 1", false,
                    true);
            Class.forName("org.apache.commons.dbcp.PoolingDriver");
            PoolingDriver driver = (PoolingDriver) DriverManager
                    .getDriver("jdbc:apache:commons:dbcp:");
            driver.registerPool(poolName, poolableConnectionFactory.getPool());
 
        } catch (Exception e) {
            e.printStackTrace();
        }
        // 開啟查詢程式
        queryThread = new QueryThread(this);
        queryThread.start();
    }
 
    /**
     * 關閉連線池
     *
     * @author jiangshuai
     * @date 2016年10月03日
     */
    private void ShutdownPool() {
        try {
            PoolingDriver driver = (PoolingDriver) DriverManager
                    .getDriver("jdbc:apache:commons:dbcp:");
            driver.closePool(poolName);
            // 關閉定時查詢
            queryThread.setStartQuery(false);
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }
 
    /**
     * 得到一個連線
     *
     * @return
     * @author jiangshuai
     * @date 2016年10月03日
     */
    public synchronized Connection getConnection() {
        Connection conn = null;
        try {
            if (connectionPool == null)
                StartPool();
            conn = DriverManager.getConnection("jdbc:apache:commons:dbcp:"
                    + poolName);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return conn;
    }
}
 
/**
* 當連線池啟動後會自動定時查詢資料庫,防止資料庫連線超時。
*
* @author jiangshuai
* @date 2016年10月03日
*/
class QueryThread extends Thread {
 
    private DBUtil dbUtil = null;
    // 是否開啟查詢
    private boolean startQuery = true;
 
    /**
     * 功能:對應的資料庫連線。
     *
     * @author jiangshuai
     * @date 2016年10月03日
     * @param dbUtil
     *            資料庫連線
     */
    public QueryThread(DBUtil dbUtil) {
        this.dbUtil = dbUtil;
    }
 
    public void run() {
        while (true) {
            try {
                if (startQuery) {
                    this.dbUtil.query("select 1");
                }
                // System.out.println(startQuery+"   123");
            } catch (Exception e) {
                e.printStackTrace();
            } finally {
                try {
                    Thread.sleep(120000);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        }
    }
 
    public void setStartQuery(boolean startQuery) {
        // System.out.println("startQuery shut:"+startQuery);
        this.startQuery = startQuery;
    }
}