Play with gevent.pool.Pool

test_pool.py

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

import gevent
import gevent.pool


class TestPool(object):

    def __init__(self, maxsize=10):
        self.pool = gevent.pool.Pool(maxsize)

    def run(self):
        for i in xrange(8):
            gworker = gevent.spawn(self._worker, i)
            self.pool.add(gworker)
            print('sizeof pool is %d' % (len(self.pool),))
            print('worker %d in? %r' % (i, gworker in self.pool))

        # add this statement to trap into the loop
        self.pool.join()

        print('All tasks done.')

    def _worker(self, pid):
        print('My pid is %d' % (pid,))
        gevent.sleep(0)
        print('%d has done.' % (pid,))


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