asgi-server-timing-middleware/smoke_test.py at master · sm-Fifteen/asgi-server-timing-middleware · GitHub

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
import fastapi
import pydantic
from fastapi import FastAPI
from asyncio import sleep as async_sleep
from time import perf_counter_ns

from asgi_server_timing import ServerTimingMiddleware

fastapi_app = fastapi.FastAPI()

@fastapi_app.get("/")
async def context_id_endpoint():
	before = perf_counter_ns()
	await async_sleep(0.5)
	after = perf_counter_ns()

	return after - before


track = {
	"1deps": (fastapi.routing.solve_dependencies,),
	"2main": (fastapi.routing.run_endpoint_function,),
	"3valid": (pydantic.fields.ModelField.validate,),
	"4encode": (fastapi.encoders.jsonable_encoder,),
	"5render": (
		fastapi.responses.JSONResponse.render,
		fastapi.responses.ORJSONResponse.render,
	),
}

fastapi_app.add_middleware(ServerTimingMiddleware, calls_to_track=track)

if __name__ == '__main__':
	import uvicorn
	uvicorn.run(fastapi_app, host="0.0.0.0", port=8000)