Skip to content

Commit 506e834

Browse files
committed
Added set_default_thread_limiter
1 parent 0b4a2f5 commit 506e834

1 file changed

Lines changed: 6 additions & 4 deletions

File tree

debug_toolbar/middleware.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,11 @@
1-
import asyncio
21
import functools
32
import json
43
import re
54
import typing as t
6-
from concurrent.futures import ThreadPoolExecutor
75
from urllib import parse
86

7+
from anyio import CapacityLimiter
8+
from anyio.lowlevel import RunVar
99
from fastapi import APIRouter, HTTPException, Request, Response, status
1010
from fastapi.staticfiles import StaticFiles
1111
from starlette.middleware.base import BaseHTTPMiddleware, RequestResponseEndpoint
@@ -50,8 +50,10 @@ def init_toolbar(self) -> None:
5050
StaticFiles(packages=[__package__]),
5151
name="debug_toolbar.static",
5252
)
53-
loop = asyncio.get_event_loop()
54-
loop.set_default_executor(ThreadPoolExecutor(1))
53+
54+
@self.router.on_event("startup")
55+
def set_default_thread_limiter():
56+
RunVar("_default_thread_limiter").set(CapacityLimiter(1))
5557

5658
async def dispatch(
5759
self,

0 commit comments

Comments
 (0)