Python多路复用Queue：实现类似Go select语句的方案-Golang-PHP中文网

Python多路复用Queue：实现类似Go select语句的方案

花韻仙語

发布： 2025-08-20 21:42:55

原创

667人浏览过

python多路复用queue：实现类似go select语句的方案

Python多路复用Queue：实现类似Go select语句的方案

在Go语言中，select语句允许同时监听多个channel，并在其中一个channel准备好时执行相应的操作。这种机制在并发编程中非常有用。然而，Python标准库中的queue.Queue并不直接支持类似的功能，即无法同时阻塞等待多个队列。本文将介绍几种在Python中实现类似多路复用queue.Queue的方法。

正如摘要中所述，Python的queue.Queue不具备Go语言select语句的功能。这意味着我们需要寻找一些替代方案来实现类似的效果。常见的解决方案包括轮询和使用通知队列。

轮询（Polling）

最直接的方法是使用轮询。这种方法会定期检查每个队列，看是否有数据可读。

import queue
import time

c1 = queue.Queue()
c2 = queue.Queue()

while True:
    try:
        i1 = c1.get_nowait()
        print("received %s from c1" % i1)
    except queue.Empty:
        pass
    try:
        i2 = c2.get_nowait()
        print("received %s from c2" % i2)
    except queue.Empty:
        pass
    time.sleep(0.1)

登录后复制

优点：

立即学习“Python免费学习笔记（深入）”；

简单易懂，实现起来非常直接。

缺点：

CPU占用率高： 即使队列为空，程序也会不断循环检查，浪费CPU资源。
响应延迟： 轮询间隔会影响响应速度。如果轮询间隔过长，可能会错过队列中的数据。
扩展性差： 当队列数量增加时，轮询的效率会线性下降。

改进方案：

指数退避： 如果连续多次轮询都发现队列为空，可以逐渐增加time.sleep()的时间，以降低CPU占用率。

使用通知队列

另一种方法是使用一个额外的通知队列。当有数据放入某个队列时，向通知队列发送一个消息，指示哪个队列有数据。

Alkaid.art

专门为Phtoshop打造的AIGC绘画插件

153

查看详情

import queue
import time
import threading

c1 = queue.Queue()
c2 = queue.Queue()
notify = queue.Queue()

def worker():
    while True:
        queue_id = notify.get()
        if queue_id == 1:
            i1 = c1.get()
            print("received %s from c1" % i1)
        elif queue_id == 2:
            i2 = c2.get()
            print("received %s from c2" % i2)

# 启动工作线程
t = threading.Thread(target=worker)
t.daemon = True  # 设置为守护线程
t.start()

# 模拟生产者向队列c1和c2放入数据，并向通知队列发送消息
def producer():
    time.sleep(1)
    c1.put("data from c1")
    notify.put(1)

    time.sleep(2)
    c2.put("data from c2")
    notify.put(2)

threading.Thread(target=producer).start()

# 让主线程等待一段时间，以便观察结果
time.sleep(5)

登录后复制

优点：

立即学习“Python免费学习笔记（深入）”；