threading.local()使用与原理剖析
前言
还是第一次摘出某个方法来专门写一篇随笔,哈哈哈。
为什么要写这个方法呢?因为它确实太重要了,包括后期的Flask
框架源码中都有它的影子。
那么我们就来瞄一眼这个东西是啥吧。
作用
在Python官方中文文档中(Python3.8.4),对它的介绍其实并不是很详细
其实他的功能非常简单,如下:
在一个全局的容器中可以存放一些线程独有的数据,这些数据应是某一线程私有的,是除了本线程外的其他线程访问不到的。
举个例子,例如你用迅雷下载的时候每条线程的下载进度不一样,我们需要将下载进度这个数据存储起来,怎么存储?自己创建一个具有线程安全性的列表或字典?那太麻烦了,请记住一点,数据怎么存不重要,关键是要方便取,取的快,取的准才是王道,所以直接用
thrading.local()
方法即可。
我来画一张吧,灵魂画师上线,能用图描述绝对不打字。
基本使用
threading.local()
:可以实例化出一个寄存柜,这个寄存柜是全局化的
寄存柜对象.你想放的东西名字 = 东西
:你可以在线程中这样放入一个独有的物件
寄存柜对象.你放过的东西的名字
:这样,你就可以将你存放进的东西拿出来。注意,只有你才能拿出来,其他人拿不了。
import threading
import time
def operate(name):
"""操作"""
print("三个月后,{0}忽然想起了自己有件东西放在柜子里,决定取出来".format(name))
print("最后他从自己的柜子里取出来了:",Locker.article)
def people(article):
name = threading.current_thread().getName() # 获取线程名
Locker.article = article # 这就表示将私有的的一件物品放在了寄存柜里
print("{0}将{1}放在了寄存柜里...".format(name,article))
time.sleep(3) # 过了三个月,这期间发生了很多事
operate(name)
if __name__ == '__main__':
Locker = threading.local() # 好了寄存柜放在全局了
t1 = threading.Thread(target=people,args=("一封情书",),name="小王")
t2 = threading.Thread(target=people,args=("一双臭袜子",),name="小李")
t1.start()
t2.start()
t1.join()
t2.join()
# ==== 执行结果 ====
"""
小王将一封情书放在了寄存柜里...
小李将一双臭袜子放在了寄存柜里...
三个月后,小王忽然想起了自己有件东西放在柜子里,决定取出来
最后他从自己的柜子里取出来了: 一封情书
三个月后,小李忽然想起了自己有件东西放在柜子里,决定取出来
最后他从自己的柜子里取出来了: 一双臭袜子
"""
原理分析
我们可以自己做一个全局字典,来实现与这个类似的功能,但是使用起来肯定不太方便(它实际上本质就是字典嵌套进行存储的):
import threading
import time
Locker = {} # 好了寄存柜放在全局了
"""
{
线程id:{"article":"存放的具体物品"},
线程id:{"article":"存放的具体物品"},
线程id:{"article":"存放的具体物品"},
}
"""
def operate(name):
"""操作"""
print("三个月后,{0}忽然想起了自己有件东西放在柜子里,决定取出来".format(name))
ident = threading.get_ident() # 获取线程ID
print("最后他从自己的柜子里取出来了:",Locker[ident]["article"])
def people(article):
name = threading.current_thread().getName() # 获取线程名
ident = threading.get_ident() # 获取线程ID
Locker[ident] = {} # 创建了一个寄存柜的小格子
Locker[ident]["article"] = article # 这就表示将私有的的一件物品放在了寄存柜里
print("{0}将{1}放在了寄存柜里...".format(name,article))
time.sleep(3) # 过了三个月,这期间发生了很多事
operate(name)
if __name__ == '__main__':
t1 = threading.Thread(target=people,args=("一封情书",),name="小王")
t2 = threading.Thread(target=people,args=("一双臭袜子",),name="小李")
t1.start()
t2.start()
t1.join()
t2.join()
# ==== 执行结果 ====
"""
小王将一封情书放在了寄存柜里...
小李将一双臭袜子放在了寄存柜里...
三个月后,小李忽然想起了自己有件东西放在柜子里,决定取出来
最后他从自己的柜子里取出来了: 一双臭袜子
三个月后,小王忽然想起了自己有件东西放在柜子里,决定取出来
最后他从自己的柜子里取出来了: 一封情书
"""
尝试自己做出一个寄存柜
这样做是不是太麻烦了?我们可以自定义一个类,让它变得更加简单,如同原本的使用方法一样。其实下面代码中采取的方法也是threading.local()
所采取的方法。
import threading
import time
class MyLocker(object):
cabinet = {} # 柜子
"""
{
线程id:{"article":"存放的具体物品"},
线程id:{"article":"存放的具体物品"},
线程id:{"article":"存放的具体物品"},
}
"""
def __getattr__(self, item):
"""当访问属性不存在时触发"""
ident = threading.get_ident()
return MyLocker.cabinet[ident][item]
def __setattr__(self, key, value):
"""试图用 . 去设置属性时触发"""
ident = threading.get_ident()
if ident in MyLocker.cabinet: # 如果在柜子里这重新赋值
MyLocker.cabinet[ident][key] = value
else:
MyLocker.cabinet[ident] = {key: value} # 如果不在则做一个小格子,并且把物件存放进来
def operate(name):
"""操作"""
print("三个月后,{0}忽然想起了自己有件东西放在柜子里,决定取出来".format(name))
print("最后他从自己的柜子里取出来了:", Locker.article)
def people(article):
name = threading.current_thread().getName() # 获取线程名
Locker.article = article # 这就表示将私有的的一件物品放在了寄存柜里
print("{0}将{1}放在了寄存柜里...".format(name, article))
time.sleep(3) # 过了三个月,这期间发生了很多事
operate(name)
if __name__ == '__main__':
Locker = MyLocker() # 好了寄存柜放在全局了
t1 = threading.Thread(target=people, args=("一封情书",), name="小王")
t2 = threading.Thread(target=people, args=("一双臭袜子",), name="小李")
t1.start()
t2.start()
t1.join()
t2.join()
# ==== 执行结果 ====
"""
小王将一封情书放在了寄存柜里...
小李将一双臭袜子放在了寄存柜里...
三个月后,小王忽然想起了自己有件东西放在柜子里,决定取出来
最后他从自己的柜子里取出来了: 一封情书
三个月后,小李忽然想起了自己有件东西放在柜子里,决定取出来
最后他从自己的柜子里取出来了: 一双臭袜子
"""
来源:oschina
链接:https://my.oschina.net/u/4347380/blog/4334024