Python 中的重入锁（RLock）详解

最新推荐文章于 2025-03-06 11:32:52 发布

有馬公生

最新推荐文章于 2025-03-06 11:32:52 发布

阅读量531

点赞数 4

分类专栏： Python以及爬虫操作文章标签：开发语言 python

本文链接：https://blog.csdn.net/Wayne_zzx1112233/article/details/144711376

版权

Python以及爬虫操作专栏收录该内容

8 篇文章

订阅专栏

引言

在多线程和多进程编程中，同步机制是确保数据一致性和线程安全的关键。重入锁（Reentrant Lock），又称递归锁，是一种特殊的锁，允许同一个线程多次获取同一把锁。本文将详细介绍重入锁的原理，并提供一个简单的 Python 实现，同时对比线程和进程的概念。

线程与进程

进程（Process）

定义：进程是操作系统进行资源分配和调度的一个独立单位，是应用程序的实例。
内存：每个进程都有自己的独立内存空间，包括代码段、数据段和堆栈。
创建和管理：进程的创建和管理涉及到较为复杂的系统资源管理，开销较大
通信：进程间通信（IPC）需要通过特定的机制，如：管道、信号、共享内存、消息队列

线程（Thread）

定义：线程是进程中的一个实体，是被系统独立调度和分配的基本单位。线程自身不拥有系统资源，只拥有少部分在运行中必不可少的资源（如执行栈），但它可以与同属一个进程的其他线程共享进程所拥有的全部资源。
内存：同一进程内的线程共享进程的内存空间，如：全局变量、静态变量等。
创建和管理：线程的创建和管理开销相对较小，因为它们不需要独立的内存空间。
通信：线程间可以直接通过读写共享变量来进行通信，无需特殊的IPC机制。

重入锁的原理

锁的基本概念

用于确保在任何时刻只有一个线程可以访问特定的资源，如果一个线程获取了锁，其他线程必须等待直到锁被释放。

重入锁的特点

可重入性：同一个线程可以多次获取同一把锁，而不会被阻塞。
计数机制：内部维护一个计数器，记录同一个线程获取锁的次数。
释放条件：只有当计数器减至零时，锁才被真正释放。

为什么需要重入锁

重入锁适用于递归函数或嵌套调用，其中同一个线程需要多次获取同一把锁。

Python 实现重入锁

以下是一个简化的 Python 实现，展示了重入锁的基本逻辑：

import threading

# 获取当前线程的ID
def get_current_thread_id():
    return threading.current_thread().ident

class ReentrantLock:
    def __init__(self):
        self.owner = None
        self.count = 0

    def acquire(self):
        current_thread_id = get_current_thread_id()
        if self.owner == current_thread_id:
            self.count += 1
            print(f"Thread {current_thread_id}: Acquired lock (reentrant). Count = {self.count}")
            return
        while True:
            if self.owner is None:
                self.owner = current_thread_id
                self.count = 1
                print(f"Thread {current_thread_id}: Acquired lock. Count = {self.count}")
                return

    def release(self):
        current_thread_id = get_current_thread_id()
        if self.owner != current_thread_id:
            raise RuntimeError("Cannot release an un-acquired lock")
        self.count -= 1
        print(f"Thread {current_thread_id}: Released lock. Count = {self.count}")
        if self.count == 0:
            self.owner = None
            print(f"Thread {current_thread_id}: Lock is now free.")

# 使用重入锁
lock = ReentrantLock()
lock.acquire()  # 线程1获取锁
lock.acquire()  # 线程1再次获取锁（重入）
lock.release()  # 线程1释放锁
lock.release()  # 线程1再次释放锁