首頁 > 後端開發 > Python教學 > python學習筆記 - ThreadLocal

python學習筆記 - ThreadLocal

高洛峰
發布: 2017-02-16 11:03:01
原創
1343 人瀏覽過

我們在寫多執行緒程式的時候,往往會遇到兩種類型的變數。

  • 一種是全域變量,多個執行緒共享。為了避免改亂為,我們在前面已經提到要加鎖。

  • 一種是局部變數。僅供一個執行緒使用,線程間相互不影響。

例如下列程式中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中文網!


相關標籤:
來源:php.cn
本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板