Skip to content

iancmcc/greenlight

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Greenlight

greenlight is a powerful decorator that helps you write code using gevent. You can decorate an existing function to run it in a greenlet:

>>> from greenlight import greenlight
>>> from gevent.greenlet import Greenlet
>>> @greenlight
... def async_squaring(n):
...     "An extremely useful function"
...     return n**2
...
>>> g = async_squaring(10)
>>> isinstance(g, Greenlet)
True
>>> g.get()
100

More interestingly, you can chain greenlets inside a greenlit generator (similar to Twisted's inlineCallbacks). This saves boilerplate; one would otherwise have to call gevent.spawn() on each function, then either link() them together, or, perhaps, call join() or get() on each one and pass the result to the next. With greenlight, you can simply yield to get results inline:

>>> @greenlight
... def square_thrice(n):
...     squared = yield async_squaring(n)
...     tothe4th = yield async_squaring(squared)
...     tothe8th = yield async_squaring(tothe4th)
...
>>> square_thrice(2).get()
256

Since a greenlit function always returns a greenlet itself, you can write code very similar to the synchronous code you normally would, while still ensuring execution order, simply by chaining greenlit functions together.

You can explicitly return values from the greenlit generator using the green_return function. If green_return is not used, the result of the last greenlet yielded is returned (as in the previous example):

>>> from greenlight import green_return
>>> @greenlight
... def greater_than_100():
...     a = 2
...     while True:
...         a = yield async_squaring(a)
...         if a>100:
...             green_return(a)
...
>>> greater_than_100().get()
256

Error handling works as you would expect:

>>> @greenlight
... def something_bad():
...     raise Exception('O NOES')
...
>>> @greenlight
... def main():
...     try:
...         hundred = yield async_squaring(10)
...         yield something_bad()
...     except Exception, e:
...         print e
...         green_return(None)
...
>>> main().get()
O NOES

greenlight normally behaves like gevent.spawn, in that it starts greenlets for you. If you don't want that to happen, you can use greenlight_nostart:

>>> from greenlight import greenlight_nostart as greenlight
>>> @greenlight
... def squared(n):
...     return n**2
...
>>> g = squared(4)
>>> g.started
False
>>> g.start(); g.get()
16

Releases

No releases published

Packages

No packages published

Languages