Недавно делал быстрый прототип асинхронного приложения в котором требовалось вызывать много синхронного кода. Да, я знаю, что это не лучший дизайн, но нужно было быстрое решение на один процесс и без очередей. Поэтому я выполнял код в потоках.
Выглядело это примерно так:
from fastapi.concurrency import run_in_threadpool
async def execute(data: DataRequest) -> DataResponse:
try:
result = await run_in_threadpool(sync_function, data)
return DataResponse(data=result)
except Exception as e:
return DataResponse(
error=str(e),
success=False,
)
В общем работает нормально. Для всех вызовов под капотом используется общий тредпул, всё работает предсказуемо.
Но потребовалось изменить количество запускаемых в пуле потоков (по умолчанию создается 40 воркеров).
Так как дело происходит с FastAPI, делается это через lifespan используя настройки anyio:
import anyio
@asynccontextmanager
async def lifespan(app: FastAPI):
limiter = anyio.to_thread.current_default_thread_limiter()
limiter.total_tokens = 100
yield
# если вдруг нужно вернуть обратно
limiter.total_tokens = 40
Зачем менять количество воркеров?
- уменьшить, если оперативки мало (один тред занимает ~8мб)
- увеличить чтобы выдержать нагрузку
Если есть предложения получше при тех же вводных - предлагайте😉
#async
被 Covid 偷走的五年
我依旧觉得 2019 年是去年，要不是写博客梳理我都不知道我干了这么多事。我说什么来着，博客给记忆增加了一个全新的维度，是最好的时间胶囊。
下一个 5 年就是 2030 了，we're literally already closer to 2050 than 2000, can you even imagine?
本文是我 2024 年 8 月 Patreon 的月度选题。本月两个命题平票，由至尊糖妈的双倍票数 tie breaker 所以选了这个话题。加入 patreon 可以跟其它赞助者一起选出下个月你想看的博客文章，并且支持本博客的持续创作。下期选题已经发布欢迎金主点击投票：
- My pet peeves
- Mid 30s vs Mid 20s
- What's your fun fact
#blog#patreon