管理数据库连接和事务的工具类

import java.sql.Connection;
import java.sql.SQLException;

import javax.sql.DataSource;

import com.mchange.v2.c3p0.ComboPooledDataSource;
import com.mchange.v2.c3p0.DataSources;

/**
 * 这个工具类就是来管理数据库连接和事务的
 * 
 * @author user
 * 
 */
public class JDBCUtil {

    // 数据连接池对象
    private static ComboPooledDataSource dataSource = new ComboPooledDataSource();;
    // 事务专用连接,把事务存入到ThreadLocal中
    private static ThreadLocal<Connection> tlConn = new ThreadLocal<Connection>();

    /**
     * 通过数据连接池获取连接
     * 
     * @throws SQLException
     */
    public static Connection getConnection() throws SQLException {
        // 如果conn不为空,证明一件调用了beginTransaction方法,事务已经开启
        Connection conn = tlConn.get();
        if (conn != null)
            return conn;
        return dataSource.getConnection();
    }

    /**
     * 获取数据连接池对象
     */
    public static DataSource getDataSource() {
        return dataSource;
    }

    /**
     * 开启事务 获取Connection,设置它的setAutoCommit(false)
     * 
     * @throws SQLException
     */
    public static void beginTransaction() throws SQLException {
        Connection conn = tlConn.get();
        if (conn != null) {
            throw new SQLException("事务已经开启,不能重复开启事务!");
        }
        // 给conn赋值
        conn = getConnection();
        // 把conn的事务设置为手动提交
        conn.setAutoCommit(false);
        // 把连接存入到ThreadLocal中
        tlConn.set(conn);
    }

    /**
     * 提交事务
     * 
     * @throws SQLException
     */
    public static void commitTransaction() throws SQLException {
        Connection conn = tlConn.get();
        if (conn == null) {
            throw new SQLException("还没有开启事务,不能提交!");
        }
        // 提交事务
        conn.commit();
        // 把连接归还,并且赋值为NULL,是为了保证下次获取,不再是开始事务的连接
        conn.close();
        tlConn.remove();
    }

    /**
     * 事务回滚
     * 
     * @throws SQLException
     */
    public static void rollbackTransaction() throws SQLException {
        Connection conn = tlConn.get();
        if (conn == null) {
            throw new SQLException("还没有开启事务,不能回滚!");
        }
        conn.rollback();
        conn.close();
        tlConn.remove();
    }

    /**
     * 释放连接
     * 
     * @throws SQLException
     */
    public static void closeConnection(Connection connection)
            throws SQLException {
        // 判断参数连接是否为事务连接,是就不关闭,不是再关闭
        // 如果conn为NULL,证明没有开启事务,那么就关闭参数的connection
        Connection conn = tlConn.get();
        if (conn == null) {
            connection.close();
        } else if (conn != connection) {
            // 如果conn不为NULL,但是与connection不是同一个连接,证明参数不是事务专用连接,那么也关闭connection
            connection.close();
        }
    }

}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值