Python 线程加锁:范围越小越好,但这真的总是对的吗?
python线程中加锁范围越小越好
问题说明
在python线程中,使用锁保证多个线程对共享数据的并发访问时,加锁的范围是一个重要的问题,是将锁放在循环外面还是里面。
对比两种情况
把锁放在循环外面:
- 代码如下:
from threading import thread, lock import time num = 0 mutex = lock() # 创建一把互斥锁 def test1(): for i in range(1000000): # 上锁 mutex.acquire() num += 1 # 解锁 mutex.release() def test2(): for i in range(1000000): # 上锁 mutex.acquire() num += 1 # 解锁 mutex.release() if __name__ == "__main__": start_time = time.time() # 开始时间 p1 = thread(target=test1) p1.start() p2 = thread(target=test2) p2.start() p1.join() p2.join() end_time = time.time() # 结束时间 print("运行时间:%.2fs" % (end_time - start_time))
- 特点:加锁范围大,整个循环体都被锁住了。
- 优点:简单易实现,可以保证共享数据的原子性。
- 缺点:效率较低,因为整个循环都会被阻塞,导致其他线程无法访问共享数据。
把锁放在for里面:
- 代码如下:
from threading import Thread, Lock import time num = 0 mutex = Lock() # 创建一把互斥锁 def test1(): for i in range(1000000): if mutex.acquire(False): # 尝试获取锁,成功则返回True,否则返回False num += 1 mutex.release() # 释放锁 def test2(): for i in range(1000000): if mutex.acquire(False): # 尝试获取锁,成功则返回True,否则返回False num += 1 mutex.release() # 释放锁 if __name__ == "__main__": start_time = time.time() # 开始时间 p1 = Thread(target=test1) p1.start() p2 = Thread(target=test2) p2.start() p1.join() p2.join() end_time = time.time() # 结束时间 print("运行时间:%.2fs" % (end_time - start_time))
- 特点:加锁范围小,只有对共享数据的操作时才会加锁。
- 优点:效率较高,只有需要时才加锁,其他时间线程可以并行运行。
- 缺点:实现起来较复杂,需要对代码进行较多的改动。
选择建议
根据实际需求来选择加锁范围。
- 如果需要保证数据绝对的原子性,可以将锁放在循环外面。
- 如果效率是首要考虑因素,可以将锁放在for里面。
常见的错误
- 无锁:不使用锁对共享数据并发访问,会导致数据不一致。
- 死锁:多个线程相互等待释放锁,导致程序无法继续执行。