废话不多说,直接进入正题
使用:
声明一个公共变量a = threading.local()
然后线程里使用这个变量 例如 a.x = set()
各个线程的a.x都是独立的, 也就是说每个线程对该set的操作都是独立的
实现:
由于没有init方法那就看new方法,实力化了_localimpl对象, 并且调用了这个对象的 create_dict()方法,并通过setattr给local对象增加了_localimpl的引用。
接下来看create_dict方法:
这里通过current_thread获取到当前线程,idt就是当前线程的id, localdict就是当前线程使用时变量的保存处(详情可以去看__getattr__和__setattr__方法),(a=threading.local() a.x=111 也就是这种形式,此时的x就保存到localdict里),然后把idt当作key,localdict当作value保存到self.dicts里,其实到这里已经很清楚了,线程的id为key,用户在对threading.local()对象进行赋值取值(getattr__和__setattr)时,只会对操作到自己线程id作为key对应到的值。
也就是说线程越多,self.dicts里的值就会越来越多,所以一定要考虑资源回收的问题
所以,即当线程结束时要从self.dicts里清除掉该线程对应的数据,那么如何实现呢?
这里用到一个weakref的技巧,
class weakref.ref(object[, callback])
创建一个弱引用对象,object是被引用的对象,callback是回调函数(当被引用对象被删除时的,会调用改函数)。
注意图中的wrthread,其callback函数是thread_deleted,即当线程结束时就会调用该函数,通过wrlocal获取到self(即_localimpl)对象,再调用dicts.pop(id)即可
不过要注意的是,当弱引用对象没有引用时callback不会生效,也就是说当wrthread这个对象不存在引用时thread_deleted不会生效,图中,在self.dicts保存了一份wrthread, 可以保证
当create_dict函数退出了,wrthread仍被引用,并且除此之外local_deleted和thread_deleted两个函数中都使用了对方的弱引用,也可以确保create_dict函数退出时弱引用声明callback不会失效。
题外话:
这里搞local_deleted貌似没什么实际意义啊,往threading.__dict__里放_localimpl也很疑惑啊,谁会通过threading去反向获取_localimpl呢,如果仅仅是为了保证弱引用被引用,self.dicts[idt] = wrthread, localdict这里已经做到了啊。
(当然也可能是我没有使用到)