Gevent队列高CPU使用率

时间:2013-05-19 22:29:04

标签: python queue gevent

为什么此代码永久使用95%的CPU?制作人不发送任何请求,但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()

2 个答案:

答案 0 :(得分:1)

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

data = q.get()

超级简单,现在您的消费者会在等待更多工作时阻止。

答案 1 :(得分:0)

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

代码的下一部分在greenlet内部的greenlet中轮询两个gevent.queue队列+ 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()

感谢您的建议!

相关问题