python学习笔记-ThreadLocal

例如下列程序中task()函数中定义的count变量就是局部变量。即使我们创建了两个线程，两者的count递增也不会相互影响，因为count是在task中定义的。

import threading


def task():
 count = 0
 for i in range(1000):
 count += 1
 print count


if __name__ == '__main__':
 t1 = threading.Thread(target=task)
 t1.start()
 t2 = threading.Thread(target=task)
 t2.start()

那么，这么处理是不是就完美了呢？其实还不是。
以上的例子我们举的是一个非常简单的例子，但是我们遇到一个比较复杂的业务逻辑的时候，比如多个局部变量，函数多重调用等，这么定义局部变量就会变得不简洁，麻烦。
函数多重调用是指，例如：
我们定义了函数，methodA(),这个方法体内调用了methodB(), methodB()方法体中又调用了methodC()...
如果我们在某一个线程中调用了methodA()并且使用了一个变量attr，那么我们就需要将attr一层一层地传递给后续的函数。

有没有一种方法，能让我们在线程中定义一个变量后，那么这个线程中的函数就都能调用，如此才叫简洁明了？
Python为我们做到了，那就是ThreadLocal.
ThreadLocal的用法只需要三步：

下面展示一下代码：

# coding=utf-8
import threading

local = threading.local() # 创建一个全局的对象


def task():
 local.count = 0 # 初始化一个线程内变量，该变量线程间互不影响。
 for i in range(1000):
 count_plus()


def count_plus():
 local.count += 1
 print threading.current_thread().name, local.count


if __name__ == '__main__':
 t1 = threading.Thread(target=task)
 t1.start()
 t2 = threading.Thread(target=task)
 t2.start()

更多python学习笔记 - ThreadLocal 相关文章请关注PHP中文网！