JDBC工具类封装v2.0

JDBC工具类封装v2.0

package com.atsyc.api.utils;

/*
 * TODO:
 *      利用线程本地变量，存储连接信息，确保一个线程的多个方法可以获取同一个connection
 *      优势：事务操作的时候 service 和 dao 属于同一个线程，不用再传递参数了
 *      大家都可以调用getConnection自动获取的是相同的连接池
 */

import com.alibaba.druid.pool.DruidDataSourceFactory;

import javax.sql.DataSource;
import java.io.IOException;
import java.io.InputStream;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.Properties;

public class JdbcUtilsV2 {
    private static DataSource dataSource = null;//连接池对象

    //创建一个线程本地变量
    private static ThreadLocal<Connection> tl = new ThreadLocal<>();

    static {
        //初始化连接池对象
        Properties properties = new Properties();
        InputStream ips = JdbcUtils.class.getClassLoader().getResourceAsStream("druid.properties");
        try {
            properties.load(ips);
        } catch (IOException e) {
            throw new RuntimeException(e);
        }

        try {
            dataSource = DruidDataSourceFactory.createDataSource(properties);
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }

    /*
     * 对外提供连接的方法
     * @return
     */

    public static Connection getConnection() throws SQLException {

        //先查看线程本地变量是否存在
        Connection connection = tl.get();

        //第一次查看没有,则从连接池获取
        if(connection == null){
            connection = dataSource.getConnection();
            tl.set(connection);
        }

        return connection;
    }

    //回收连接的方法
    public static void freeConnection() throws SQLException {

        Connection connection = tl.get();
        if(connection != null){
            tl.remove();//清空线程本地变量数据
            connection.setAutoCommit(true);//事务状态回顾false
            connection.close();//回收到连接池即可
        }

    }
}

在BankDao中工具类的使用

package com.atsyc.api.transactionnew;

//bank表的数据库操作方法存储类

import com.atsyc.api.utils.JdbcUtils;
import com.atsyc.api.utils.JdbcUtilsV2;

import java.sql.Connection;
import java.sql.PreparedStatement;

public class BankDao {
    /*
     * 加钱的数据库操作方法（jdbc）
     * account 加钱的账号
     * money 加钱的金额
     */
    public void add(String account,int money) throws Exception {

        Connection connection = JdbcUtilsV2.getConnection();

        String sql = "UPDATE t_bank SET money = money + ? WHERE account = ? ;";
        PreparedStatement statement = connection.prepareStatement(sql);
        statement.setObject(1,money);
        statement.setObject(2,account);
        statement.executeUpdate();
        statement.close();
        System.out.println("加钱成功！");
    }

    /*
     * 减钱的数据库操作方法（jdbc）
     * account 减钱的帐号
     * money 减钱的金额
     */
    public void sub(String account,int money) throws Exception {

        Connection connection = JdbcUtilsV2.getConnection();

        String sql = "UPDATE t_bank SET money = money - ? WHERE account = ? ;";
        PreparedStatement statement = connection.prepareStatement(sql);
        statement.setObject(1,money);
        statement.setObject(2,account);
        statement.executeUpdate();
        statement.close();
        System.out.println("减钱成功！");
    }
}

在BankService中工具类的使用

package com.atsyc.api.transactionnew;

//银行卡业务方法，调用dao方法

import com.atsyc.api.transaction.BankDao;
import com.atsyc.api.utils.JdbcUtilsV2;
import org.junit.Test;

import java.sql.Connection;
import java.sql.DriverManager;

/*
 * TODO：
 *      事物添加是在业务方法中
 *      利用try catch代码块，开启事务和提交事务，还有事务回滚
 *      将connection传入dao层即可，dao只负责使用，不要close();
 */

public class BankService {

    @Test
    public void start() throws Exception {
        transfer("user1","user2",500);
    }

    public void transfer(String addAccount,String subAccount,int money) throws Exception {
        BankDao bankDao = new BankDao();

        Connection connection = JdbcUtilsV2.getConnection();

        //一个事务的最基本要求，必须是同一个连接对象connection

        //一个转账方法属于一个事务（加钱，减钱）

        try{
            //开启事务(从业务层开启)

            //关闭事务提交
            connection.setAutoCommit(false);

            //执行数据库动作
            bankDao.add(addAccount,money);
            System.out.println("----------");
            bankDao.sub(subAccount,money);

            //事务提交
            connection.commit();
        }catch (Exception e) {
            //事务回滚
            connection.rollback();
            //抛出异常信息
            throw e;
        }finally {
            {
                JdbcUtilsV2.freeConnection();
            }
        }
    }
}