aiothrift
Asyncio implementation for thrift protocol, which is heavily based on thriftpy.
This project is still in early develop state and thus is not recommended for production usage.
Documentation: https://aiothrift.readthedocs.org/
Installation
$ pip install aiothrift
Usage example
Thrift file
service PingPong { string ping(), i64 add(1:i32 a, 2:i64 b), }
Server
import asyncio
import thriftpy
from aiothrift.server import make_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(
make_server(pingpong_thrift.PingPong, Dispatcher(), ('127.0.0.1', 6000), loop=loop))
try:
loop.run_forever()
except KeyboardInterrupt:
pass
server.close()
loop.run_until_complete(server.wait_closed())
loop.close()
Client
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, ('127.0.0.1', 6000), loop=loop, timeout=2)
print(await conn.ping())
print(await conn.add(5, 6))
conn.close()
loop.run_until_complete(go())
loop.close()
Or use ConnectionPool
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,
('127.0.0.1', 6000), loop=loop, timeout=2)
async with pool.get() as conn:
print(await conn.ping())
print(await conn.add(5, 6))
pool.close()
yield from 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.
Requirements
- Python >= 3.4.2
- async-timeout
- thriftpy
LICENSE
aiothrift
is offered under the MIT license.