在Python中通过queue模块实现线程安全的队列

2023-02-27 00:00:00 线程队列模块

在Python中，queue模块提供了线程安全的队列实现，可以在多线程中安全地实现数据的共享和传递。queue模块提供了三种类型的队列：

Queue：普通队列，先进先出(FIFO)。
LifoQueue：后进先出(LIFO)队列。
PriorityQueue：优先级队列，按元素的优先级进行排序。
以下是一个使用Queue实现多线程的示例：

import threading
import queue

def worker(queue):
    """线程执行的任务"""
    while True:
        item = queue.get()
        if item is None:
            # 遇到空元素时退出循环
            break
        print(item)
        queue.task_done()

# 创建队列
queue = queue.Queue()

# 启动多个线程，共同处理队列中的数据
for i in range(5):
    t = threading.Thread(target=worker, args=(queue,))
    t.start()

# 向队列中添加数据
for i in range(10):
    queue.put(i)

# 等待队列中的所有数据被处理完成
queue.join()

# 添加空元素，让线程退出循环
for i in range(5):
    queue.put(None)

# 等待线程退出
for t in threading.enumerate():
    if t is not threading.current_thread():
        t.join()

在上面的示例中，首先创建了一个队列，并启动了多个线程来共同处理队列中的数据。然后向队列中添加数据，并等待队列中的所有数据被处理完成。最后向队列中添加空元素，让线程退出循环，等待线程退出。在这个过程中，线程之间安全地共享了队列中的数据，并按照FIFO的顺序进行处理。

相关文章