Недавно делал быстрый прототип асинхронного приложения в котором требовалось вызывать много синхронного кода. Да, я знаю, что это не лучший дизайн, но нужно было быстрое решение на один процесс и без очередей. Поэтому я выполнял код в потоках.
Выглядело это примерно так:
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
#Murmuration#PKM 这个有意思,[《中国的地图每张都是错的》](https://telegra.ph/%E6%AF%8F%E5%BC%A0%E4%B8%AD%E5%9B%BD%E5%9C%B0%E5%9B%BE%E9%83%BD%E6%98%AF%E9%94%99%E7%9A%84%E8%BF%98%E6%98%AF%E6%95%85%E6%84%8F%E7%9A%84-05-15),这是翻译版本。原版在这里 [《Every map of China is wrong》](https://medium.com/@anastasia.bizyayeva/every-map-of-china-is-wrong-bc2bce145db2) 。🤷我说怎么地图啥都飘呢,哈哈哈