DataSource工具类

package utils;

import com.mchange.v2.c3p0.ComboPooledDataSource;

import javax.sql.DataSource;
import java.sql.Connection;
import java.sql.SQLException;

/**
 * 数据源工具
 */
public class DataSourceUtils {
    /**
     * 创建数据源(c3p0)
     */
    private static DataSource dataSource = new ComboPooledDataSource();
    /**
     * ThreadLocal: 可以为每个线程创建一个副本。每个线程可以访问自己内部的副本
     */
    private static ThreadLocal<Connection> tl = new ThreadLocal<Connection>();

    public static DataSource getDataSource() {
        return dataSource;
    }

    /**
     * 当 DBUtils需要手动控制事务时,调用该方法获得一个连接
     *
     * @return
     * @throws SQLException
     */
    public static Connection getConnection() throws SQLException {
        // 获得一个连接
        Connection con = tl.get();
        // 判断是否第一次取连接,第一次取连接时,ThreadLocal中没有连接
        if (con == null) {
            // 如果是第一次取连接,则这个连接为空,创建一个连接
            con = dataSource.getConnection();
            tl.set(con);
        }
        return con;
    }

    /**
     * 开启事务
     *
     * @throws SQLException
     */
    public static void startTransaction() throws SQLException {
        Connection con = getConnection();
        if (con != null) {
            // 开启事务,true表示自动提交事务
            con.setAutoCommit(false);
        }
    }

    /**
     * 提交事务
     *
     * @throws SQLException
     */
    public static void commitTransaction() throws SQLException {
        Connection con = getConnection();
        if (con != null) {
            con.commit();

        }
    }

    /**
     * 事务回滚
     *
     * @throws SQLException
     */
    public static void rollback() throws SQLException {
        Connection con = getConnection();
        if (con != null) {
            con.rollback();
        }
    }

    /**
     * 从 ThreadLocal中释放并且关闭Connection,并结束事务
     *
     * @throws SQLException
     */
    public static void close() throws SQLException {
        Connection connection = getConnection();
        if (connection != null) {
            // 清空ThreadLocal
            tl.remove();
            // 关闭连接
            connection.close();
            connection = null;
        }
    }

}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值