JBDC工具类

Posted 2020-10-06

package Util;

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

import javax.sql.DataSource;

import com.mchange.v2.c3p0.ComboPooledDataSource;

/**
 * 使用本类的方法，必须提供c3p0-copnfig.xml文件
 * @author pplsunny
 */
public class C3P0Util {
    
    // 饿汉式
    private static DataSource ds = new ComboPooledDataSource();
    
    /**
     * 它为null表示没有事务
     * 它不为null表示有事务
     * 当开启事务时，需要给它赋值
     * 当结束事务时，需要给它赋值为null
     * 并且在开启事务时，让dao的多个方法共享这个Connection
     */
    private static ThreadLocal<Connection> tl = new ThreadLocal<Connection>();
    
    public static DataSource getDataSource() {
        return ds;
    }
    
    /**
     * dao使用本方法来获取连接
     * @return
     * @throws SQLException
     */
    public static Connection getConnection()  {
        /*
         * 如果有事务，返回当前事务的con
         * 如果没有事务，通过连接池返回新的con
         */
        Connection con = tl.get();//获取当前线程的事务连接
        if(con != null) return con;
        try {
            return ds.getConnection();
        } catch (SQLException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        return con;
    }
    
    /**
     * 开启事务
     * @throws SQLException 
     */
    public static void beginTransaction() throws SQLException {
        Connection con = tl.get();//获取当前线程的事务连接
        if(con != null) throw new SQLException("已经开启了事务，不能重复开启！");
        con = ds.getConnection();//给con赋值，表示开启了事务
        con.setAutoCommit(false);//设置为手动提交
        tl.set(con);//把当前事务连接放到tl中
    }
    
    /**
     * 提交事务
     * @throws SQLException 
     */
    public static void commitTransaction() throws SQLException {
        Connection con = tl.get();//获取当前线程的事务连接
        if(con == null) throw new SQLException("没有事务不能提交！");
        con.commit();//提交事务
        con.close();//关闭连接
        con = null;//表示事务结束！
        tl.remove();
    }
    
    /**
     * 回滚事务
     * @throws SQLException 
     */
    public static void rollbackTransaction() throws SQLException {
        Connection con = tl.get();//获取当前线程的事务连接
        if(con == null) throw new SQLException("没有事务不能回滚！");
        con.rollback();
        con.close();
        con = null;
        tl.remove();
    }
    
    /**
     * 利用不定参接受需要关闭的对象
     * 释放Connection，PrepeareStatement, ResultSet.
     * @param con pStmt rs
     * @throws SQLException 
     */
    public static void releaseConnection(Object...object) throws SQLException {
        for(Object T: object){
            if(T instanceof Connection){
                Connection con = tl.get();//获取当前线程的事务连接
                if(T != con) {//如果参数连接，与当前事务连接不同，说明这个连接不是当前事务，可以关闭！
                    if(T != null &&!((Connection) T).isClosed()) {//如果参数连接没有关闭，关闭之！
                        ((Connection)T).close();
                    }
                }
            }
            if(T instanceof PreparedStatement){
                if(T != null){
                    ((PreparedStatement) T).close();
                    T=null;
                }
            }
            if(T instanceof ResultSet){
                if(T != null){
                    ((ResultSet) T).close();
                    T=null;
                }
            }
        }
    }
}