Python如何实现多线程安全队列？queue模块线程同步

queue

import queue
import threading
import time

# 创建一个线程安全的队列
q = queue.Queue()

def worker(num):
    while True:
        try:
            # 从队列中获取一个任务，如果没有任务则阻塞，直到有任务
            item = q.get(timeout=1) # 设置超时时间，避免无限阻塞
            print(f"线程 {num} 正在处理: {item}")
            time.sleep(1)  # 模拟处理任务的时间
            print(f"线程 {num} 完成处理: {item}")
        except queue.Empty:
            print(f"线程 {num} 退出，队列为空")
            break  # 队列为空，退出线程
        finally:
            q.task_done() # 通知队列，任务已完成

# 创建多个线程
threads = []
for i in range(3):
    t = threading.Thread(target=worker, args=(i,))
    threads.append(t)
    t.daemon = True  # 设置为守护线程，主线程退出时自动退出
    t.start()

# 向队列中添加任务
for i in range(10):
    q.put(i)

# 等待队列中的所有任务完成
q.join()

print("所有任务完成")

queue

q.get()

q.task_done()

q.join()

queue.Queue

maxsize

queue.Full

import queue
import threading
import time

q = queue.Queue(maxsize=5)  # 创建一个最大容量为5的队列

def producer():
    for i in range(10):
        try:
            q.put(i, timeout=1) # 设置超时时间，避免无限阻塞
            print(f"生产者添加: {i}")
            time.sleep(0.5)
        except queue.Full:
            print(f"队列已满，生产者等待...")
            time.sleep(1)  # 等待一段时间后重试

def consumer(num):
    while True:
        try:
            item = q.get(timeout=1)
            print(f"消费者 {num} 消费: {item}")
            time.sleep(1)
            q.task_done()
        except queue.Empty:
            print(f"消费者 {num} 退出，队列为空")
            break

# 创建生产者和消费者线程
producer_thread = threading.Thread(target=producer)
consumer_thread1 = threading.Thread(target=consumer, args=(1,))
consumer_thread2 = threading.Thread(target=consumer, args=(2,))

producer_thread.start()
consumer_thread1.start()
consumer_thread2.start()

producer_thread.join()
q.join() # 等待队列为空

print("所有任务完成")

queue.Queue

queue.Queue

queue.LifoQueue

queue.PriorityQueue

collections.deque

threading.Lock