ThreadLocal来管理事务

 

ThreadLocal (扩展)

1 ThreadLocal API

ThreadLocal类只有三个方法:

l  void set(T value):保存值;

l  T get():获取值;

l  void remove():移除值。

2 ThreadLocal的内部是Map

ThreadLocal内部其实是个Map来保存数据。虽然在使用ThreadLocal时只给出了值,没有给出键,其实它内部使用了当前线程做为键。

class MyThreadLocal<T> {

    private Map<Thread,T> map = new HashMap<Thread,T>();

    public void set(T value) {

       map.put(Thread.currentThread(), value);

    }

   

    public void remove() {

       map.remove(Thread.currentThread());

    }

   

    public T get() {

       return map.get(Thread.currentThread());

    }

}

Service事务(扩展)

  在Service中使用ThreadLocal来完成事务,为将来学习Spring事务打基础!

1 DAO中的事务

在DAO中处理事务真是“小菜一碟”。

public void xxx() {

   Connection con = null;

   try {

      con = JdbcUtils.getConnection();

      con.setAutoCommitted(false);

      QueryRunner qr = new QueryRunner();

      String sql = …;

      Object[] params = …;

      qr.update(con, sql, params);
     

sql = …;

      Object[] params = …;

      qr.update(con, sql, params);

      con.commit();

} catch(Exception e) {

    try {

       if(con != null) {con.rollback();}

} catch(Exception e) {}

} finally {

    try {

       con.close();

} catch(Exception e) {}

}

}

2 Service才是处理事务的地方

  我们要清楚一件事,DAO中不是处理事务的地方,因为DAO中的每个方法都是对数据库的一次操作,而Service中的方法才是对应一个业务逻辑。也就是说我们需要在Service中的一方法中调用DAO的多个方法,而这些方法应该在一起事务中。

怎么才能让DAO的多个方法使用相同的Connection呢?方法不能再自己来获得Connection,而是由外界传递进去。

public void daoMethod1(Connection con, …) {

}

public void daoMethod2(Connection con, …) {

}

在Service中调用DAO的多个方法时,传递相同的Connection就可以了。

public class XXXService() {

   private XXXDao dao = new XXXDao();

   public void serviceMethod() {

      Connection con = null;

      try {

         con = JdbcUtils.getConnection();

         con.setAutoCommitted(false);

         dao.daoMethod1(con, …);

         dao.doaMethod2(con, …);

         com.commint();

} catch(Exception e) {

   try {

      con.rollback();

} catch(Exception e) {}

} finally {

   try {

     con.close();

   } catch(Exception e) {}

}

}

}

但是,在Service中不应该出现Connection,它应该只在DAO中出现,因为它是JDBC的东西,JDBC的东西是用来连接数据库的,连接数据库是DAO的事儿!!!但是,事务是Service的事儿,不能放到DAO中!!!

3 修改JdbcUtils

我们把对事务的开启和关闭放到JdbcUtils中,在Service中调用JdbcUtils的方法来完成事务的处理,但在Service中就不会再出现Connection这一“禁忌”了。

DAO中的方法不用再让Service来传递Connection了。DAO会主动从JdbcUtils中获取Connection对象,这样,JdbcUtils成为了DAO和Service的中介!

我们在JdbcUtils中添加beginTransaction()和rollbackTransaction(),以及commitTransaction()方法。这样在Service中的代码如下:

public class XXXService() {

   private XXXDao dao = new XXXDao();

   public void serviceMethod() {

   try {

      JdbcUtils.beginTransaction();

      dao.daoMethod1(…);

      dao.daoMethod2(…);

      JdbcUtils.commitTransaction();

} catch(Exception e) {

   JdbcUtils.rollbackTransaction();

}

}

}

DAO

public void daoMethod1(…) {

  Connection con = JdbcUtils.getConnection();

}

public void daoMethod2(…) {

  Connection con = JdbcUtils.getConnection();

}

在Service中调用了JdbcUtils.beginTransaction()方法时,JdbcUtils要做准备好一个已经调用了setAutoCommitted(false)方法的Connection对象,因为在Service中调用JdbcUtils.beginTransaction()之后,马上就会调用DAO的方法,而在DAO方法中会调用JdbcUtils.getConnection()方法。这说明JdbcUtils要在getConnection()方法中返回刚刚准备好的,已经设置了手动提交的Connection对象。

在JdbcUtils中创建一个Connection con属性,当它为null时,说明没有事务!当它不为null时,表示开启了事务。

l  在没有开启事务时,可以调用“开启事务”方法;

l  在开启事务后,可以调用“提交事务”和“回滚事务”方法;

l  getConnection()方法会在con不为null时返回con,再con为null时,从连接池中返回连接。

beginTransaction()

判断con是否为null,如果不为null,就抛出异常!

如果con为null,那么从连接池中获取一个Connection对象,赋值给con!然后设置它为“手动提交”。

getConnection()

判断con是否为null,如果为null说明没有事务,那么从连接池获取一个连接返回;

如果不为null,说明已经开始了事务,那么返回con属性返回。这说明在con不为null时,无论调用多少次getConnection()方法,返回的都是同个Connection对象。

commitTransaction()

判断con是否为null,如果为null,说明没有开启事务就提交事务,那么抛出异常;

如果con不为null,那么调用con的commit()方法来提交事务;

调用con.close()方法关闭连接;

con = null,这表示事务已经结束!

rollbackTransaction()

判断con是否为null,如果为null,说明没有开启事务就回滚事务,那么抛出异常;

如果con不为null,那么调用con的rollback()方法来回滚事务;

调用con.close()方法关闭连接;

con = null,这表示事务已经结束!

JdbcUtils.java

public class JdbcUtils {

    private static DataSource dataSource = new ComboPooledDataSource();

    private static Connection con = null;

 

    public static DataSource getDataSource() {

       return dataSource;

    }

 

    public static Connection getConnection() throws SQLException {

       if(con == null) {

           return dataSource.getConnection();

       }

       return con;

    }

   

    public static void beginTranscation() throws SQLException {

       if(con != null) {

           throw new SQLException("事务已经开启,在没有结束当前事务时,不能再开启事务!");

       }

       con = dataSource.getConnection();

       con.setAutoCommit(false);

    }

   

    public static void commitTransaction() throws SQLException {

       if(con == null) {

           throw new SQLException("当前没有事务,所以不能提交事务!");

       }

       con.commit();

       con.close();

       con = null;

    }

   

    public static void rollbackTransaction() throws SQLException {

       if(con == null) {

           throw new SQLException("当前没有事务,所以不能回滚事务!");

       }

       con.rollback();

       con.close();

       con = null;      

    }

}

4 再次修改JdbcUtils

现在JdbcUtils有个问题,如果有两个线程!第一个线程调用了beginTransaction()方法,另一个线程再调用beginTransaction()方法时,因为con已经不再为null,所以就会抛出异常了。

我们希望JdbcUtils可以多线程环境下被使用!这说明最好的方法是为每个线程提供一个Connection,这样每个线程都可以开启自己的事务了。

还记得ThreadLocal类么?

public class JdbcUtils {

    private static DataSource dataSource = new ComboPooledDataSource();

    private static ThreadLocal<Connection> tl[涛1]  = new ThreadLocal<Connection>();

 

    public static DataSource getDataSource() {

       return dataSource;

    }

 

    public static Connection getConnection() throws SQLException {

       Connection con = tl.get();[涛2] 

       if(con == null) {

           return dataSource.getConnection();

       }

       return con;

    }

   

    public static void beginTranscation() throws SQLException {

       Connection con = tl.get();[涛3] 

       if(con != null[涛4] ) {

           throw new SQLException("事务已经开启,在没有结束当前事务时,不能再开启事务!");

       }

       con = dataSource.getConnection();[涛5] 

       con.setAutoCommit(false);[涛6] 

       tl.set(con);[涛7] 

    }

   

    public static void commitTransaction() throws SQLException {

       Connection con = tl.get();[涛8] 

       if(con == null[涛9] ) {

           throw new SQLException("当前没有事务,所以不能提交事务!");

       }

       con.commit();[涛10] 

       con.close();[涛11] 

       tl.remove();[涛12] 

    }

   

    public static void rollbackTransaction() throws SQLException {

       Connection con = tl.get();

       if(con == null) {

           throw new SQLException("当前没有事务,所以不能回滚事务!");

       }

       con.rollback();

       con.close();

       tl.remove();

    }

}

5 转账示例

public class AccountDao {

    public void updateBalance(String name, double balance) throws SQLException {

       String sql = "update account set balance=balance+? where name=?";

       Connection con = JdbcUtils.getConnection();

       //修改的代码....

    }

}

public class AccountService {

    private AccountDao dao = new AccountDao();

   

    public void transfer(String from, String to, double balance) {

       try {

           JdbcUtils.beginTranscation();

           dao.updateBalance(from, -balance);

           dao.updateBalance(to, balance);

           JdbcUtils.commitTransaction();

       } catch(Exception e) {

           try {

               JdbcUtils.rollbackTransaction();

           } catch (SQLException e1) {

              throw new RuntimeException(e);

           }

       }

    }

}

       AccountService as = new AccountService();

       as.transfer("zs", "ls", 100);


tl其实是一个Map,每个线程都可以来保存自己的Connection

获取当前线程的连接

获取当前线程的连接

如果当前线程已经有了连接,说明已经开启了事务

如果当前线程没有连接,说明还没有开启事务,那么就获取一个连接

设置连接为手动提交

把连接放到tl中,当前线程就有了连接了。

获取当前线程的连接

如果当前线程没有连接,那么就不能提交事务

如果当前线程有连接,那么就提交事务

关闭连接

把连接从tl中移除,当前线程就没有连接了,表示事务结束

原文地址:https://www.cnblogs.com/onlyxue121226/p/6938961.html