Merge pull request #558 from Zheaoli/master
Add a new example by using aiomysql
This commit is contained in:
commit
fa69892f70
62
examples/sanic_aiomysql_with_global_pool.py
Normal file
62
examples/sanic_aiomysql_with_global_pool.py
Normal file
|
@ -0,0 +1,62 @@
|
||||||
|
# encoding: utf-8
|
||||||
|
"""
|
||||||
|
You need the aiomysql
|
||||||
|
"""
|
||||||
|
import os
|
||||||
|
|
||||||
|
import aiomysql
|
||||||
|
|
||||||
|
from sanic import Sanic
|
||||||
|
from sanic.response import json
|
||||||
|
|
||||||
|
database_name = os.environ['DATABASE_NAME']
|
||||||
|
database_host = os.environ['DATABASE_HOST']
|
||||||
|
database_user = os.environ['DATABASE_USER']
|
||||||
|
database_password = os.environ['DATABASE_PASSWORD']
|
||||||
|
app = Sanic()
|
||||||
|
|
||||||
|
|
||||||
|
@app.listener("before_server_start")
|
||||||
|
async def get_pool(app, loop):
|
||||||
|
"""
|
||||||
|
the first param is the global instance ,
|
||||||
|
so we can store our connection pool in it .
|
||||||
|
and it can be used by different request
|
||||||
|
:param args:
|
||||||
|
:param kwargs:
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
app.pool = {
|
||||||
|
"aiomysql": await aiomysql.create_pool(host=database_host, user=database_user, password=database_password,
|
||||||
|
db=database_name,
|
||||||
|
maxsize=5)}
|
||||||
|
async with app.pool['aiomysql'].acquire() as conn:
|
||||||
|
async with conn.cursor() as cur:
|
||||||
|
await cur.execute('DROP TABLE IF EXISTS sanic_polls')
|
||||||
|
await cur.execute("""CREATE TABLE sanic_polls (
|
||||||
|
id serial primary key,
|
||||||
|
question varchar(50),
|
||||||
|
pub_date timestamp
|
||||||
|
);""")
|
||||||
|
for i in range(0, 100):
|
||||||
|
await cur.execute("""INSERT INTO sanic_polls
|
||||||
|
(id, question, pub_date) VALUES ({}, {}, now())
|
||||||
|
""".format(i, i))
|
||||||
|
|
||||||
|
|
||||||
|
@app.route("/")
|
||||||
|
async def test():
|
||||||
|
result = []
|
||||||
|
data = {}
|
||||||
|
async with app.pool['aiomysql'].acquire() as conn:
|
||||||
|
async with conn.cursor() as cur:
|
||||||
|
await cur.execute("SELECT question, pub_date FROM sanic_polls")
|
||||||
|
async for row in cur:
|
||||||
|
result.append({"question": row[0], "pub_date": row[1]})
|
||||||
|
if result or len(result) > 0:
|
||||||
|
data['data'] = res
|
||||||
|
return json(data)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
app.run(host="127.0.0.1", workers=4, port=12000)
|
Loading…
Reference in New Issue
Block a user