python多线程中使用锁的简单用例

àì夳堔傛蜴生んèń 2023-10-12 09:07 11阅读 0赞

在Python中，我们可以使用多种同步机制来保护共享资源的访问，包括锁、信号量、条件变量等。这里以锁为例，介绍如何使用锁来保护共享资源的访问。

首先，我们需要创建一个锁对象，使用`threading.Lock()`即可创建一个锁对象。然后，在访问共享资源的代码块前获取锁，在访问完成后释放锁。这样，同一时刻只有一个线程能够访问共享资源，从而避免了数据竞争和其他问题。

以下是一个简单的使用锁的示例，该示例启动两个线程并让它们分别对共享变量`count`进行加1操作。

import threading
    
    # 共享变量
    count = 0
    
    # 创建一个锁对象
    lock = threading.Lock()
    
    # 定义一个函数，对共享变量进行加1操作
    def increment():
        global count
        for i in range(10000):
            # 获取锁
            lock.acquire()
            count += 1
            # 释放锁
            lock.release()
    
    # 创建两个线程
    t1 = threading.Thread(target=increment)
    t2 = threading.Thread(target=increment)
    
    # 启动线程
    t1.start()
    t2.start()
    
    # 等待线程完成
    t1.join()
    t2.join()
    
    # 打印最终的计数值
    print("Final count: ", count)

在这个例子中，我们创建了一个锁对象`lock`，并定义了一个函数`increment`，它对共享变量`count`进行加1操作。在访问`count`变量前，我们使用`lock.acquire()`获取锁，在访问完成后使用`lock.release()`释放锁。这样，同一时刻只有一个线程能够访问`count`变量，从而避免了数据竞争和其他问题。

需要注意的是，在使用锁或其他同步机制时，要避免死锁问题，即多个线程因为互相等待对方释放锁而无法继续执行的情况。因此，在编写代码时，要仔细考虑锁的获取和释放顺序，避免出现死锁问题。