快捷導(dǎo)航

Python中線程鎖的使用介紹

更新時(shí)間：2022年01月07日 09:30:23 作者：LiveEveryDay

大家好，本篇文章主要講的是Python中線程鎖的使用介紹，感興趣的同學(xué)趕快來(lái)看一看吧，對(duì)你有幫助的話記得收藏一下，方便下次瀏覽

前言

當(dāng)有多個(gè)線程，且它們同時(shí)訪問(wèn)同一資源時(shí)，需要考慮如何避免線程沖突。解決辦法是使用線程鎖。鎖由Python的threading模塊提供，并且它最多被一個(gè)線程所持有。當(dāng)一個(gè)線程試圖獲取一個(gè)已經(jīng)鎖在資源上的鎖時(shí)，該線程通常會(huì)暫停運(yùn)行，直到這個(gè)鎖被釋放。看看下面的不具備鎖功能的例子：

#!/usr/bin/env python3
# -*- coding:utf-8 -*-
# Author: LiveEveryDay
 
 
import threading
 
total = 0
 
 
def update_total(amount):
    global total
    total += amount
    print(total)
 
 
if __name__ == '__main__':
    for i in range(10):
        my_thread = threading.Thread(target=update_total, args=(5,))
        my_thread.start()
 
''' ------ Running Results ------
510
15
20
25
30
3540
45
50
'''

如果往以上代碼添加 time.sleep 函數(shù)并給出不同長(zhǎng)度的時(shí)間，可能會(huì)讓這個(gè)例子更有意思。無(wú)論如何，這里的問(wèn)題是，一個(gè)線程可能已經(jīng)調(diào)用 update_total 函數(shù)并且還沒(méi)有更新完成，此時(shí)另一個(gè)線程也有可能調(diào)用它并且嘗試更新內(nèi)容。根據(jù)操作執(zhí)行順序的不同，該值可能只被增加一次。

給它添加鎖后：

方式一：使用try/finally，確保鎖肯定會(huì)被釋放。

#!/usr/bin/env python3
# -*- coding:utf-8 -*-
# Author: LiveEveryDay
 
 
import threading
 
total = 0
lock = threading.Lock()
 
 
def update_total(amount):
    global total
    lock.acquire()
    try:
        total += amount
    finally:
        print(total)
        lock.release()
 
 
if __name__ == '__main__':
    for i in range(10):
        my_thread = threading.Thread(target=update_total, args=(5,))
        my_thread.start()
 
''' ------ Running Results ------
5
10
15
20
25
30
35
40
45
50
'''

如上，在我們做任何處理之前就獲取鎖。然后嘗試更新 total 的值，最后打印出 total 的當(dāng)前值并釋放鎖。

方式二：with語(yǔ)句避免使用try/finally。

事實(shí)上，我們可以使用 Python 的 with 語(yǔ)句避免使用 try/finally 這種較為繁瑣的語(yǔ)句：

#!/usr/bin/env python3
# -*- coding:utf-8 -*-
# Author: LiveEveryDay
 
 
import threading
 
total = 0
lock = threading.Lock()
 
 
def update_total(amount):
    global total
    with lock:
        total += amount
        print(total)
 
 
if __name__ == '__main__':
    for i in range(10):
        my_thread = threading.Thread(target=update_total, args=(5,))
        my_thread.start()
 
''' ------ Running Results ------
5
10
15
20
25
30
35
40
45
50
'''