42 lines
817 B
Python
42 lines
817 B
Python
"""
|
|
Uses a trace function to switch greenlets at unexpected times.
|
|
|
|
In the trace function, we switch from the current greenlet to another
|
|
greenlet, which switches
|
|
"""
|
|
import greenlet
|
|
|
|
g1 = None
|
|
g2 = None
|
|
|
|
switch_to_g2 = False
|
|
|
|
def tracefunc(*args):
|
|
print('TRACE', *args)
|
|
global switch_to_g2
|
|
if switch_to_g2:
|
|
switch_to_g2 = False
|
|
g2.switch()
|
|
print('\tLEAVE TRACE', *args)
|
|
|
|
def g1_run():
|
|
print('In g1_run')
|
|
global switch_to_g2
|
|
switch_to_g2 = True
|
|
greenlet.getcurrent().parent.switch()
|
|
print('Return to g1_run')
|
|
print('Falling off end of g1_run')
|
|
|
|
def g2_run():
|
|
g1.switch()
|
|
print('Falling off end of g2')
|
|
|
|
greenlet.settrace(tracefunc)
|
|
|
|
g1 = greenlet.greenlet(g1_run)
|
|
g2 = greenlet.greenlet(g2_run)
|
|
|
|
g1.switch()
|
|
print('Falling off end of main')
|
|
g2.switch()
|