Python线程锁

软件发布|下载排行|最新软件

当前位置:首页IT学院IT技术

Python线程锁

LiveEveryDay   2022-05-23 我要评论

前言

当有多个线程,且它们同时访问同一资源时,需要考虑如何避免线程冲突。解决办法是使用线程锁。锁由Python的threading模块提供,并且它最多被一个线程所持有。当一个线程试图获取一个已经锁在资源上的锁时,该线程通常会暂停运行,直到这个锁被释放。看看下面的不具备锁功能的例子:

#!/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 函数并给出不同长度的时间,可能会让这个例子更有意思。无论如何,这里的问题是,一个线程可能已经调用 update_total 函数并且还没有更新完成,此时另一个线程也有可能调用它并且尝试更新内容。根据操作执行顺序的不同,该值可能只被增加一次。

给它添加锁后:

方式一:使用try/finally,确保锁肯定会被释放。

#!/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 的当前值并释放锁。

方式二:with语句避免使用try/finally。

事实上,我们可以使用 Python 的 with 语句避免使用 try/finally 这种较为繁琐的语句:

#!/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
'''

总结

Copyright 2022 版权所有 软件发布 访问手机版

声明:所有软件和文章来自软件开发商或者作者 如有异议 请与本站联系 联系我们