github/datasette
1
0
Fork 0
mirror of https://github.com/simonw/datasette.git synced 2025-12-10 16:51:24 +01:00
Code Issues Projects Releases Packages Wiki Activity

tracer.trace_child_tasks() for asyncio.gather tracing

Browse source 
Also added documentation for datasette.tracer module.

Closes #1576
This commit is contained in:
Simon Willison 2022-02-04 21:19:49 -08:00
commit da53e0360d
4 changed files with 111 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

12
tests/plugins/my_plugin.py
View file

@ -1,5 +1,7 @@
import asyncio
from datasette import hookimpl
from datasette.facets import Facet
from datasette import tracer
from datasette.utils import path_with_added_args
from datasette.utils.asgi import asgi_send_json, Response
import base64
@ -270,6 +272,15 @@ def register_routes():
def asgi_scope(scope):
return Response.json(scope, default=repr)
async def parallel_queries(datasette):
db = datasette.get_database()
with tracer.trace_child_tasks():
one, two = await asyncio.gather(
db.execute("select coalesce(sleep(0.1), 1)"),
db.execute("select coalesce(sleep(0.1), 2)"),
)
return Response.json({"one": one.single_value(), "two": two.single_value()})
return [
(r"/one/$", one),
(r"/two/(?P<name>.*)$", two),
@ -281,6 +292,7 @@ def register_routes():
(r"/add-message/$", add_message),
(r"/render-message/$", render_message),
(r"/asgi-scope$", asgi_scope),
(r"/parallel-queries$", parallel_queries),
]