python - Gevent队列高CPU使用率

Question

为什么这段代码永久使用 95% 的 CPU？Producer 没有发送任何请求，但是 cpu 很热。如何在消费者 greenlet 中不添加显式非零睡眠超时的情况下降低 CPU 使用率？

from gevent import queue
from gevent import sleep
from gevent import Greenlet

def cons(q):
    while True:
        try:
            data = q.get_nowait()
        except queue.Empty:
            sleep(0)
            continue
        print data

def prod(q):
    while True:
        #q.put_nowait('hello')
        sleep(1)


if __name__ == '__main__':

    q = queue.Queue()
    cons1 = Greenlet.spawn(cons, q)
    prod1 = Greenlet.spawn(prod, q)
    cons1.join()
    prod1.join()

Answer 1

在第 8 行，使用 get() 而不是 get_nowait()

data = q.get()

超级简单，现在你的消费者会在等待更多工作时阻塞。

Answer 2

所有推荐的最佳解决方案：在每个队列的单独greenlets中使用queue.get阻塞调用。

代码的下一部分轮询两个 gevent.queue 队列 + 在 greenlet 内的 greenlets 内的 zmq.green 套接字 =) CPU 负载接近于零！

def _zmq_poll(self):
    while True:
        sockets = super(Core, self)._zmq_poll()
        if sockets:
            if sockets.get(self.z_ctr) == zmq.POLLIN:
                self.ctr_process()

def ctr_process(self):
    self.send_log('new CTR: %s')

def req_process(self):
    while True:
        req = self.q_req_core.get()
        self.send_log('new req: %s' % req)
        self.q_core_wrk.put(req)

def wrk_process(self):
    while True:
        wrk_result = self.q_wrk_core.get()
        self.send_log('new result: %s' % wrk_result)
        self.q_core_res.put(wrk_result)
        self.test_wrk_result(wrk_result)

def _run(self):
    self.slaves_start()
    req_process = Greenlet(self.req_process)
    wrk_process = Greenlet(self.wrk_process)
    zmq_poll = Greenlet(self._zmq_poll)
    req_process.start()
    wrk_process.start()
    zmq_poll.start()
    zmq_poll.join()
    req_process.join()
    wrk_process.join()
    self._cleanup()

感谢您的建议！