forked from samuelcolvin/aiohttp-vs-sanic-vs-japronto
/
run_aiohttp.py
44 lines (30 loc) · 1.04 KB
/
run_aiohttp.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
import asyncio
import random
import ujson
from aiohttp import web
from shared import setup_db
async def startup(app):
app['db'] = await setup_db(app.loop)
async def cleanup(app):
await app['db'].close()
async def index(request):
data = {'message': 'hello world'}
body = ujson.dumps(data)
return web.Response(body=body.encode(), content_type='application/json')
async def db(request):
user_id = random.randrange(1, 1000)
async with request.app['db'].acquire() as conn:
username = await conn.fetchval('SELECT name from users WHERE id=$1', user_id)
data = {'message': f'hello {username}'}
body = ujson.dumps(data)
return web.Response(body=body.encode(), content_type='application/json')
# import tokio
# asyncio.set_event_loop_policy(tokio.TokioLoopPolicy())
import uvloop
asyncio.set_event_loop_policy(uvloop.EventLoopPolicy())
app = web.Application()
app.on_startup.append(startup)
app.on_cleanup.append(cleanup)
app.router.add_get('/', index)
app.router.add_get('/db', db)
web.run_app(app, port=8000)