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的用法只需要三步驟：

• 定義一個物件 threading.local

• 在線程內給該物件綁定參數。所有綁定的參數都是執行緒隔離的。

• 在執行緒內呼叫。

下面展示一下程式碼：

# 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中文網！