|
@@ -1,24 +1,14 @@
|
|
|
import asyncio
|
|
|
import asyncpg
|
|
|
import os
|
|
|
-import jinja2
|
|
|
-from fastapi import FastAPI
|
|
|
+from fastapi import FastAPI, Request
|
|
|
+from fastapi.templating import Jinja2Templates
|
|
|
from starlette.responses import HTMLResponse, JSONResponse, PlainTextResponse
|
|
|
from random import randint, sample
|
|
|
-from operator import itemgetter
|
|
|
-from urllib.parse import parse_qs
|
|
|
|
|
|
-
|
|
|
-READ_ROW_SQL = 'SELECT "randomnumber", "id" FROM "world" WHERE id = $1'
|
|
|
+READ_ROW_SQL = 'SELECT "id", "randomnumber" FROM "world" WHERE id = $1'
|
|
|
WRITE_ROW_SQL = 'UPDATE "world" SET "randomnumber"=$1 WHERE id=$2'
|
|
|
-ADDITIONAL_ROW = [0, 'Additional fortune added at request time.']
|
|
|
-
|
|
|
-
|
|
|
-def load_fortunes_template():
|
|
|
- path = os.path.join('templates', 'fortune.html')
|
|
|
- with open(path, 'r') as template_file:
|
|
|
- template_text = template_file.read()
|
|
|
- return jinja2.Template(template_text)
|
|
|
+ADDITIONAL_ROW = [0, "Additional fortune added at request time."]
|
|
|
|
|
|
|
|
|
def get_num_queries(queries):
|
|
@@ -35,40 +25,39 @@ def get_num_queries(queries):
|
|
|
|
|
|
|
|
|
connection_pool = None
|
|
|
-sort_fortunes_key = itemgetter(1)
|
|
|
-template = load_fortunes_template()
|
|
|
|
|
|
app = FastAPI()
|
|
|
|
|
|
+templates = Jinja2Templates(directory="templates")
|
|
|
+
|
|
|
|
|
|
@app.on_event("startup")
|
|
|
async def setup_database():
|
|
|
global connection_pool
|
|
|
connection_pool = await asyncpg.create_pool(
|
|
|
- user=os.getenv('PGUSER', 'benchmarkdbuser'),
|
|
|
- password=os.getenv('PGPASS', 'benchmarkdbpass'),
|
|
|
- database='hello_world',
|
|
|
- host='tfb-database',
|
|
|
- port=5432
|
|
|
+ user=os.getenv("PGUSER", "benchmarkdbuser"),
|
|
|
+ password=os.getenv("PGPASS", "benchmarkdbpass"),
|
|
|
+ database="hello_world",
|
|
|
+ host="tfb-database",
|
|
|
+ port=5432,
|
|
|
)
|
|
|
|
|
|
|
|
|
[email protected]('/json')
|
|
|
[email protected]("/json")
|
|
|
async def json_serialization():
|
|
|
- return JSONResponse({'message': 'Hello, world!'})
|
|
|
+ return JSONResponse({"message": "Hello, world!"})
|
|
|
|
|
|
|
|
|
[email protected]('/db')
|
|
|
[email protected]("/db")
|
|
|
async def single_database_query():
|
|
|
row_id = randint(1, 10000)
|
|
|
-
|
|
|
async with connection_pool.acquire() as connection:
|
|
|
number = await connection.fetchval(READ_ROW_SQL, row_id)
|
|
|
|
|
|
- return JSONResponse({'id': row_id, 'randomNumber': number})
|
|
|
+ return JSONResponse({"id": row_id, "randomNumber": number})
|
|
|
|
|
|
|
|
|
[email protected]('/queries')
|
|
|
[email protected]("/queries")
|
|
|
async def multiple_database_queries(queries = None):
|
|
|
num_queries = get_num_queries(queries)
|
|
|
row_ids = sample(range(1, 10000), num_queries)
|
|
@@ -78,27 +67,26 @@ async def multiple_database_queries(queries = None):
|
|
|
statement = await connection.prepare(READ_ROW_SQL)
|
|
|
for row_id in row_ids:
|
|
|
number = await statement.fetchval(row_id)
|
|
|
- worlds.append({'id': row_id, 'randomNumber': number})
|
|
|
+ worlds.append({"id": row_id, "randomNumber": number})
|
|
|
|
|
|
return JSONResponse(worlds)
|
|
|
|
|
|
|
|
|
[email protected]('/fortunes')
|
|
|
-async def fortunes():
|
|
|
[email protected]("/fortunes")
|
|
|
+async def fortunes(request: Request):
|
|
|
async with connection_pool.acquire() as connection:
|
|
|
- fortunes = await connection.fetch('SELECT * FROM Fortune')
|
|
|
+ fortunes = await connection.fetch("SELECT * FROM Fortune")
|
|
|
|
|
|
fortunes.append(ADDITIONAL_ROW)
|
|
|
- fortunes.sort(key=sort_fortunes_key)
|
|
|
- content = template.render(fortunes=fortunes)
|
|
|
- return HTMLResponse(content)
|
|
|
+ fortunes.sort(key=lambda row: row[1])
|
|
|
+ return templates.TemplateResponse("fortune.html", {"fortunes": fortunes, "request": request})
|
|
|
|
|
|
|
|
|
[email protected]('/updates')
|
|
|
[email protected]("/updates")
|
|
|
async def database_updates(queries = None):
|
|
|
num_queries = get_num_queries(queries)
|
|
|
updates = [(row_id, randint(1, 10000)) for row_id in sample(range(1, 10000), num_queries)]
|
|
|
- worlds = [{'id': row_id, 'randomNumber': number} for row_id, number in updates]
|
|
|
+ worlds = [{"id": row_id, "randomNumber": number} for row_id, number in updates]
|
|
|
|
|
|
async with connection_pool.acquire() as connection:
|
|
|
statement = await connection.prepare(READ_ROW_SQL)
|
|
@@ -109,6 +97,6 @@ async def database_updates(queries = None):
|
|
|
return JSONResponse(worlds)
|
|
|
|
|
|
|
|
|
[email protected]('/plaintext')
|
|
|
[email protected]("/plaintext")
|
|
|
async def plaintext():
|
|
|
- return PlainTextResponse(b'Hello, world!')
|
|
|
+ return PlainTextResponse(b"Hello, world!")
|