在多線程環(huán)境下,每個(gè)線程都有自己的數(shù)據(jù)。一個(gè)線程使用自己的局部變量比使用全局變量好,因?yàn)榫植孔兞恐挥芯€程自己能看見毅桃,不會(huì)影響其他線程缸匪,而全局變量的修改必須加鎖翁狐。但是局部變量也有問題,就是在函數(shù)調(diào)用的時(shí)候凌蔬,傳遞起來很麻煩
如果用一個(gè)全局dict存放所有的Student對(duì)象露懒,然后以thread自身作為key獲得線程對(duì)應(yīng)的Student對(duì)象如何闯冷?這種方式理論上是可行的,它最大的優(yōu)點(diǎn)是消除了std對(duì)象在每層函數(shù)中的傳遞問題懈词,但是蛇耀,每個(gè)函數(shù)獲取std的代碼有點(diǎn)丑。
有沒有更簡(jiǎn)單的方式坎弯?ThreadLocal應(yīng)運(yùn)而生纺涤,不用查找dict,ThreadLocal幫你自動(dòng)做這件事:
import threading
# 創(chuàng)建全局ThreadLocal對(duì)象:
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é)果:
Hello, Alice (in Thread-A) Hello, Bob (in Thread-B)
全局變量local_school就是一個(gè)ThreadLocal對(duì)象抠忘,每個(gè)Thread對(duì)它都可以讀寫student屬性撩炊,但互不影響。你可以把local_school看成全局變量崎脉,但每個(gè)屬性如local_school.student都是線程的局部變量拧咳,可以任意讀寫而互不干擾,也不用管理鎖的問題囚灼,ThreadLocal內(nèi)部會(huì)處理骆膝。
可以理解為全局變量local_school是一個(gè)dict,不但可以用local_school.student灶体,還可以綁定其他變量阅签,如local_school.teacher等等。
ThreadLocal最常用的地方就是為每個(gè)線程綁定一個(gè)數(shù)據(jù)庫(kù)連接蝎抽,HTTP請(qǐng)求政钟,用戶身份信息等,這樣一個(gè)線程的所有調(diào)用到的處理函數(shù)都可以非常方便地訪問這些資源织中。