详解ThreadLocal

提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档


1、ThreadLocal介绍

1.1 官方介绍

ThreadLocal类用来提供线程内部的局部变量。这种变量在多线程环境下访问(通过get和set方法访问)时能够保证各个线程的变量独立于其他线程的变量。ThreadLocal实例通常来说都是private static 类型的,用来关联线程和应用上下文。

我们可以得知ThreadLocal的作用是:提供线程内的局部变量,不同的线程之间不会互相干扰,这种变量在线程的周期内起作用,减少同一个线程内多个函数或者组件之间一些公共变量传递的复杂度

总结:

  1. 线程并发:在多线程并发场景下
  2. 传递数据:我们可以通过ThreadLocal在同一线程,不同组件中传递公共变量
  3. 线程隔离:每个线程的变量都是独立的,不会互相影响

1.2 基本用法

1.2.1 常用方法

在使用之前,先介绍几个ThreadLocal常用的方法

  • ThreadLocal:创建ThreadLocal对象
  • public void set(T value):设置当前线程绑定的局部变量
  • public T get():获取当前线程绑定的局部变量
  • public void remove():移除当前线程绑定的局部变量

1.2.2 使用案例

首先看下面一个案例:5个线程并发地打印数据

public class MyDemo01 {
    
    
    //变量
    private String content;

    public String getContent() {
    
    
        return content;
    }

    public void setContent(String content) {
    
    
        this.content = content;
    }


    public static void main(String[] args) {
    
    
        MyDemo01 demo = new MyDemo01();

        for (int i = 0; i < 5; i++) {
    
    
            int j =i;
            new Thread(()->{
    
    
                demo.setContent(Thread.currentThread().getName()+"的数据");
                System.out.println("--------------");
                System.out.println(Thread.currentThread().getName()+"---->"+demo.getContent());
            },"线程"+j).start();
        }
    }
}

如果不加控制,容易产生线程安全问题:

--------------
--------------
线程1---->线程1的数据
线程0---->线程1的数据
--------------
线程2---->线程2的数据
--------------
线程4---->线程4的数据
--------------
线程3---->线程3的数据

那怎么解决呢,其实第一反应就是就加锁,比如说使用synchronized


public class MyDemo01 {
    
    
    //变量
    private String content;

    public String getContent() {
    
    
        return content;
    }

    public void setContent(String content) {
    
    
        this.content = content;
    }


    public static void main(String[] args) {
    
    
        MyDemo01 demo = new MyDemo01();

        for (int i = 0; i < 5; i++) {
    
    
            int j =i;
            new Thread(()->{
    
    
                synchronized (MyDemo01.class){
    
    
                    demo.setContent(Thread.currentThread().getName()+"的数据");
                    System.out.println("--------------");
                    System.out.println(Thread.currentThread().getName()+"---->"+demo.getContent());
                }

            },"线程"+j).start();
        }
    }
}
--------------
线程0---->线程0的数据
--------------
线程1---->线程1的数据
--------------
线程2---->线程2的数据
--------------
线程3---->线程3的数据
--------------
线程4---->线程4的数据

加锁确实可以解决并发安全性问题,但是加锁的本质是把这个5线程在执行synchronized 代码时,串行的取执行,而且synchronized 锁本身就比较重,同步代码块势必会影响性能。

有没有更好的解决方式呢?哎,那就是今天的主角——ThreadLocal

/**
 * 需求: 线程隔离
 *        在多线程并发场景下,每个线程的变量都是相互独立的
 *        线程A: 设置(变量1)       获取(变量1)
 *        线程B: 设置(变量2)       获取(变量2)
 *        
 *        ThreadLocal:
 *           set(): 将变量绑定到当前线程
 *           get():获取当前线程绑定的变量
 *        
 */
public class MyDemo01 {
    
    

    ThreadLocal<String> t1 = new ThreadLocal<>();
    //变量
    private String content;

    public String getContent() {
    
    
//        return content;
        String s = t1.get();
        return s;
    }

    public void setContent(String content) {
    
    
//        this.content = content;
        //变量绑定到当前线程
        t1.set(content);
    }


    public static void main(String[] args) {
    
    
        MyDemo01 demo = new MyDemo01();

        for (int i = 0; i < 5; i++) {
    
    
            int j =i;
            new Thread(()->{
    
    
                    demo.setContent(Thread.currentThread().getName()+"的数据");
                    System.out.println("--------------");
                    System.out.println(Thread.currentThread().getName()+"---->"+demo.getContent());

            },"线程"+j).start();
        }
    }
}

--------------
线程2---->线程2的数据
--------------
线程1---->线程1的数据
--------------
--------------
线程3---->线程3的数据
线程0---->线程0的数据
--------------
线程4---->线程4的数据

1.3 ThreadLocal与synchronized的区别

虽然ThreadLocal模式与synchronized关键字都用于处理多线程并发访问变量的问题,不过二者处理问题的角度和思路不同:

synchronized ThreadLocal
原理 同步机制采用“以时间换空间”的方式,只提供了一份变量,让不同的线程排队访问 ThreadLoca采用“以空间换时间”的方式,为每一线程提供了一份变量的副本,从而实现互相访问而互不干扰
侧重点 多个线程之间访问资源的同步 多线程中让每个线程之间的数据相互隔离

总结:在刚刚的案例中,虽然使用ThreadLocal和synchronized都能解决问题,但是使用ThreadLocal更为合理,因为这样可以使程序拥有更高的并发性

2、运用场景_事务案例

通过以上的介绍,我们已经了解了ThreadLocal的基本使用以及特点,但是它运用在什么场景中呢?接下来我们看一个案例:事务操作

2.1 转账案例

首先我们先构建一个简单的转账场景:有一个数据表account,里面有两个用户jack和rose,用jack给rose转账

案例的实现主要用:mysql数据库、JDBC和C3P0框架,以下是详细代码

初始数据:

在这里插入图片描述
在这里插入图片描述
AccountDao :

public class AccountDao {
    
    
    public void out(String outUser, int money) throws SQLException {
    
    
        String sql = "update account set money = money - ? where name = ?";

        Connection conn = JdbcUtils.getConnection();
        PreparedStatement pstm = conn.prepareStatement(sql);
        pstm.setInt(1,money);
        pstm.setString(2,outUser);
        pstm.executeUpdate();

        JdbcUtils.release(pstm,conn);
    }

    public void in(String inUser, int money) throws SQLException {
    
    
        String sql = "update account set money = money + ? where name = ?";

        Connection conn = JdbcUtils.getConnection();
        PreparedStatement pstm = conn.prepareStatement(sql);
        pstm.setInt(1,money);
        pstm.setString(2,inUser);
        pstm.executeUpdate();

        JdbcUtils.release(pstm,conn);
    }
}

AccountService :

public class AccountService {
    
    

    public boolean transfer(String outUser, String inUser, int money) {
    
    
        AccountDao ad = new AccountDao();
        try {
    
    
            // 转出
            ad.out(outUser, money);
            // 转入
            ad.in(inUser, money);
        } catch (Exception e) {
    
    
            e.printStackTrace();
            return false;
        }
        return true;
    }
}

JdbcUtils :

public class JdbcUtils {
    
    
    //c3p0 数据库连接池对象属性
    private static final ComboPooledDataSource ds = new ComboPooledDataSource();

    //获取连接
    public static Connection getConnection() throws SQLException {
    
    
        return ds.getConnection();
    }

    //释放资源
    public static void  release(AutoCloseable... ios){
    
    
        for (AutoCloseable io: ios) {
    
    
            if (io!=null){
    
    
                try {
    
    
                    io.close();
                } catch (Exception e) {
    
    
                    e.printStackTrace();
                }
            }
        }
    }

    public static void commitAndClose(Connection conn) {
    
    
        try {
    
    
            if(conn != null){
    
    
                //提交事务
                conn.commit();
                //释放连接
                conn.close();
            }
        } catch (SQLException e) {
    
    
            e.printStackTrace();
        }
    }

    public static void rollbackAndClose(Connection conn) {
    
    
        try {
    
    
            if(conn != null){
    
    
                //回滚事务
                conn.rollback();
                //释放连接
                conn.close();
            }
        } catch (SQLException e) {
    
    
            e.printStackTrace();
        }
    }
}

c3p0的配置

<?xml version="1.0" encoding="UTF-8"?>
<c3p0-config>

    <default-config>
        <property name="initialPoolSize">5</property>
        <property name="maxPoolSize">10</property>
        <property name="checkoutTimeout">3000</property>


        <property name="driverClass">com.mysql.cj.jdbc.Driver</property>
        <property name="jdbcUrl">jdbc:mysql://localhost:3306/test</property>
        <property name="user">root</property>
        <property name="password">root</property>
    </default-config>


</c3p0-config>

如果转账过程中发生异常,比如:

public class AccountService {
    
    
    public boolean transfer(String outUser, String inUser, int money) {
    
    
        AccountDao ad = new AccountDao();
        try {
    
    
            // 转出
            ad.out(outUser, money);
            // 模拟转账过程中的异常
            int i = 1/0;
            // 转入
            ad.in(inUser, money);
        } catch (Exception e) {
    
    
            e.printStackTrace();
            return false;
        }
        return true;
    }
}

那么Jack转处钱,Rose却不能转入钱

在这里插入图片描述
解决办法,引入事务:
在这里插入图片描述

2.2 常规解决方式

public class AccountService {
    
    
    public boolean transfer(String outUser, String inUser, int money) {
    
    
        AccountDao ad = new AccountDao();
        //线程并发情况下,为了保证每个线程使用各自的connection,故加锁
        synchronized (AccountService.class) {
    
    

            Connection conn = null;
            try {
    
    
                conn = JdbcUtils.getConnection();
                //开启事务
                conn.setAutoCommit(false);
                // 转出
                ad.out(conn, outUser, money);
                // 模拟转账过程中的异常
//            int i = 1/0;
                // 转入
                ad.in(conn, inUser, money);
                //事务提交
                JdbcUtils.commitAndClose(conn);
            } catch (Exception e) {
    
    
                e.printStackTrace();
                //事务回滚
                JdbcUtils.rollbackAndClose(conn);
                return false;
            }
            return true;
        }
    }
}

public class AccountDao {
    
    
    public void out(Connection conn, String outUser, int money) throws SQLException{
    
    
        String sql = "update account set money = money - ? where name = ?";
        //注释从连接池获取连接的代码,使用从service中传递过来的connection
//        Connection conn = JdbcUtils.getConnection();
        PreparedStatement pstm = conn.prepareStatement(sql);
        pstm.setInt(1,money);
        pstm.setString(2,outUser);
        pstm.executeUpdate();
        //连接不能在这里释放,service层中还需要使用
//        JdbcUtils.release(pstm,conn);
        JdbcUtils.release(pstm);
    }

    public void in(Connection conn, String inUser, int money) throws SQLException {
    
    
        String sql = "update account set money = money + ? where name = ?";
//        Connection conn = JdbcUtils.getConnection();
        PreparedStatement pstm = conn.prepareStatement(sql);
        pstm.setInt(1,money);
        pstm.setString(2,inUser);
        pstm.executeUpdate();
//        JdbcUtils.release(pstm,conn);
        JdbcUtils.release(pstm);
    }
}

对上述进行一个说明
1、为了保证AccountService 和AccountDao 使用同一个Connection ,解决方式是AccountService 把conn 传给AccountDao
2、为啥子加锁来保证一个线程一个conn ?是因为conn 发生了逃逸,在抛出异常前释放链接是不安全的。

上述虽然解决了问题,但是很不优雅,总不能每次都传conn 吧,这样代码耦合了,加锁也会降低并发性

2.3 ThreadLocal解决方案

首先改造JdbcUtils ,我们把连接对象Connection 与当前线程进行绑定,因为线程在第一次获取Connection 时,没有绑定过,锁有要进行判断,如果Connection 是null,就从连接池中获取一个连接对象和当前线程进行绑定,后续当前线程再获取时,ThreadLocal就能get到之前绑定的Connection 了。

public class JdbcUtils {
    
    
    
    static  ThreadLocal<Connection> t1 = new ThreadLocal<>();
    
    //c3p0 数据库连接池对象属性
    private static final ComboPooledDataSource ds = new ComboPooledDataSource();
    // 
    //获取连接
    /*
     *原来:直接从线程池中拿, getConnection会拿到不一样的链接
     * 
     * 现在:1、直接获取当前线程绑定的链接对象
     *       2、如果连接是空的
     *            2.1 再去连接池中去拿
     *            2.2 将次连接对象与当前线程进行绑定
     *       
     */
    public static Connection getConnection() throws SQLException {
    
    
        Connection conn = t1.get();
        if (conn ==null){
    
      //因为当前线程第一次获取的时候,没有绑定过
            conn = ds.getConnection();
            t1.set(conn);
        }
        return conn;
    }
 public static void commitAndClose(Connection conn) {
    
    
        try {
    
    
            if(conn != null){
    
    
                //提交事务
                conn.commit();
                //解绑当前线程的连接对象,防止内存泄漏(后面解释)
                t1.remove();
                //释放连接
                conn.close();
            }
        } catch (SQLException e) {
    
    
            e.printStackTrace();
        }
    }

改造AccountService 、AccountDao

public class AccountService {
    
    
    public boolean transfer(String outUser, String inUser, int money) {
    
    
        AccountDao ad = new AccountDao();
       
        synchronized (AccountService.class) {
    
    

            Connection conn = null;
            try {
    
    
               //第一次获取Connection ,会从连接池获取一个连接,绑定到当前线程的ThreadLocal
                conn = JdbcUtils.getConnection(); 
                //开启事务
                conn.setAutoCommit(false);
                // 转出
                ad.out(outUser, money);
                // 模拟转账过程中的异常
//            int i = 1/0;
                // 转入
                ad.in( inUser, money);
                //事务提交
                JdbcUtils.commitAndClose(conn);
            } catch (Exception e) {
    
    
                e.printStackTrace();
                //事务回滚
                JdbcUtils.rollbackAndClose(conn);
                return false;
            }
            return true;
        }
    }
}

public class AccountDao {
    
    
    public void out(C String outUser, int money) throws SQLException{
    
    
        String sql = "update account set money = money - ? where name = ?";
        //再次获取Connection ,会直接从ThreadLocal中去拿
        Connection conn = JdbcUtils.getConnection();
        PreparedStatement pstm = conn.prepareStatement(sql);
        pstm.setInt(1,money);
        pstm.setString(2,outUser);
        pstm.executeUpdate();
        //连接不能在这里释放,service层中还需要使用
//        JdbcUtils.release(pstm,conn);
        JdbcUtils.release(pstm);
    }

    public void in( String inUser, int money) throws SQLException {
    
    
        String sql = "update account set money = money + ? where name = ?";
         //再次获取Connection ,会直接从ThreadLocal中去拿
        Connection conn = JdbcUtils.getConnection();
        PreparedStatement pstm = conn.prepareStatement(sql);
        pstm.setInt(1,money);
        pstm.setString(2,inUser);
        pstm.executeUpdate();
        JdbcUtils.release(pstm,conn);
        JdbcUtils.release(pstm);
    }
}

这样就实现了无锁无传参的优化,因此并发性更高,代码更优雅

2.4 ThreadLocal方案的好处

从上述的案例中我们可以看到, 在一些特定场景下,ThreadLocal方案有两个突出的优势:

  • 传递数据 : 保存每个线程绑定的数据,在需要的地方可以直接获取, 避免参数直接传递带来的代码耦合问题

  • 线程隔离 : 各线程之间的数据相互隔离却又具备并发性,避免同步方式带来的性能损失

3、ThreadLocal的内部结构

了解JDK8之前:
在这里插入图片描述

每个ThreadLocal都创建一个Map,然后用线程作为Map的key,要存储的局部变量作为Map的value,这样就能达到各个线程的局部变量隔离的效果。这是最简单的设计方法,JDK最早期的ThreadLocal 确实是这样设计的,但现在早已不是了。

JDK8:每个Thread维护一个ThreadLocalMap,这个Map的key是ThreadLocal实例本身,value才是真正要存储的值Object。

具体的过程是这样的:

  • 每个Thread线程内部都有一个Map (ThreadLocalMap)

  • Map里面存储ThreadLocal对象(key)和线程的变量副本(value)

  • Thread内部的Map是由ThreadLocal维护的,由ThreadLocal负责向map获取和设置线程的变量值。

  • ​ 对于不同的线程,每次获取副本值时,别的线程并不能获取到当前线程的副本值,形成了副本的隔离,互不干扰
    在这里插入图片描述

这样设计的好处

​ 这个设计与我们一开始说的设计刚好相反,这样设计有如下两个优势:

  • 这样设计之后每个Map存储的Entry数量就会变少。因为之前的存储数量由Thread的数量决定,现在是由ThreadLocal的数量决定。在实际运用当中,往往ThreadLocal的数量要少于Thread的数量。
  • 当Thread销毁之后,对应的ThreadLocalMap也会随之销毁,能减少内存的使用

4、ThreadLocal的核心方法源码

​ 基于ThreadLocal的内部结构,我们继续分析它的核心方法源码,更深入的了解其操作原理。

除了构造方法之外, ThreadLocal对外暴露的方法有以下4个:
在这里插入图片描述

4.1 get()

/*
设置当前线程对应的ThreadLocal的值
@param value:将要保存在当前线程对应的ThreadLocal的值
*/
  public void set(T value) {
    
    
         //获取当前线程对象
        Thread t = Thread.currentThread();
        //获取此线程对象中维护的ThreadLocalMap对象
        ThreadLocalMap map = getMap(t);
        //判断map是否存在
        if (map != null)
            //调用map.set设置此实体entry(this是当前ThreadLocal,Vavue是你要绑定的值)
            map.set(this, value);
        else
            //当前线程Thread不存在ThreadLcoalMap对象
            //则调用createMap进行ThreadLcoalMap对象的初始化
            //并将t(当前现车给)和value(t对应的值)作为第一个entry放在ThreadLcoalMap中
            createMap(t, value);
    }

  /**
     * 
     * 获取当前线程Thread对应维护的ThreadLcoalMap的值
     *
     * @param  t 当前线程
     * @return the map 对应维护的ThreadLcoalMap
     */
    ThreadLocalMap getMap(Thread t) {
    
    
        return t.threadLocals;
    }

/**
     * 
     * 创建当前线程Thread对应维护的ThreadLcoalMap
     *
     * @param t 当前线程
     * @param firstValue 存放到map中的第一个entry
     */
    void createMap(Thread t, T firstValue) {
    
    
       //  这里的this是调用此方法的ThreadLcoal   
        t.threadLocals = new ThreadLocalMap(this, firstValue);
    }

代码执行流程

​ A. 首先获取当前线程,并根据当前线程获取一个Map

​ B. 如果获取的Map不为空,则将参数设置到Map中(当前ThreadLocal的引用作为key)

​ C. 如果Map为空,则给该线程创建 Map,并设置初始值

4.2 get()

   /**
     * 返回当前线程中保存的ThreadLocal的值
     * 如果当前线程没有ThreadLcoal变量
     *  则它会通过 {@link #initialValue} 方法初始化
     *
     * @return t 返回当前线程对应的ThreadLocal的值
     */
    public T get() {
    
    
        //获取当前线程对象
        Thread t = Thread.currentThread();
        //获取当前线程Thread对象维护对应的ThreadLcoalMap对象
        ThreadLocalMap map = getMap(t);
        //如果没有map
        if (map != null) {
    
    
            //以当前线程的ThreadLocal为key,调用getEntry获取对应的存储实体
            ThreadLocalMap.Entry e = map.getEntry(this);
            //判断e是否存在
            if (e != null) {
    
    
                @SuppressWarnings("unchecked")
                //如果存在直接取出e对应的value,value就是当前线程对应的ThreadLcoal值
                T result = (T)e.value;
                return result;
            }
        }
        /*
        *   初始化:有两种情况执行当前代码
        *     1、map不存在,表示此线程没有维护的ThreadLocalMap对象
        *      2、map存在。但是没有与当选线程ThreadlLocal关联的entry
        *
        */
        return setInitialValue();
    }



/**
     * 
     *  初始化
     *
     * @return the initial value  初始化后的值
     */
    private T setInitialValue() {
    
    
         // 调用initialValue获取初始化的值
         // 此方法可以被子类重写,如果不重写则默认返回null
        T value = initialValue();
        //获取当前线程对应
        Thread t = Thread.currentThread();
        //获取此线程中维护的ThreadLocalMap 对象
        ThreadLocalMap map = getMap(t);
        //判断map是否存在
        if (map != null)
           //存在则调用map.set设置此实体entry
            map.set(this, value);
        else
            //当前线程Thread不存在ThreadLcoalMap对象
            //则调用createMap进行ThreadLcoalMap对象的初始化
            //并将t(当前现车给)和value(t对应的值)作为第一个entry放在ThreadLcoalMap中
            createMap(t, value);
        return value;
    }

代码执行流程

​ A. 首先获取当前线程, 根据当前线程获取一个Map

​ B. 如果获取的Map不为空,则在Map中以ThreadLocal的引用作为key来在Map中获取对应的Entry e,否则转到D

​ C. 如果e不为null,则返回e.value,否则转到D

​ D. Map为空或者e为空,则通过initialValue函数获取初始值value,然后用ThreadLocal的引用和value作为firstKey和firstValue创建一个新的Map

总结: 先获取当前线程的 ThreadLocalMap 变量,如果存在则返回值,不存在则创建并返回初始值。

4.3 remove()

 /**
     * 删除当前线程中保存的ThreadLocal对应的实体entry
     */
     public void remove() {
    
    
        // 获取当前线程对象中维护的ThreadLocalMap对象
         ThreadLocalMap m = getMap(Thread.currentThread());
        // 如果此map存在
         if (m != null)
            // 存在则调用map.remove
            // 以当前ThreadLocal为key删除对应的实体entry
             m.remove(this);
     }

代码执行流程

​ A. 首先获取当前线程,并根据当前线程获取一个Map

​ B. 如果获取的Map不为空,则移除当前ThreadLocal对象对应的entry

4.4 initialValue()

/**
  * 返回当前线程对应的ThreadLocal的初始值
  
  * 此方法的第一次调用发生在,当线程通过get方法访问此线程的ThreadLocal值时
  * 除非线程先调用了set方法,在这种情况下,initialValue 才不会被这个线程调用。
  * 通常情况下,每个线程最多调用一次这个方法。
  *
  * <p>这个方法仅仅简单的返回null {@code null};
  * 如果程序员想ThreadLocal线程局部变量有一个除null以外的初始值,
  * 必须通过子类继承{@code ThreadLocal} 的方式去重写此方法
  * 通常, 可以通过匿名内部类的方式实现
  *
  * @return 当前ThreadLocal的初始值
  */
protected T initialValue() {
    
    
    return null;
}


此方法的作用是 返回该线程局部变量的初始值。

(1) 这个方法是一个延迟调用方法,从上面的代码我们得知,在set方法还未调用而先调用了get方法时才执行,并且仅执行1次。

(2)这个方法缺省实现直接返回一个null。

(3)如果想要一个除null之外的初始值,可以重写此方法。(备注: 该方法是一个protected的方法,显然是为了让子类覆盖而设计的)

5、ThreadLocalMap源码分析

​ 在分析ThreadLocal方法的时候,我们了解到ThreadLocal的操作实际上是围绕ThreadLocalMap展开的。ThreadLocalMap的源码相对比较复杂, 我们从以下三个方面进行讨论。

5.1 基本结构

ThreadLocalMap是ThreadLocal的内部类,没有实现Map接口,用独立的方式实现了Map的功能,其内部的Entry也是独立实现。
在这里插入图片描述
在这里插入图片描述
(1) 成员变量

    /**
     * 初始容量 —— 必须是2的整次幂
     */
    private static final int INITIAL_CAPACITY = 16;

    /**
     * 存放数据的table,Entry类的定义在下面分析
     * 同样,数组长度必须是2的整次幂。
     */
    private Entry[] table;

    /**
     * 数组里面entrys的个数,可以用于判断table当前使用量是否超过阈值。
     */
    private int size = 0;

    /**
     * 进行扩容的阈值,表使用量大于它的时候进行扩容。
     */
    private int threshold; // Default to 0
    

跟HashMap类似,INITIAL_CAPACITY代表这个Map的初始容量;table 是一个Entry 类型的数组,用于存储数据;size 代表表中的存储数目; threshold 代表需要扩容时对应 size 的阈值。

(2) 存储结构 - Entry

/*
 * Entry继承WeakReference,并且用ThreadLocal作为key.
 * 如果key为null(entry.get() == null),意味着key不再被引用,
 * 因此这时候entry也可以从table中清除。
 */
static class Entry extends WeakReference<ThreadLocal<?>> {
    
    
    /** The value associated with this ThreadLocal. */
    Object value;

    Entry(ThreadLocal<?> k, Object v) {
    
    
        super(k);
        value = v;
    }
}

在ThreadLocalMap中,也是用Entry来保存K-V结构数据的。不过Entry中的key只能是ThreadLocal对象,这点在构造方法中已经限定死了。

​ 另外,Entry继承WeakReference,也就是key(ThreadLocal)是弱引用,其目的是将ThreadLocal对象的生命周期和线程生命周期解绑。

5.2 弱引用和内存泄漏

有些程序员在使用ThreadLocal的过程中会发现有内存泄漏的情况发生,就猜测这个内存泄漏跟Entry中使用了弱引用的key有关系。这个理解其实是不对的。

​ 我们先来回顾这个问题中涉及的几个名词概念,再来分析问题
(1) 内存泄漏相关概念

  • Memory overflow:内存溢出,没有足够的内存提供申请者使用。

    新对象的产生速度大于旧对象的回收速度

  • Memory leak: 内存泄漏是指程序中已动态分配的堆内存由于某种原因程序未释放或无法释放,造成系统内存的浪费,导致程序运行速度减慢甚至系统崩溃等严重后果。内存泄漏的堆积终将导致内存溢出。

    对象长期存活在内存中,且这个对象没什么用

(2) 弱引用相关概念

Java中的引用有4种类型: 强、软、弱、虚。当前这个问题主要涉及到强引用和弱引用:

  • 强引用(“Strong” Reference),就是我们最常见的普通对象引用,只要还有强引用指向一个对象,就能表明对象还“活着”,垃圾回收器就不会回收这种对象。

  • 弱引用(WeakReference),垃圾回收器一旦发现了只具有弱引用的对象,不管当前内存空间足够与否,都会回收它的内存。

(3) 如果key使用强引用

​ 假设ThreadLocalMap中的key使用了强引用,那么会出现内存泄漏吗?

​ 此时ThreadLocal的内存图(实线表示强引用)如下:
在这里插入图片描述
假设在业务代码中使用完ThreadLocal ,threadLocal Ref被回收了。

​ 但是因为threadLocalMap的Entry强引用了threadLocal,造成threadLocal无法被回收。

​ 在没有手动删除这个Entry以及CurrentThread依然运行的前提下,始终有强引用链 threadRef->currentThread->threadLocalMap->entry,Entry就不会被回收(Entry中包括了ThreadLocal实例和value),导致Entry内存泄漏。

​ 也就是说,ThreadLocalMap中的key使用了强引用, 是无法完全避免内存泄漏的。

左边局部变量表中的都是GCroots,经过可达性分析,ThreadLcoal可达,且是强引用,所以不能回收,主要问题在于,ThreadLcoal使用完了应该被GC回收,但是当前与ThreadLcoal绑定的线程还未执行完,当前线程对象是GCRoots,经过可达性分析,能够引用到已经执行完的ThreadLcoal,这样就出现了问题:ThreadLocal的生命周期和当前线程Thread绑定在了一起,如果线程一直不结束,ThreadLcoal就不会被回收

(5)如果key使用弱引用

​ 那么ThreadLocalMap中的key使用了弱引用,会出现内存泄漏吗?

​ 此时ThreadLocal的内存图(实线表示强引用,虚线表示弱引用)如下:
在这里插入图片描述
同样假设在业务代码中使用完ThreadLocal ,threadLocal Ref被回收了。

​ 由于ThreadLocalMap只持有ThreadLocal的弱引用,没有任何强引用指向threadlocal实例, 所以threadlocal就可以顺利被gc回收,此时Entry中的key=null。

​ 但是在没有手动删除这个Entry以及CurrentThread依然运行的前提下,也存在有强引用链 threadRef->currentThread->threadLocalMap->entry -> value ,value不会被回收, 而这块value永远不会被访问到了,导致value内存泄漏。

​ 也就是说,ThreadLocalMap中的key使用了弱引用, 也有可能内存泄漏。

(6)出现内存泄漏的真实原因

​ 比较以上两种情况,我们就会发现,内存泄漏的发生跟ThreadLocalMap中的key是否使用弱引用是没有关系的。那么内存泄漏的的真正原因是什么呢?

​ 细心的同学会发现,在以上两种内存泄漏的情况中,都有两个前提:

1、没有手动删除这个Entry
2、CurrentThread依然运行

​ 第一点很好理解,只要在使用完ThreadLocal,调用其remove方法删除对应的Entry,就能避免内存泄漏。

​ 第二点稍微复杂一点, 由于ThreadLocalMap是Thread的一个属性,被当前线程所引用,所以它的生命周期跟Thread一样长。那么在使用完ThreadLocal之后,如果当前Thread也随之执行结束,ThreadLocalMap自然也会被gc回收,从根源上避免了内存泄漏。

​ 综上,ThreadLocal内存泄漏的根源是:由于ThreadLocalMap的生命周期跟Thread一样长,如果没有手动删除对应key就会导致内存泄漏。

(7) 为什么使用弱引用

​ 根据刚才的分析, 我们知道了: 无论ThreadLocalMap中的key使用哪种类型引用都无法完全避免内存泄漏,跟使用弱引用没有关系。

​ 要避免内存泄漏有两种方式:

使用完ThreadLocal,调用其remove方法删除对应的Entry

使用完ThreadLocal,当前Thread也随之运行结束

相对第一种方式,第二种方式显然更不好控制,特别是使用线程池的时候,线程结束是不会销毁的。

也就是说,只要记得在使用完ThreadLocal及时的调用remove,无论key是强引用还是弱引用都不会有问题。那么为什么key要用弱引用呢?

​ 事实上,在ThreadLocalMap中的set/getEntry方法中,会对key为null(也即是ThreadLocal为null)进行判断,如果为null的话,那么是会对value置为null的。

​ 这就意味着使用完ThreadLocal,CurrentThread依然运行的前提下,就算忘记调用remove方法,弱引用比强引用可以多一层保障:弱引用的ThreadLocal会被回收,对应的value在下一次ThreadLocalMap调用set,get,remove中的任一方法的时候会被清除,从而避免内存泄漏。
这也就解释清除了,为什么在案例中,释放连接的是否,手动删除entry

 public static void commitAndClose(Connection conn) {
    
    
        try {
    
    
            if(conn != null){
    
    
                //提交事务
                conn.commit();
                //解绑当前线程的连接对象,防止内存泄漏(后面解释)
                t1.remove();
                //释放连接
                conn.close();
            }
        } catch (SQLException e) {
    
    
            e.printStackTrace();
        }
    }

5.3 hash冲突的解决

​ hash冲突的解决是Map中的一个重要内容。我们以hash冲突的解决为线索,来研究一下ThreadLocalMap的核心源码。
(1) 首先从ThreadLocal的set() 方法入手

  public void set(T value) {
    
    
        Thread t = Thread.currentThread();
        ThreadLocal.ThreadLocalMap map = getMap(t);
        if (map != null)
            //调用了ThreadLocalMap的set方法
            map.set(this, value);
        else
            createMap(t, value);
    }
    
    ThreadLocal.ThreadLocalMap getMap(Thread t) {
    
    
        return t.threadLocals;
    }

    void createMap(Thread t, T firstValue) {
    
    
        	//调用了ThreadLocalMap的构造方法
        t.threadLocals = new ThreadLocal.ThreadLocalMap(this, firstValue);
    }

这个方法我们刚才分析过, 其作用是设置当前线程绑定的局部变量 :

​ A. 首先获取当前线程,并根据当前线程获取一个Map

​ B. 如果获取的Map不为空,则将参数设置到Map中(当前ThreadLocal的引用作为key)

​ (这里调用了ThreadLocalMap的set方法)

​ C. 如果Map为空,则给该线程创建 Map,并设置初始值

​ (这里调用了ThreadLocalMap的构造方法)

这段代码有两个地方分别涉及到ThreadLocalMap的两个方法, 我们接着分析这两个方法。

(2)构造方法ThreadLocalMap(ThreadLocal<?> firstKey, Object firstValue)

 /*
  * firstKey : 本ThreadLocal实例(this)
  * firstValue : 要保存的线程本地变量
  */
ThreadLocalMap(ThreadLocal<?> firstKey, Object firstValue) {
    
    
        //初始化table
        table = new ThreadLocal.ThreadLocalMap.Entry[INITIAL_CAPACITY];
        //计算索引(重点代码)
        int i = firstKey.threadLocalHashCode & (INITIAL_CAPACITY - 1);
        //设置值
        table[i] = new ThreadLocal.ThreadLocalMap.Entry(firstKey, firstValue);
        size = 1;
        //设置阈值
        setThreshold(INITIAL_CAPACITY);
    }

​ 构造函数首先创建一个长度为16的Entry数组,然后计算出firstKey对应的索引,然后存储到table中,并设置size和threshold。

​ 重点分析: int i = firstKey.threadLocalHashCode & (INITIAL_CAPACITY - 1)。

a. 关于firstKey.threadLocalHashCode:

 	private final int threadLocalHashCode = nextHashCode();
    
    private static int nextHashCode() {
    
    
        return nextHashCode.getAndAdd(HASH_INCREMENT);
    }
//AtomicInteger是一个提供原子操作的Integer类,通过线程安全的方式操作加减,适合高并发情况下的使用
    private static AtomicInteger nextHashCode =  new AtomicInteger();
     //特殊的hash值
    private static final int HASH_INCREMENT = 0x61c88647;

这里定义了一个AtomicInteger类型,每次获取当前值并加上HASH_INCREMENT,HASH_INCREMENT = 0x61c88647,这个值跟斐波那契数列(黄金分割数)有关,其主要目的就是为了让哈希码能均匀的分布在2的n次方的数组里, 也就是Entry[] table中,这样做可以尽量避免hash冲突。

b. 关于& (INITIAL_CAPACITY - 1)

​ 计算hash的时候里面采用了hashCode & (size - 1)的算法,这相当于取模运算hashCode % size的一个更高效的实现。正是因为这种算法,我们要求size必须是2的整次幂,这也能保证在索引不越界的前提下,使得hash发生冲突的次数减小。

(3) ThreadLocalMap中的set方法

private void set(ThreadLocal<?> key, Object value) {
    
    
        ThreadLocal.ThreadLocalMap.Entry[] tab = table;
        int len = tab.length;
        //计算索引(重点代码,刚才分析过了)
        int i = key.threadLocalHashCode & (len-1);
        /**
         * 使用线性探测法查找元素(重点代码)
         */
        for (ThreadLocal.ThreadLocalMap.Entry e = tab[i];
             e != null;
             e = tab[i = nextIndex(i, len)]) {
    
    
            ThreadLocal<?> k = e.get();
            //ThreadLocal 对应的 key 存在,直接覆盖之前的值
            if (k == key) {
    
    
                e.value = value;
                return;
            }
            // key为 null,但是值不为 null,说明之前的 ThreadLocal 对象已经被回收了,
           // 当前数组中的 Entry 是一个陈旧(stale)的元素
            if (k == null) {
    
    
                //用新元素替换陈旧的元素,这个方法进行了不少的垃圾清理动作,防止内存泄漏
                replaceStaleEntry(key, value, i);
                return;
            }
        }
    
    	//ThreadLocal对应的key不存在并且没有找到陈旧的元素,则在空元素的位置创建一个新的Entry。
            tab[i] = new Entry(key, value);
            int sz = ++size;
            /**
             * cleanSomeSlots用于清除那些e.get()==null的元素,
             * 这种数据key关联的对象已经被回收,所以这个Entry(table[index])可以被置null。
             * 如果没有清除任何entry,并且当前使用量达到了负载因子所定义(长度的2/3),那么进行				 * rehash(执行一次全表的扫描清理工作)
             */
            if (!cleanSomeSlots(i, sz) && sz >= threshold)
                rehash();
}

 /**
     * 获取环形数组的下一个索引
     */
    private static int nextIndex(int i, int len) {
    
    
        return ((i + 1 < len) ? i + 1 : 0);
    }


代码执行流程:

A. 首先还是根据key计算出索引 i,然后查找i位置上的Entry,

B. 若是Entry已经存在并且key等于传入的key,那么这时候直接给这个Entry赋新的value值,

C. 若是Entry存在,但是key为null,则调用replaceStaleEntry来更换这个key为空的Entry,

D. 不断循环检测,直到遇到为null的地方,这时候要是还没在循环过程中return,那么就在这个null的位置新建一个Entry,并且插入,同时size增加1。

​ 最后调用cleanSomeSlots,清理key为null的Entry,最后返回是否清理了Entry,接下来再判断sz 是否>= thresgold达到了rehash的条件,达到的话就会调用rehash函数执行一次全表的扫描清理。

重点分析 : ThreadLocalMap使用线性探测法来解决哈希冲突的

​ 该方法一次探测下一个地址,直到有空的地址后插入,若整个空间都找不到空余的地址,则产生溢出。

​ 举个例子,假设当前table长度为16,也就是说如果计算出来key的hash值为14,如果table[14]上已经有值,并且其key与当前key不一致,那么就发生了hash冲突,这个时候将14加1得到15,取table[15]进行判断,这个时候如果还是冲突会回到0,取table[0],以此类推,直到可以插入。

​ 按照上面的描述,可以把Entry[] table看成一个环形数组。

猜你喜欢

转载自blog.csdn.net/m0_45364328/article/details/125022916