Asyncio implementation for thrift protocol, which is heavily based on thriftpy.
Documentation: https://aiothrift.readthedocs.org/
$ pip install aiothrift
service PingPong { string ping(), i64 add(1:i32 a, 2:i64 b), }
import asyncio
import thriftpy
from aiothrift.server import create_server
pingpong_thrift = thriftpy.load('pingpong.thrift', module_name='pingpong_thrift')
class Dispatcher:
def ping(self):
return "pong"
async def add(self, a, b):
await asyncio.sleep(1)
return a + b
loop = asyncio.get_event_loop()
server = loop.run_until_complete(
create_server(pingpong_thrift.PingPong, Dispatcher(), loop=loop))
try:
loop.run_forever()
except KeyboardInterrupt:
pass
server.close()
loop.run_until_complete(server.wait_closed())
loop.close()
import thriftpy
import asyncio
import aiothrift
loop = asyncio.get_event_loop()
pingpong_thrift = thriftpy.load('pingpong.thrift', module_name='pingpong_thrift')
async def go():
conn = await aiothrift.create_connection(pingpong_thrift.PingPong, loop=loop)
print(await conn.ping())
print(await conn.add(5, 6))
conn.close()
loop.run_until_complete(go())
loop.close()
import thriftpy
import asyncio
import aiothrift
loop = asyncio.get_event_loop()
pingpong_thrift = thriftpy.load('pingpong.thrift', module_name='pingpong_thrift')
async def go():
pool = await aiothrift.create_pool(pingpong_thrift.PingPong, loop=loop)
async with pool.get() as conn:
print(await conn.ping())
print(await conn.add(5, 6))
pool.close()
await pool.wait_closed()
loop.run_until_complete(go())
loop.close()
It's just that simple to begin with aiothrift
, and you are not forced to use aiothrift
on both server and client side.
So if you already have a normal thrift server setup, feel free to create an async thrift client to communicate with that server.
- Python >= 3.4.2
- async-timeout
- thriftpy
aiothrift
is offered under the MIT license.