Python线程安全单例在单元测试中阻塞问题排查及方案咨询

你的单元测试出现阻塞，核心原因是使用了不可重入的threading.Lock，且单例类的初始化逻辑中可能存在递归调用自身构造函数的场景：

• threading.Lock是不可重入锁，同一线程在未释放锁的情况下再次尝试获取锁会直接阻塞。
• 当单元测试触发单例初始化时，如果__init__方法中又调用了该类的构造函数（比如MyClass()），当前线程会先进入__call__方法获取锁，然后在初始化过程中再次调用__call__，尝试第二次获取锁，导致线程死锁阻塞。
• 线上环境没有触发这种递归调用场景，所以运行正常。

另外，你的第二种实现（简单锁）还存在双重检查锁逻辑不完整的问题：获取锁后没有再次检查实例是否已存在，多线程下可能导致重复创建实例（虽然Python的GIL一定程度上缓解了这个问题，但逻辑上不严谨）。

方案1：使用可重入锁修复现有实现 #

将threading.Lock替换为threading.RLock，允许同一线程多次获取锁，避免递归初始化时的阻塞，同时完善双重检查锁的逻辑：

from threading import RLock


class Singleton(type):
    _instances = {}
    _lock: RLock = RLock()

    def __call__(cls, *args, **kwargs):
        if cls not in cls._instances:
            with cls._lock:
                # 锁内再次检查，避免多线程竞争时重复创建
                if cls not in cls._instances:
                    cls._instances[cls] = super().__call__(*args, **kwargs)
        return cls._instances[cls]

方案2：利用Python导入机制实现天然线程安全单例 #

Python的模块在导入时会自动保证线程安全，且整个程序生命周期内只会被初始化一次，这是最简单可靠的线程安全单例实现方式：

比如创建一个my_singleton.py模块：

# my_singleton.py
class MySingleton:
    def __init__(self):
        # 初始化逻辑
        pass

# 模块导入时创建实例，全局唯一
instance = MySingleton()

使用时直接导入：

from my_singleton import instance

方案3：基于装饰器的线程安全单例 #

如果需要更灵活的单例定义（比如给多个类添加单例特性），可以用装饰器结合RLock实现：

import functools
from threading import RLock


def singleton(cls):
    _instances = {}
    _lock = RLock()

    @functools.wraps(cls)
    def wrapper(*args, **kwargs):
        if cls not in _instances:
            with _lock:
                if cls not in _instances:
                    _instances[cls] = cls(*args, **kwargs)
        return _instances[cls]

    return wrapper

# 使用示例
@singleton
class MyClass:
    pass

内容的提问来源于stack exchange，提问作者Ema Il