Count current greenlet's CPU cost

time-so-far.py

import gevent

# GreenletProfiler 0.1 from https://pypi.python.org/pypi/GreenletProfiler
import GreenletProfiler

MILLION = 1000 * 1000


def print_greenlet_cpu_time():
    # Find self in list of "thread stats", actually greenlet stats.
    greenlet_id = id(gevent.getcurrent())
    for stat in GreenletProfiler.get_thread_stats():
        if stat.id == greenlet_id:
            print stat.sched_count, stat.ttot
            break


def foo():
    for i in range(20 * MILLION):
        if not i % (2 * MILLION):
            gevent.sleep(0)

    print 'foo done'


def bar():
    for i in range(10 * MILLION):
        if not i % MILLION:
            gevent.sleep(0)

            print_greenlet_cpu_time()

    print 'bar done'


GreenletProfiler.set_clock_type('cpu')
GreenletProfiler.start()
foo_greenlet = gevent.spawn(foo)
bar_greenlet = gevent.spawn(bar)
foo_greenlet.join()
bar_greenlet.join()
GreenletProfiler.stop()