examples/bench_spawn.py: add geventraw and geventpool options

examples/bench_spawn.py

@@ -7,9 +7,10 @@ N = 20000
 counter = 0
 
 
-def incr(**kwargs):
+def incr(sleep, **kwargs):
     global counter
     counter += 1
+    sleep(0)
 
 if not sys.argv[1:]:
     sys.exit('USAGE: python bench_spawn.py none|gevent|eventlet [eventlet_hub]')
@@ -21,10 +22,26 @@ if sys.argv[-1]=='kwargs':
     kwargs = {'foo': 1, 'bar': 'hello'}
     del sys.argv[-1]
 
+def noop(p):
+    pass
+
+def test(spawn, sleep):
+    start = time()
+    for _ in xrange(N):
+        spawn(incr, sleep, **kwargs)
+    delta = time() - start
+    print 'spawning: %.1f microseconds per greenlet' % (delta*1000000.0/N)
+    assert counter == 0, counter
+    start = time()
+    sleep(0)
+    delta = time() - start
+    assert counter == N, (counter, N)
+    print 'sleep(0): %.1f microseconds per greenlet' % (delta*1000000.0/N)
+
 if sys.argv[1]=='none':
     start = time()
     for _ in xrange(N):
-        incr(**kwargs)
+        incr(noop, **kwargs)
     delta = time() - start
     assert counter == N, (counter, N)
     print '%.2f microseconds' % (delta*1000000.0/N)
@@ -32,49 +49,34 @@ elif sys.argv[1]=='gevent':
     import gevent
     print gevent.__file__
     from gevent import spawn, sleep
+    test(spawn, sleep)
+elif sys.argv[1]=='geventraw':
+    import gevent
+    print gevent.__file__
+    from gevent import sleep
+    from gevent.rawgreenlet import spawn
+    test(spawn, sleep)
+elif sys.argv[1]=='geventpool':
+    import gevent
+    print gevent.__file__
+    from gevent import sleep
+    from gevent.pool import Pool
+    p = Pool()
+    test(p.spawn, sleep)
     start = time()
-    for _ in xrange(N):
-        spawn(incr, **kwargs)
-    delta = time() - start
-    print 'spawning: %d microseconds' % (delta*1000000.0/N)
-    assert counter == 0, counter
-    start = time()
-    sleep(0)
+    p.join()
     delta = time() - start
-    assert counter == N, (counter, N)
-    print 'switching: %d microseconds' % (delta*1000000.0/N)
+    print 'joining: %.1f microseconds per greenlet' % (delta*1000000.0/N)
 elif sys.argv[1]=='eventlet':
     import eventlet
     print eventlet.__file__
     from eventlet.api import spawn, sleep, use_hub
     if sys.argv[2:]:
         use_hub(sys.argv[2])
-    start = time()
-    for _ in xrange(N):
-        spawn(incr, **kwargs)
-    delta = time() - start
-    print 'spawning: %d microseconds' % (delta*1000000.0/N)
-    assert counter == 0, counter
-    start = time()
-    sleep(0)
-    delta = time() - start
-    assert counter == N, (counter, N)
-    print 'switching: %d microseconds' % (delta*1000000.0/N)
-elif sys.argv[1]=='eventletproc':
+    test(spawn, sleep)
+elif sys.argv[1]=='eventlet1':
     from eventlet.proc import spawn_greenlet as spawn
     from eventlet.api import sleep, use_hub
     if sys.argv[2:]:
         use_hub(sys.argv[2])
-    start = time()
-    for _ in xrange(N):
-        spawn(incr, **kwargs)
-    delta = time() - start
-    print 'spawning: %d microseconds' % (delta*1000000.0/N)
-    assert counter == 0, counter
-    start = time()
-    sleep(0)
-    delta = time() - start
-    assert counter == N, (counter, N)
-    print 'switching: %d microseconds' % (delta*1000000.0/N)
-
-
+    test(spawn, sleep, sleep)