Python如何实现无中断原子代码执行？解决字典并发修改遍历失败问题

关于“无上下文切换”代码块的实现 #

在CPython中，由于全局解释器锁（GIL）的机制，线程切换会自动发生在字节码执行间隙、IO操作时或达到预设的字节码计数阈值时。严格意义上完全禁止上下文切换的代码块很难实现，因为GIL的释放逻辑是解释器内置的，强行通过底层API干预容易引发死锁或其他不稳定问题。

但如果你的核心需求是保证一段代码执行期间，其他线程不会执行冲突操作（逻辑上的原子性），可以用threading.Lock实现：

import threading

atomic_lock = threading.Lock()

with atomic_lock:
    do_stuff()
    do_other_stuff()

这种方式下，其他线程尝试获取同一把锁时会被阻塞，直到当前线程释放锁。虽然当前线程执行期间可能发生GIL切换，但其他线程拿到GIL后因为拿不到锁会立刻阻塞，最终效果是这段代码的执行不会被其他冲突操作打断。

遍历字典时防止修改的解决方案 #

针对你遍历字典时因其他线程修改导致报错的问题，有三种实用方案：

1. 用互斥锁保护所有字典操作 #

创建一个全局锁，所有访问（遍历、增删改）字典的代码都用这个锁包裹：

import threading

dict_lock = threading.Lock()
my_dict = {}

# 遍历操作
with dict_lock:
    for key in my_dict:
        # 处理逻辑
        print(key)

# 修改操作（所有修改都要加锁）
with dict_lock:
    my_dict["new_key"] = "new_value"

这种方式能从根本上避免遍历期间字典被修改，是最稳妥的方案。

2. 遍历字典的副本 #

如果不需要实时获取最新的字典内容，可以在遍历前生成字典键的副本：

# 遍历副本，原字典修改不影响遍历过程
for key in list(my_dict.keys()):
    print(key)

这种方式不需要加锁，实现简单，但遍历的是副本生成时刻的字典状态，无法反映后续的修改。

3. 封装线程安全字典类 #

自己实现一个带锁的字典类，把锁的逻辑封装起来，避免手动加锁的繁琐：

import threading

class ThreadSafeDict:
    def __init__(self):
        self._inner_dict = {}
        self._lock = threading.Lock()
    
    def __getitem__(self, key):
        with self._lock:
            return self._inner_dict[key]
    
    def __setitem__(self, key, value):
        with self._lock:
            self._inner_dict[key] = value
    
    def keys(self):
        with self._lock:
            # 返回键的副本，避免外部遍历期间字典被修改
            return list(self._inner_dict.keys())

# 使用示例
safe_dict = ThreadSafeDict()
safe_dict["a"] = 1

# 遍历
for key in safe_dict.keys():
    print(key)

# 修改
safe_dict["b"] = 2

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