验证自己对主线程到 hub greenlet 的切换和 greenlets 间的切换

test_mainthread_coroutines.py

#!/usr/bin/env python2.7
#coding: utf-8

import gevent


class TestMainCo(object):
    def __init__(self, mtimeout=2, gtimeout=3, numtasks=10):
        self.mtimeout = mtimeout
        self.gtimeout = gtimeout
        self.numtasks = numtasks

    def run(self):
        tasks = [gevent.spawn(self._task, i) for i in xrange(self.numtasks)]
        gevent.joinall(tasks, timeout=self.mtimeout)
        print('Tasks done!')

    def _task(self, pid):
        print('Task %d begin' % (pid,))
        gevent.sleep(self.gtimeout)
        print('Task %d done' % (pid,))


if __name__ == '__main__':
    test = TestMainCo()
    test.run()