Python中ThreadLocal變量是什么

這篇文章主要介紹了Python中ThreadLocal變量是什么，具有一定借鑒價值，需要的朋友可以參考下。希望大家閱讀完這篇文章后大有收獲。下面讓小編帶著大家一起了解一下。

ThreadLocal

在多線程環(huán)境下，每個線程都有自己的數(shù)據(jù)。一個線程使用自己的局部變量比使用全局變量好，因為局部變量只有線程自己能看見，不會影響其他線程，而全局變量的修改必須加鎖。

但是局部變量也有問題，就是在函數(shù)調(diào)用的時候，傳遞起來很麻煩：

def process_student(name):
    std = Student(name)
    # std是局部變量，但是每個函數(shù)都要用到它，因此必須傳進去
    do_task_1(std)
    do_task_2(std)
def do_task_1(std):
    do_subtask_1(std)
    do_subtask_2(std)
def do_task_2(std):
    do_subtask_2(std)
    do_subtask_2(std)

每個函數(shù)一層一層調(diào)用都這么傳參數(shù)那還得了？用全局變量？也不行，因為每個線程處理的不同Student對象，不能共享。

如果用一個全局dict存放所有的Student對象，然后以thread自身作為key獲得線程對應(yīng)的Student對象如何？

global_dict = {}
def std_thread(name):
    std = Student(name)
    # 把std放到全局變量global_dict中：
    global_dict[threading.current_thread()] = std
    do_task_1()
    do_task_2()
def do_task_1():
    # 不傳入std,而是根據(jù)當(dāng)前線程查找：
    std = global_dict[threading.current_thread()]
    ...
def do_task_2(arg):
    # 任何函數(shù)都可以查找出當(dāng)前線程的std變量
    std = global_dict[threading.current_thread()]
    ...

這種方式理論上是可行的，它最大的優(yōu)點是消除了std對象在每層函數(shù)中的傳遞問題，但是，每個函數(shù)獲取std的代碼有點丑。

有沒有更簡單的方式？

ThreadLocal應(yīng)運而生，不用查找dict，ThreadLocal幫你自動做這件事：

# _*_ coding:utf-8 _*_
import threading
# 創(chuàng)建全局ThreadLocal對象：
local_school = threading.local()
def process_student():
    # 獲取當(dāng)前線程關(guān)聯(lián)的student
    std = local_school.student
    print('Hello, %s (in %s)' % (std, threading.current_thread().name))
def process_thread(name):
    # 綁定ThreadLocal的student
    local_school.student = name
    process_student()
t1 = threading.Thread(target=process_thread, args=('Alice',), name='Thread-A')
t2 = threading.Thread(target=process_thread, args=('Bob',), name='Thread-B')
t1.start()
t2.start()
t1.join()
t2.join()
# 執(zhí)行結(jié)果
$ python thread_local.py
Hello, Alice (in Thread-A)
 Hello, Bob (in Thread-B)
$ python thread_local.py
Hello, Alice (in Thread-A)
Hello, Bob (in Thread-B)

全局變量local_school就是一個ThreadLocal對象，每個Thread對它都可以讀寫student屬性，但互不影響。你可以把local_school看成是全局變量，但每個屬性例如local_school.student都是線程的局部變量，可以任意讀寫而互不干擾，也不用管理鎖的問題，ThreadLocal內(nèi)部會處理。

可以理解為全局變量local_school是一個dict，不但可以用local_school.student，還可以綁定其它變量，如local_school.teacher等等。

ThreadLocal最常用的地方就是為每個線程綁定一個數(shù)據(jù)庫連接，HTTP請求，用戶身份信息等，這樣一個線程的所有調(diào)用到的處理函數(shù)都可以非常方便地訪問這些資源。

一個ThreadLocal變量雖然是全局變量，但每個線程都只能讀寫自己線程的獨立副本，互不干擾。ThreadLocal解決了參數(shù)在一個線程中各個函數(shù)之間互相傳遞的問題。

感謝你能夠認真閱讀完這篇文章，希望小編分享Python中ThreadLocal變量是什么內(nèi)容對大家有幫助，同時也希望大家多多支持億速云，關(guān)注億速云行業(yè)資訊頻道，遇到問題就找億速云，詳細的解決方法等著你來學(xué)習(xí)!