Update fastapi orjson (replace #5594) (#5628)

* :sparkles: Add FastAPI with ORJSON

* update orjson

* update requirements of default fastapi

* fix typo

Co-authored-by: Sebastián Ramírez <[email protected]>

frameworks/Python/fastapi/app_orjson.py

@@ -0,0 +1,113 @@
+import asyncio
+import asyncpg
+import os
+import jinja2
+from fastapi import FastAPI
+from fastapi.responses import HTMLResponse, ORJSONResponse, PlainTextResponse
+from random import randint
+from operator import itemgetter
+from functools import partial
+
+_randint = partial(randint, 1, 10000)
+
+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."]
+
+
+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,
+    )
+
+
+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)
+
+
+def get_num_queries(queries):
+    try:
+        query_count = int(queries)
+    except (ValueError, TypeError):
+        return 1
+
+    if query_count < 1:
+        return 1
+    if query_count > 500:
+        return 500
+    return query_count
+
+
+connection_pool = None
+sort_fortunes_key = itemgetter(1)
+template = load_fortunes_template()
+loop = asyncio.get_event_loop()
+loop.run_until_complete(setup_database())
+
+
+app = FastAPI()
+
+
[email protected]("/json")
+async def json_serialization():
+    return ORJSONResponse({"message": "Hello, world!"})
+
+
[email protected]("/db")
+async def single_database_query():
+    async with connection_pool.acquire() as connection:
+        record = await connection.fetchrow(READ_ROW_SQL, _randint())
+
+    return ORJSONResponse({"id": record['id'], "randomNumber": record['randomnumber']})
+
+
[email protected]("/queries")
+async def multiple_database_queries(queries=None):
+    num_queries = get_num_queries(queries)
+    worlds = tuple(map(lambda _: {"id": _randint(), "randomNumber": None}, range(num_queries)))
+
+    async with connection_pool.acquire() as connection:
+        statement = await connection.prepare(READ_ROW_SQL)
+        for world in worlds:
+            world["randomNumber"] = await statement.fetchval(world["id"])
+
+    return ORJSONResponse(worlds)
+
+
[email protected]("/fortunes")
+async def fortunes():
+    async with connection_pool.acquire() as connection:
+        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)
+
+
[email protected]("/updates")
+async def database_updates(queries=None):
+    num_queries = get_num_queries(queries)
+    updates = [(_randint(), _randint()) for _ in range(num_queries)]
+    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)
+        for row_id, number in updates:
+            await statement.fetchval(row_id)
+        await connection.executemany(WRITE_ROW_SQL, updates)
+
+    return ORJSONResponse(worlds)
+
+
[email protected]("/plaintext")
+async def plaintext():
+    return PlainTextResponse(b"Hello, world!")

frameworks/Python/fastapi/benchmark_config.json

@@ -24,6 +24,29 @@
         "display_name": "FastAPI",
         "notes": "",
         "versus": "None"
+      },
+      "orjson": {
+        "json_url": "/json",
+        "fortune_url": "/fortunes",
+        "plaintext_url": "/plaintext",
+        "db_url": "/db",
+        "query_url": "/queries?queries=",
+        "update_url": "/updates?queries=",
+        "port": 8080,
+        "approach": "Realistic",
+        "classification": "Micro",
+        "database": "Postgres",
+        "framework": "FastAPI",
+        "language": "Python",
+        "flavor": "Python3",
+        "orm": "Raw",
+        "platform": "None",
+        "webserver": "None",
+        "os": "Linux",
+        "database_os": "Linux",
+        "display_name": "FastAPI",
+        "notes": "",
+        "versus": "None"
       }
     }
   ]

frameworks/Python/fastapi/fastapi-orjson.dockerfile

@@ -0,0 +1,9 @@
+FROM python:3.8
+
+ADD ./ /fastapi
+
+WORKDIR /fastapi
+
+RUN pip3 install -r /fastapi/requirements-orjson.txt
+
+CMD gunicorn app_orjson:app -k uvicorn.workers.UvicornWorker -c fastapi_conf.py

frameworks/Python/fastapi/requirements-orjson.txt

@@ -0,0 +1,6 @@
+asyncpg==0.20.1
+gunicorn==20.0.4
+Jinja2==2.11.1
+fastapi==0.54.1
+orjson==2.6.5
+uvicorn==0.11.3