【18.6 python中的线程通信】

python中的线程通信

---

在Python中，线程通信是多个线程之间交换信息或数据的过程。由于线程共享进程的内存空间，因此它们可以直接访问和修改同一份数据。然而，这种共享访问也带来了数据竞争和同步问题。为了确保线程之间的正确通信和数据一致性，Python提供了多种线程通信机制。

1. 共享全局变量

最简单直接的方式是通过共享全局变量来进行线程间通信。然而，这种方式需要仔细管理同步问题，以避免数据竞争和条件竞争。

import threading

# 全局变量
shared_data = 0

def worker():
    global shared_data
    shared_data += 1

# 创建并启动线程
t1 = threading.Thread(target=worker)
t2 = threading.Thread(target=worker)

t1.start()
t2.start()

# 等待线程完成
t1.join()
t2.join()

print(shared_data)  # 可能的结果不是2，因为存在数据竞争

2. 使用锁（Locks）

锁是线程同步的基本机制之一，它可以防止多个线程同时访问共享资源。在Python中，threading模块提供了Lock类来实现锁。

import threading

shared_data = 0
lock = threading.Lock()

def worker():
    global shared_data
    with lock:  # 使用with语句自动管理锁的获取和释放
        shared_data += 1

# 创建并启动线程
t1 = threading.Thread(target=worker)
t2 = threading.Thread(target=worker)

t1.start()
t2.start()

t1.join()
t2.join()

print(shared_data)  # 正确输出2

3. 事件（Events）

Event是线程间通信的另一种方式，它允许一个线程等待另一个线程的通知。Event对象管理一个内部标志，该标志可以被设置为True或False。线程可以等待这个标志变为True，或者设置这个标志为True以通知等待的线程。

import threading

def worker(event):
    print('Worker waiting for event')
    event.wait()  # 等待事件被设置
    print('Worker got event')

event = threading.Event()

t = threading.Thread(target=worker, args=(event,))
t.start()

# 做一些工作...

# 通知线程
event.set()

4. 条件变量（Condition）

条件变量比锁更高级，它允许线程在等待某个条件成立时挂起。当条件变量上的某个条件满足时，一个或多个等待的线程将被唤醒以继续执行。

import threading

def worker(cond, item):
    with cond:
        print(f'Worker {item} is waiting')
        cond.wait()  # 等待条件变量上的某个条件满足
        print(f'Worker {item} was notified')

cond = threading.Condition()

t1 = threading.Thread(target=worker, args=(cond, 'A'))
t2 = threading.Thread(target=worker, args=(cond, 'B'))

t1.start()
t2.start()

# 做一些工作...

with cond:
    # 通知所有等待的线程
    cond.notify_all()

5. 队列（Queues）

队列是线程安全的，它们被设计用来在多线程程序中安全地交换数据。queue模块提供了多种队列实现，包括线程安全的队列。

import queue
import threading

def worker(q):
    while True:
        item = q.get()  # 从队列中获取项目
        if item is None:
            break
        print(f'Processing {item}')
        q.task_done()  # 表示前一个项目已处理完毕

q = queue.Queue()

t = threading.Thread(target=worker, args=(q,))
t.start()

# 发送数据到线程
q.put('some work')
q.put(None)  # 发送结束信号

t.join()

以上是Python中线程通信的一些主要机制。选择哪种机制取决于你的具体需求和场景。