Java常用工具类封装——连接数据库的工具类

来源：互联网发布：淘宝卖家网编辑：程序博客网时间：2024/06/09 18:20

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

阅读全文

0 0