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;

}

}