ThreadLocal的使用场景分析

目录

一.ThreadLocal介绍

二.使用场景1——数据库事务问题

  2.1 问题背景

  2.2 方案1-修改接口传参

  2.3 方案2-使用ThreadLocal

三.使用场景2——日志追踪问题

四.其他使用场景

 

 

一.ThreadLocal介绍

  我们知道,变量从作用域范围进行分类,可以分为“全局变量”、“局部变量”两种:

  1.全局变量(global variable),比如类的静态属性(加static关键字),在类的整个生命周期都有效;

  2.局部变量(local variable),比如在一个方法中定义的变量,作用域只是在当前方法内,方法执行完毕后,变量就销毁(释放)了;

  使用全局变量,当多个线程同时修改静态属性,就容易出现并发问题,导致脏数据;而局部变量一般来说不会出现并发问题(在方法中开启多线程并发修改局部变量,仍可能引起并发问题);

  再看ThreadLocal,从名称上就能知道,它可以用来保存局部变量,只不过这个“局部”是指“线程”作用域,也就是说,该变量在该线程的整个生命周期中有效。

 

二.使用场景1——数据库事务问题

2.1问题背景

  下面介绍示例,UserService调用UserDao删除用户信息,涉及到两张表的操作,所以用到了数据库事务:

  数据库封装类DbUtils

public class DbUtils {

    // 使用C3P0连接池
    private static ComboPooledDataSource dataSource = new ComboPooledDataSource("dev");

    public static Connection getConnectionFromPool() throws SQLException {
        return dataSource.getConnection();
    }

    // 省略其他方法.....
}

  UserService代码如下:  

public class UserService {

    private UserDao userDao;

    public void deleteUserInfo(Integer id, String operator) {
        Connection connection = null;
        try {
            // 从连接池中获取一个连接
            connection = DbUtils.getConnectionFromPool();
            // 因为涉及事务操作,所以需要关闭自动提交
            connection.setAutoCommit(false);

            // 事务涉及两步操作,删除用户表,增加操作日志
            userDao.deleteUserById(id);
            userDao.addOperateLog(id, operator);

            connection.commit();
        } catch (SQLException e) {
            // 回滚操作
            try {
                if (connection != null) {
                    connection.rollback();
                }
            } catch (SQLException ex) {
            }
        } finally {
            DbUtils.freeConnection(connection);
        }
    }
}

  下面是UserDao,省略了部分代码:

package cn.ganlixin.dao;
import cn.ganlixin.util.DbUtils;
import java.sql.Connection;

/**
 * @author ganlixin
 * @create 2020-06-12
 */
public class UserDao {

    public void deleteUserById(Integer id) {
        // 从连接池中获取一个数据连接
        Connection connection = DbUtils.getConnectionFromPool();

        // 利用获取的数据库连接,执行sql...........删除用户表的一条数据

        // 归还连接给连接池
        DbUtils.freeConnection(connection);
    }

    public void addOperateLog(Integer id, String operator) {
        // 从连接池中获取一个数据连接
        Connection connection = DbUtils.getConnectionFromPool();

        // 利用获取的数据库连接,执行sql...........插入一条记录到操作日志表

        // 归还连接给连接池
        DbUtils.freeConnection(connection);
    }
}

  上面的代码乍一看,好像没啥问题,但是仔细看,其实是存在问题的!!问题出在哪儿呢?就出在从数据库连接池获取连接哪个位置。

  1.UserService会从数据库连接池获取一个连接,关闭该连接的自动提交;

  2.UserService然后调用UserDao的两个接口进行数据库操作;

  3.UserDao的两个接口,都会从数据库连接池获取一个连接,然后执行sql;

  注意,第1步和第3步获得的连接不一定是同一个!!!!这才是关键。

  如果UserService和UserDao获取的数据库连接不是同一个,那么UserService中关闭自动提交的数据库连接,并不是UserDao接口中执行sql的数据库连接,当userService中捕获异常,即使执行rollback,userDao中的sql已经执行完了,并不会回滚,所以数据已经出现不一致!!!

 

2.2方案1-修改接口传参

  上面的例子中,因为UserService和UserDao获取的连接不是同一个,所以并不能保证事务原子性;那么只要能够解决这个问题,就可以保证了

  可以修改userDao中的代码,不要每次在UserDao中从数据库连接池获取连接,而是增加一个参数,该参数就是数据库连接,有UserService传入,这样就能保证UserService和UserDao使用同一个数据库连接了

public class UserDao {

    public void deleteUserById(Connection connection, Integer id) {
        // 利用传入的数据库连接,执行sql...........删除用户表的一条数据
    }

    public void addOperateLog(Connection connection, Integer id, String operator) {
        // 利用传入的数据库连接,执行sql...........插入一条记录到操作日志表
    }
}

  UserService调用接口时,传入数据库连接,修改代码后如下:

// 事务涉及两步操作,删除用户表,增加操作日志
// 新增参数传入数据库连接,保证UserService和UserDao使用同一个连接
userDao.deleteUserById(connection, id);
userDao.addOperateLog(connection, id, operator);

  这样做,的确是能解决数据库事务的问题,但是并不推荐这样做,耦合度太高,不利于维护,修改起来也不方便;

 

2.3使用ThreadLocal解决

  ThreadLocal可以保存当前线程有效的变量,正好适合解决这个问题,而且改动的点也特别小,只需要在DbUtils获取连接的时候,将获取到的连接存到ThreadLocal中即可:

public class DbUtils {

    // 使用C3P0连接池
    private static ComboPooledDataSource dataSource = new ComboPooledDataSource("dev");

    // 创建threadLocal对象,保存每个线程的数据库连接对象
    private static ThreadLocal threadLocal = new ThreadLocal<>();

    public static Connection getConnectionFromPool() throws SQLException {
        if (threadLocal.get() == null) {
            threadLocal.set(dataSource.getConnection());
        }

        return threadLocal.get();
    }

    // 省略其他方法.....
}

  然后UserService和UserDao中,恢复最初的版本,UserService和UserDao中都调用DbUtils获取数据库连接,此时他们获取到的连接则是同一个Connection对象,就可以解决数据库事务问题了。

 

三.使用场景2——日志追踪问题

  如果理解了场景1的数据库事务问题,那么对于本小节的日志追踪,光看标题就知道是怎么回事了;

  开发过程时,会在项目中打很多的日志,一般来说,查看日志的时候,都是通过关键字去找日志,这就需要我们在打日志的时候明确的写入某些标识,比如用户ID、订单号、流水号...

  如果业务比较复杂,那么一个请求的处理流程就会比较长,如果将这么一长串的流程给串起来,也可以通过前面说的用户ID、订单号、流水号来串,但有个问题,某些接口没有用户ID或者订单号作为参数!!!!这个时候,当然可以像场景1中给接口增加用户ID或者订单号作为参数,但是这样实现起来,除非想被炒鱿鱼,否则就别这样做。

  此时可以就可以使用ThreadLocal,封装一个工具类,提供唯一标识(可以是用户ID、订单号、或者是分布式全局ID),示例如下:

package cn.ganlixin.util;

/**
 * 描述:
 * 日志追踪工具类,设置和获取traceId,
 * 此处的traceId使用snowFlake雪花数算法,详情可以参考:https://www.cnblogs.com/-beyond/p/12452632.html
 *
 * @author ganlixin
 * @create 2020-06-12
 */
public class TraceUtils {
    // 创建ThreadLocal静态属性,存Long类型的uuid
    private static final ThreadLocal threadLocal = new ThreadLocal<>();

    // 全局id生成器(雪花数算法)
    private static final SnowFlakeIdGenerator generator = new SnowFlakeIdGenerator(1, 1);

    public static void setUuid(String uuid) {
        // 雪花数算法
        threadLocal.set(generator.nextId());
    }

    public static Long getUuid() {
        if (threadLocal.get() == null) {
            threadLocal.set(generator.nextId());
        }
        return threadLocal.get();
    }
}

  

  使用示例:

@Slf4j
public class UserService {

    private UserDao userDao;

    public void deleteUserInfo(Integer id, String operator) {
        log.info("traceId:{}, id:{}, operator:{}", TraceUtils.getUuid(), id, operator);
        
        //.....
    }
}

@Slf4j
public class UserDao {

    public void deleteUserById(Connection connection, Integer id) {
        log.info("traceId:{}, id:{}", TraceUtils.getUuid(), id);
    }
}

  

 四.其他场景

  其他场景,其实就是利用ThreadLocal“线程私有且线程间互不影响”特性,除了上面的两个场景,常见的还有用来记录用户的登录状态(当然也可以用session或者cookie实现)。

 

  原文地址:https://www.cnblogs.com/-beyond/p/13111015.html 

你可能感兴趣的:(ThreadLocal的使用场景分析)