美文网首页
Spring-利用ThreadLocal解决线程安全问题

Spring-利用ThreadLocal解决线程安全问题

作者: sherlock_6981 | 来源:发表于2018-02-28 09:23 被阅读0次

    ThreadLocal是什么

    ThreadLocal,顾名思义,它不是一个线程,而是线程的一个本地化对象。当工作于多线程中的对象使用ThreadLocal维护变量时,ThreadLocal为每个使用该变量的线程分配一个独立的变量副本。所以每一个线程都可以独立地改变自己的副本,而不会影响其他线程所对应的副本。从线程的角度看,这个变量就像是线程的本地变量,这也是类名中“Local”所要表达的意思。

    ThreadLocal的方法很简单,主要的就是4个方法

    1

    2

    3

    4

    5

    6

    7

    8

    9

    10

    11

    // 设置当前线程的线程局部变量的值

    voidset(T value)

    // 返回当前线程所对应的线程局部变量

    public Tget()

    // 将当前线程局部变量的值删除,目的是为了减少内存的占用,加快内存回收的速度

    public voidremove()

    // 返回该线程局部变量的初始值,让子类继承而设计的。这个方法是一个延迟调用方法,在线程第一次调用get()或set(Object)时才执行,并且仅执行1次。ThreadLocal中的默认实现直接返回一个null。

    protected TinitialValue()

    ThreadLocal的实现思路

    那么ThreadLocal是如何做到为每一个线程维护一份独立的变量副本呢?其实实现思路很简单:在ThreadLocal类中有一个Map,用于存储没一个线程的变量副本,Map中元素的键为线程对象,而值对应线程的变量副本。我们自己也可以提供一个简单的实现版本:

    1

    2

    3

    4

    5

    6

    7

    8

    9

    10

    11

    12

    13

    14

    15

    16

    17

    18

    19

    20

    21

    22

    23

    24

    25

    publicclass SimpleThreadLocal {

    privateMap valueMap = Collections.synchronizedMap(newHashMap<>());

    publicvoidset(ObjectnewValue) {

    valueMap.set(Thread.currentThread(), newValue);

    }

    publicObjectget() {

    Thread currentThread = Thread.currentThread();

    Objecto = valueMap.get(currentThread);

    if(o ==null&& !valueMap.containsKey(currentThread)) {

    o = initialValue();

    valueMap.put(currentThread, o);

    }

    returno;

    }

    publicvoidremove() {

    valueMap.remove(Thread.currentThread());

    }

    publicObjectinitialValue() {

    returnnull;

    }

    }

    ThreadLocal的运用

    下面是我在项目中运用ThreadLocal来实现多线程情况下,用户信息的管理Session:

    1

    2

    3

    4

    5

    6

    7

    8

    9

    10

    11

    12

    13

    14

    15

    16

    17

    18

    19

    20

    21

    22

    23

    24

    25

    26

    27

    28

    29

    30

    31

    32

    33

    34

    35

    36

    37

    38

    packagecom.benjamin.common.session;

    importjava.util.HashMap;

    importjava.util.Map;

    /**

    * Created by piqiu on 2/27/16.

    * 用户Session信息保管类

    */

    publicclass UserSession {

    /** 保存用户的信息 **/

    privatestaticfinalThreadLocal> userThreadLocal =newThreadLocal<>();

    protectedUserSession(){}

    publicstaticObjectget(Stringattribute) {

    Map userInfo = userThreadLocal.get();

    Objecto =null;

    if(userInfo !=null) {

    o = userInfo.get(attribute);

            }

    returno;

        }

    publicstatic Tget(Stringattribute, Class clazz) {

    return(T)get(attribute);

        }

    publicstaticvoidset(Objectattribute,Objectvalue) {

    Map userInfo = userThreadLocal.get();

    if(userInfo ==null) {

    userInfo =newHashMap<>();

    userThreadLocal.set(userInfo);

            }

            userInfo.put(attribute, value);

        }

    }

    ThreadLocal与Thread同步机制的比较

    ThreadLocal和线程同步机制都是为了解决多线程中相同变量的访问冲突问题,那么,ThreadLocal和线程同步机制相比有什么优势呢?

    在同步机制中,通过对象的锁机制保证同一时间只有一个线程访问变量。这时该变量是多个线程共享的,使用同步机制要求程序缜密的分析什么时候对变量进行读写,什么时候需要锁定某个对象,什么时候释放对象锁等繁杂的问题,程序设计和编写难度相对较大

    而ThreadLocal则从另一个角度来解决多线程的并发访问。ThreadLocal为每一个线程提供一个独特的变量副本,从而隔离了多个线程对访问数据的冲突。因为每一个线程都拥有自己的变量副本,从而也就没有必要对该变量进行同步了。ThreadLocal提供了线程安全的对象封装,在编写多线程代码时,可以把不安全的变量封装进ThreadLocal。

    概括起来说,对于多线程资源共享的问题,同步机制采用了“以时间换空间”的方式:访问串行化,对象共享化。而ThreadLocal采用了“以空间换时间”的方式:访问并行化,对象独享化。前者仅提供一份变量,让不同的线程排队访问,而后者为每一个线程都提供了一份变量,因此可以同时访问而互不影响。

    Spring利用ThreadLocal解决线程安全问题

    我们知道在一般情况下,只有无状态的Bean才可以在多线程环境下共享,在Spring中,绝大部分Bean都可以声明为singleton作用域。就是因为Spring对一些Bean(如RequestContextHolder、TransactionSynchronizationManager、LocaleContextHolder)中非线程安全的“状态性对象”采用ThreadLocal进行封装,让它们也成为线程安全的“状态性对象”,因为有状态的Bean就能够以singleton方式在多线程中正常工作了。

    通过查看TransactionSynchronizationManager源码可以发现Spring事务的工作机制,我们可以自己利用ThreadLocal来实现自定义的Session和Transaction管理:

    1

    2

    3

    4

    5

    6

    7

    8

    9

    10

    11

    12

    13

    14

    15

    16

    17

    18

    19

    20

    21

    22

    23

    24

    25

    26

    27

    28

    29

    30

    31

    32

    33

    34

    35

    36

    37

    38

    39

    40

    41

    42

    43

    44

    45

    46

    47

    48

    49

    50

    51

    52

    53

    54

    55

    56

    57

    58

    59

    60

    61

    62

    63

    64

    65

    66

    67

    68

    69

    70

    71

    72

    73

    74

    75

    76

    77

    78

    79

    80

    81

    82

    83

    84

    85

    86

    87

    88

    89

    90

    91

    92

    93

    94

    95

    96

    97

    packagecom.benjamin.common;

    importorg.hibernate.Session;

    importorg.hibernate.SessionFactory;

    importorg.hibernate.Transaction;

    importorg.hibernate.cfg.Configuration;

    importorg.slf4j.Logger;

    importorg.slf4j.LoggerFactory;

    importjava.io.File;

    /**

    * Created by piqiu on 2/27/16.

    */

    publicclass HibernateUtil {

    privatestaticfinalLogger logger = LoggerFactory.getLogger(HibernateUtil.class);

    /** 存储hibernate session的ThreadLocal **/

    privatestaticfinalThreadLocal sessionThreadLocal =newThreadLocal<>();

    /** 存储事务n的ThreadLocal **/

    privatestaticfinalThreadLocal transactionThreadLocal =newThreadLocal<>();

    /** session工厂 **/

    privatestaticSessionFactory sessionFactory;

    /**

        * 初始化sessionFactory

        * @param file

        */

    publicstaticvoidinitSessionFactory(File file) {

    Configuration configuration =newConfiguration();

            configuration.configure(file);

            sessionFactory = configuration.buildSessionFactory();

        }

    /**

        * 取得当前线程绑定的session

        * @return

        */

    publicstaticSession currentSession() {

    Session session = sessionThreadLocal.get();

    if(session ==null) {

                session = sessionFactory.openSession();

    sessionThreadLocal.set(session);

            }

    returnsession;

        }

    /**

        * 关闭session

        */

    publicstaticvoidcloseSession() {

    Session session = sessionThreadLocal.get();

    if(session !=null) {

    sessionThreadLocal.set(null);

    session.clear();

                session.close();

            }

        }

    /**

        * 取得当前session的事务

        * @return

        */

    publicstaticTransaction transaction() {

    Transaction transaction = transactionThreadLocal.get();

    if(transaction ==null) {

                transaction = currentSession().beginTransaction();

    transactionThreadLocal.set(transaction);

            }

    returntransaction;

        }

    /**

        * 提交事务

        */

    publicstaticvoidcommitTransaction() {

    Transaction transaction = transactionThreadLocal.get();

    transactionThreadLocal.set(null);

    if(transaction !=null) {

                transaction.commit();

            }

        }

    /**

        * 回滚事务

        */

    publicstaticvoidrollbackTransaction() {

    Transaction transaction = transactionThreadLocal.get();

    transactionThreadLocal.set(null);

    if(transaction !=null) {

                transaction.rollback();

            }

        }

    }

    相关文章

      网友评论

          本文标题:Spring-利用ThreadLocal解决线程安全问题

          本文链接:https://www.haomeiwen.com/subject/owtbxftx.html