ThreadLocal使用案例_动力节点Java学院整理
用户提出一个需求：当修改产品价格的时候，需要记录操作日志，什么时候做了什么事情。
想必这个案例，只要是做过应用系统的小伙伴们，都应该遇到过吧？无外乎数据库里就两张表：product与log，用两条SQL语句应该可以解决问题：
updateproductsetprice=?whereid=? insertintolog(created,description)values(?,?)
But！要确保这两条SQL语句必须在同一个事务里进行提交，否则有可能update提交了，但insert却没有提交。如果这样的事情真的发生了，我们肯定会被用户指着鼻子狂骂：“为什么产品价格改了，却看不到什么时候改的呢？”。
聪明的我在接到这个需求以后，是这样做的：
首先，我写一个DBUtil的工具类，封装了数据库的常用操作：
publicclassDBUtil{
//数据库配置
privatestaticfinalStringdriver="com.mysql.jdbc.Driver";
privatestaticfinalStringurl="jdbc:mysql://localhost:3306/demo";
privatestaticfinalStringusername="root";
privatestaticfinalStringpassword="root";
//定义一个数据库连接
privatestaticConnectionconn=null;
//获取连接
publicstaticConnectiongetConnection(){
try{
Class.forName(driver);
conn=DriverManager.getConnection(url,username,password);
}catch(Exceptione){
e.printStackTrace();
}
returnconn;
}
//关闭连接
publicstaticvoidcloseConnection(){
try{
if(conn!=null){
conn.close();
}
}catch(Exceptione){
e.printStackTrace();
}
}
}
里面搞了一个static的Connection，这下子数据库连接就好操作了，牛逼吧！
然后，我定义了一个接口，用于给逻辑层来调用：
publicinterfaceProductService{
voidupdateProductPrice(longproductId,intprice);
}
根据用户提出的需求，我想这个接口完全够用了。根据productId去更新对应Product的price，然后再插入一条数据到log表中。
其实业务逻辑也不太复杂，于是我快速地完成了ProductService接口的实现类：
publicclassProductServiceImplimplementsProductService{
privatestaticfinalStringUPDATE_PRODUCT_SQL="updateproductsetprice=?whereid=?";
privatestaticfinalStringINSERT_LOG_SQL="insertintolog(created,description)values(?,?)";
publicvoidupdateProductPrice(longproductId,intprice){
try{
//获取连接
Connectionconn=DBUtil.getConnection();
conn.setAutoCommit(false);//关闭自动提交事务（开启事务）
//执行操作
updateProduct(conn,UPDATE_PRODUCT_SQL,productId,price);//更新产品
insertLog(conn,INSERT_LOG_SQL,"Createproduct.");//插入日志
//提交事务
conn.commit();
}catch(Exceptione){
e.printStackTrace();
}finally{
//关闭连接
DBUtil.closeConnection();
}
}
privatevoidupdateProduct(Connectionconn,StringupdateProductSQL,longproductId,intproductPrice)throwsException{
PreparedStatementpstmt=conn.prepareStatement(updateProductSQL);
pstmt.setInt(1,productPrice);
pstmt.setLong(2,productId);
introws=pstmt.executeUpdate();
if(rows!=0){
System.out.println("Updateproductsuccess!");
}
}
privatevoidinsertLog(Connectionconn,StringinsertLogSQL,StringlogDescription)throwsException{
PreparedStatementpstmt=conn.prepareStatement(insertLogSQL);
pstmt.setString(1,newSimpleDateFormat("yyyy-MM-ddHH:mm:ssSSS").format(newDate()));
pstmt.setString(2,logDescription);
introws=pstmt.executeUpdate();
if(rows!=0){
System.out.println("Insertlogsuccess!");
}
}
}
代码的可读性还算不错吧？这里我用到了JDBC的高级特性Transaction了。暗自庆幸了一番之后，我想是不是有必要写一个客户端，来测试一下执行结果是不是我想要的呢？于是我偷懒，直接在ProductServiceImpl中增加了一个main()方法：
publicstaticvoidmain(String[]args){
ProductServiceproductService=newProductServiceImpl();
productService.updateProductPrice(1,3000);
}
我想让productId为1的产品的价格修改为3000。于是我把程序跑了一遍，控制台输出：
Updateproductsuccess! Insertlogsuccess!
应该是对了。作为一名专业的程序员，为了万无一失，我一定要到数据库里在看看。没错！product表对应的记录更新了，log表也插入了一条记录。这样就可以将ProductService接口交付给别人来调用了。
几个小时过去了，QA妹妹开始骂我：“我靠！我才模拟了10个请求，你这个接口怎么就挂了？说是数据库连接关闭了！”。
听到这样的叫声，让我浑身打颤，立马中断了我的小视频，赶紧打开IDE，找到了这个ProductServiceImpl这个实现类。好像没有Bug吧？但我现在不敢给她任何回应，我确实有点怕她的。
我突然想起，她是用工具模拟的，也就是模拟多个线程了！那我自己也可以模拟啊，于是我写了一个线程类：
publicclassClientThreadextendsThread{
privateProductServiceproductService;
publicClientThread(ProductServiceproductService){
this.productService=productService;
}
@Override
publicvoidrun(){
System.out.println(Thread.currentThread().getName());
productService.updateProductPrice(1,3000);
}
}
我用这线程去调用ProduceService的方法，看看是不是有问题。此时，我还要再修改一下main()方法：
//publicstaticvoidmain(String[]args){
//ProductServiceproductService=newProductServiceImpl();
//productService.updateProductPrice(1,3000);
//}
publicstaticvoidmain(String[]args){
for(inti=0;i<10;i++){
ProductServiceproductService=newProductServiceImpl();
ClientThreadthread=newClientThread(productService);
thread.start();
}
}
我也模拟10个线程吧，我就不信那个邪了！
运行结果真的让我很晕、很晕：
Thread-1 Thread-3 Thread-5 Thread-7 Thread-9 Thread-0 Thread-2 Thread-4 Thread-6 Thread-8 Updateproductsuccess! Insertlogsuccess! Updateproductsuccess! Insertlogsuccess! Updateproductsuccess! Insertlogsuccess! Updateproductsuccess! Insertlogsuccess! Updateproductsuccess! Insertlogsuccess! Updateproductsuccess! Insertlogsuccess! Updateproductsuccess! Insertlogsuccess! Updateproductsuccess! Insertlogsuccess! Updateproductsuccess! Insertlogsuccess! Updateproductsuccess! Insertlogsuccess! com.mysql.jdbc.exceptions.jdbc4.MySQLNonTransientConnectionException:Nooperationsallowedafterconnectionclosed. atsun.reflect.NativeConstructorAccessorImpl.newInstance0(NativeMethod) atsun.reflect.NativeConstructorAccessorImpl.newInstance(NativeConstructorAccessorImpl.java:39) atsun.reflect.DelegatingConstructorAccessorImpl.newInstance(DelegatingConstructorAccessorImpl.java:27) atjava.lang.reflect.Constructor.newInstance(Constructor.java:513) atcom.mysql.jdbc.Util.handleNewInstance(Util.java:411) atcom.mysql.jdbc.Util.getInstance(Util.java:386) atcom.mysql.jdbc.SQLError.createSQLException(SQLError.java:1015) atcom.mysql.jdbc.SQLError.createSQLException(SQLError.java:989) atcom.mysql.jdbc.SQLError.createSQLException(SQLError.java:975) atcom.mysql.jdbc.SQLError.createSQLException(SQLError.java:920) atcom.mysql.jdbc.ConnectionImpl.throwConnectionClosedException(ConnectionImpl.java:1304) atcom.mysql.jdbc.ConnectionImpl.checkClosed(ConnectionImpl.java:1296) atcom.mysql.jdbc.ConnectionImpl.commit(ConnectionImpl.java:1699) atcom.smart.sample.test.transaction.solution1.ProductServiceImpl.updateProductPrice(ProductServiceImpl.java:25) atcom.smart.sample.test.transaction.ClientThread.run(ClientThread.java:18)
我靠！竟然在多线程的环境下报错了，果然是数据库连接关闭了。怎么回事呢？我陷入了沉思中。于是我Copy了一把那句报错信息，在百度、Google，还有OSC里都找了，解答实在是千奇百怪。
我突然想起，既然是跟Connection有关系，那我就将主要精力放在检查Connection相关的代码上吧。是不是Connection不应该是static的呢？我当初设计成static的主要是为了让DBUtil的static方法访问起来更加方便，用static变量来存放Connection 也提高了性能啊。怎么搞呢？
原来要使每个线程都拥有自己的连接，而不是共享同一个连接，否则线程1有可能会关闭线程2的连接，所以线程2就报错了。一定是这样！
我赶紧将DBUtil给重构了：
publicclassDBUtil{
//数据库配置
privatestaticfinalStringdriver="com.mysql.jdbc.Driver";
privatestaticfinalStringurl="jdbc:mysql://localhost:3306/demo";
privatestaticfinalStringusername="root";
privatestaticfinalStringpassword="root";
//定义一个用于放置数据库连接的局部线程变量（使每个线程都拥有自己的连接）
privatestaticThreadLocalconnContainer=newThreadLocal();
//获取连接
publicstaticConnectiongetConnection(){
Connectionconn=connContainer.get();
try{
if(conn==null){
Class.forName(driver);
conn=DriverManager.getConnection(url,username,password);
}
}catch(Exceptione){
e.printStackTrace();
}finally{
connContainer.set(conn);
}
returnconn;
}
//关闭连接
publicstaticvoidcloseConnection(){
Connectionconn=connContainer.get();
try{
if(conn!=null){
conn.close();
}
}catch(Exceptione){
e.printStackTrace();
}finally{
connContainer.remove();
}
}
}
我把Connection放到了ThreadLocal中，这样每个线程之间就隔离了，不会相互干扰了。
此外，在getConnection()方法中，首先从ThreadLocal中（也就是connContainer中） 获取Connection，如果没有，就通过JDBC来创建连接，最后再把创建好的连接放入这个ThreadLocal中。可以把 ThreadLocal看做是一个容器，一点不假。
同样，我也对closeConnection()方法做了重构，先从容器中获取Connection，拿到了就close掉，最后从容器中将其remove掉，以保持容器的清洁。
这下应该行了吧？我再次运行main()方法：
Thread-0 Thread-2 Thread-4 Thread-6 Thread-8 Thread-1 Thread-3 Thread-5 Thread-7 Thread-9 Updateproductsuccess! Insertlogsuccess! Updateproductsuccess! Insertlogsuccess! Updateproductsuccess! Insertlogsuccess! Updateproductsuccess! Insertlogsuccess! Updateproductsuccess! Insertlogsuccess! Updateproductsuccess! Insertlogsuccess! Updateproductsuccess! Insertlogsuccess! Updateproductsuccess! Insertlogsuccess! Updateproductsuccess! Insertlogsuccess! Updateproductsuccess! Insertlogsuccess!
我去！总算是解决了，QA妹妹，你应该会对我微笑一下吧？
感谢您的关注，分享是一种快乐，也希望得到您的支持与批评！
注意：该示例仅用于说明 TheadLocal的基本用法。在实际工作中，推荐使用连接池来管理数据库连接。示例中的代码仅作参考，使用前请酌情考虑。