ThreadLocal底层解析

package com.yzh.bank.utils;

import java.sql.*;
import java.util.Map;
import java.util.ResourceBundle;

public class DBUtil {
    private static ResourceBundle bundle = ResourceBundle.getBundle("resources/jdbc");
    private static String driver = bundle.getString("driver");
    private static String url = bundle.getString("url");;
    private static String user = bundle.getString("user");;
    private static String password = bundle.getString("password");;


    /**
     * 创建了一个ThreadLocal类，并且加了一个Connection参数.
     * 这个变量包含了：
     * Class ThreadLocal<T>{
     *      静态成员Map，在类加载是加载并且指加载一次
     *     private static Map<Thread,T> map = new HashMap<>();
     *
     *     根据key(当前线程获取)绑定 value(数据T)，
     *     在这里其实就是对当前线程绑定一个连接对象
     *     public void set(T obj){
     *        map.put(Thread.currentThread(),obj);
     *     }
     *
     *
     *     //根据当前线程返回key返回对应value(T)，在这里其实就是返回当前线程对应的连接对象
     *      public T get(){
     *         map.get(Thread.currentThread());
     *      }
     *
     *      //移除当前线程对应的value(T)，在这里其实就是移除当前线程对应的连接对象
     *      public void remove(){
     *      *         map.put(Thread.currentThread(),obj);
     *      *      }
     *
     * }
     *
     */
    //创建一个ThreadLocal对象，他的源码在上面，把T换成Connection
    private static ThreadLocal<Connection> local = new ThreadLocal<>();


    public DBUtil() {}
    static{
        try {
            Class.forName(driver);
        } catch (ClassNotFoundException e) {
            e.printStackTrace();
        }
    }
    public static Connection getConnection(){
        //Connection connection  = DriverManager.getConnection(url, user, password);
        //使用ThreadLocal获取连接对象(根据当前线程返回一个连接对象，但是注意了
        // 他底层不会帮我们真正DriverManager.getConnection(url, user, password);这样
        // 一个连接对象，他只会获取你已经存在的连接对象，那也就是说你第一次获取其实是获取连接对象的
        // 那么此时就得判断null，若为null则在这里创建一个连接对象),并且你还得把这个连接对象传入ThreadLocal
        //中，让他绑定当前线程，这样子只要你通过这个DBUtil去获取连接对象时他底层都会local.get()，然后判断
        //你的当前线程去拿到对应且唯一的那一个value值：连接对象。
        //最后，DBUtil工具类中conn.close();之前，要调用local.remove()，把当前这个连接对象
        //从ThreadLocal中的Map集合移除，因为Tomcat支持线程池，当你在这里conn.close();时
        // 他只会把这个连接对象放会连接池中，那很有可能别的对象可以从线程池中可以拿到这个你用过的
        //连接对象啊，那么此时这个连接对象还存在于ThreadLocal中的Map集合，那很显然这样就会出问题了。
        Connection conn = local.get();
        //若为null则创建
        if (conn == null) {
            try {
                conn  = DriverManager.getConnection(url, user, password);
                local.set(conn);
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }
        //若不为则直接返回
        return conn;
    }



    public  static void close(Connection conn, Statement statement, ResultSet resultSet){
        if(resultSet != null){
            try {
                resultSet.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }
        if(statement != null){
            try {
                statement.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }
        if(conn != null){
            try {
                conn.close();
                local.remove();
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }
    }
}