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中移除，当前线程就没有连接了，表示事务结束