Python学习笔记__10.3章 ThreadLocal

1、概览

在多线程环境下,每个线程都有自己的数据。一个线程使用自己的局部变量比使用全局变量好,因为局部变量只有线程自己能看见,不会影响其他线程,而全局变量的修改必须加锁。但线程的局部变量,在函数调用时很麻烦

  1. 方式一:一层一层传

def process_student(name):

std = Student(name) # std是局部变量,但是每个函数都要用它,因此必须传进去:

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)

  1. 方式二:用一个全局dict存放所有的Student对象,然后以thread自身作为key获得线程对应的Student对象

global_dict = {}

def std_thread(name):

std = Student(name)

global_dict[threading.current_thread()] = std # 以thread自身作为key,把std放到全局变量global_dict中

do_task_1()

do_task_2()

def do_task_1():

std = global_dict[threading.current_thread()] # 不传入std,而是根据当前线程查找

def do_task_2()

std = global_dict[threading.current_thread()] # 任何函数都可以查找出当前线程的std变量

  1. 方式三:ThreadLocal

import threading

local_school = threading.local() # 创建全局ThreadLocal对象

def process_student():

std = local_school.student # 获取当前线程绑定在student的变量

print('Hello, %s (in %s)' % (std, threading.current_thread().name))

def process_thread(name):

local_school.student = name # 为ThreadLocal加student属性,绑定线程变量

process_student() # 调用函数

t1 = threading.Thread(target= process_thread, args=('Alice',), name='Thread-A') #创建线程,线程名为Thread-A

t2 = threading.Thread(target= process_thread, args=('Bob',), name='Thread-B')

t1.start()

t2.start()

t1.join()

t2.join()

# 执行结果:

Hello, Alice (in Thread-A)

Hello, Bob (in Thread-B)

全局变量local_school是一个ThreadLocal对象。可以把它看成全局变量。任何线程绑定变量在它的不同属性上。以达成互不干扰的效果

2、小结

ThreadLocal最常用的地方就是为每个线程绑定一个数据库连接,HTTP请求,用户身份信息等,这样一个线程的所有调用到的处理函数都可以非常方便地访问这些资源。

一个ThreadLocal变量虽然是全局变量,但每个线程都只能读写自己线程的独立副本,互不干扰。ThreadLocal解决了参数在一个线程中各个函数之间互相传递的问题。

猜你喜欢

转载自blog.51cto.com/12758568/2117003