在多线程环境下,每个线程都有自己的数据。一个线程使用自己的局部变量比使用全局变量好,因为局部变量只有线程自己能看见,不会影响其他线程,而全局变量的修改必须加锁。
但是局部变量也有问题,就是在函数调用的时候,传递起来很麻烦。
一个参数可能会传递很多次
ThreadLocal就是为此而准备的。
import threading
#一个ThreadLocal对象
local_user_file = threading.local()
def process_user():
# 获取当前线程关联的user
user = local_user_file.user
filename = local_user_file.filename
print('usr : %s download : %s (thread:%s)' % (user, filename, threading.current_thread().name))
def process_thread(name, filename):
# 绑定ThreadLocal的user
local_user_file.user = name
local_user_file.filename = filename
process_user()
if __name__=='__main__':
t1 = threading.Thread(target=process_thread, args=('Tom', '五年高考三年模拟'), name='Tom-thread')
t2 = threading.Thread(target=process_thread, args=('Jack', 'C语言'), name='Jack-thread')
t1.start()
t2.start()
t1.join()
t2.join()
#打印结果
usr : Tom download : 五年高考三年模拟 (thread:Tom-thread)
usr : Jack download : C语言 (thread:Jack-thread)
全局变量local_school就是一个ThreadLocal对象,每个Thread对它都可以读写student属性,但互不影响。你可以把local_school看成全局变量,但每个属性如local_school.student都是线程的局部变量,可以任意读写而互不干扰,也不用管理锁的问题,ThreadLocal内部会处理。
可以理解为全局变量local_school是一个dict,不但可以用local_school.student,还可以绑定其他变量,如local_school.teacher等等。
ThreadLocal最常用的地方就是为每个线程绑定一个数据库连接,HTTP请求,用户身份信息等,这样一个线程的所有调用到的处理函数都可以非常方便地访问这些资源。