65 lines
1.7 KiB
Python
65 lines
1.7 KiB
Python
from twisted.internet import defer
|
|
from twisted.python import failure
|
|
from eventlet.support.greenlet import greenlet
|
|
from eventlet import greenlib
|
|
from eventlet.api import get_hub, spawn
|
|
|
|
def block_on(deferred):
|
|
cur = [greenlet.getcurrent()]
|
|
def cb(value):
|
|
if cur:
|
|
greenlib.switch(cur[0], value)
|
|
return value
|
|
def eb(err):
|
|
if cur:
|
|
greenlib.switch(cur[0], exc=(err.type, err.value, err.tb))
|
|
return err
|
|
deferred.addCallback(cb)
|
|
deferred.addErrback(eb)
|
|
try:
|
|
return get_hub().switch()
|
|
finally:
|
|
del cur[0]
|
|
|
|
def _putResultInDeferred(deferred, f, args, kwargs):
|
|
try:
|
|
result = f(*args, **kwargs)
|
|
except:
|
|
f = failure.Failure()
|
|
deferred.errback(f)
|
|
else:
|
|
deferred.callback(result)
|
|
|
|
def deferToGreenThread(func, *args, **kwargs):
|
|
d = defer.Deferred()
|
|
spawn(_putResultInDeferred, d, func, args, kwargs)
|
|
return d
|
|
|
|
def callInGreenThread(func, *args, **kwargs):
|
|
return spawn(func, *args, **kwargs)
|
|
|
|
|
|
if __name__=='__main__':
|
|
import sys
|
|
num = int(sys.argv[1])
|
|
from twisted.internet import reactor
|
|
def test():
|
|
print block_on(reactor.resolver.getHostByName('www.google.com'))
|
|
print block_on(reactor.resolver.getHostByName('###'))
|
|
if num==0:
|
|
test()
|
|
elif num==1:
|
|
spawn(test)
|
|
from eventlet.api import sleep
|
|
sleep(5)
|
|
elif num==2:
|
|
from eventlet.twistedutil import join_reactor
|
|
spawn(test)
|
|
reactor.run()
|
|
elif num==3:
|
|
from eventlet.twistedutil import join_reactor
|
|
print "fails because it's impossible to use block_on from the mainloop"
|
|
reactor.callLater(0, test)
|
|
reactor.run()
|
|
|