feat(admin): шкала загрузки источников, отладка и загрузка работ в корпус
Заливка корпуса была чёрным ящиком: у источника только last_status (idle/running/done/error), без «сколько из скольки», без причины падения и без способа остановить начатое. Теперь каждый запуск создаёт строку parse_runs, куда воркер раз в ~2с пишет стадию, счётчики и журнал событий. Админка: - шкала загрузки у каждого источника (0→50% выборка, 50→100% индексация), раскрытая строка — журнал прогона по шагам с таймингами; - «Запустить всё» / «Остановить всё» и остановка по одному источнику (кооперативная отмена: воркер останавливается сам, не рвя запись в базу); - пакетное добавление источников (тип + список тем), тип pmc в форме; - загрузка PDF/DOCX/TXT прямо в базу сравнения (index.ingest_upload); - страница «Отладка»: воркеры Celery и их текущие таски, очереди RabbitMQ, покрытие корпуса эмбеддингами, зависшие и упавшие прогоны, конфиг бэкендов. Защита от краш-лупа по consumer_timeout RabbitMQ (docs/DR-HA.md §6), без неё массовый запуск 170+ источников гарантированно ронял воркер: - PARSER_TIME_BUDGET_S (1500с) — прогон закругляется сам и помечается partial; - worker_prefetch_multiplier=1 — таймаут считается от ДОСТАВКИ сообщения, и с дефолтным префетчем очередь долгих run_parser убивала канал на задачах, которые ещё не начинались. Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
This commit is contained in:
@@ -12,7 +12,7 @@ import xml.etree.ElementTree as ET
|
||||
from typing import Any
|
||||
|
||||
import httpx
|
||||
from base import BaseParser
|
||||
from base import BaseParser, ProgressCallback
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
@@ -46,6 +46,7 @@ class ArxivParser(BaseParser):
|
||||
limit: int = 500,
|
||||
categories: list[str] | None = None,
|
||||
year_from: int | None = None,
|
||||
progress_cb: ProgressCallback | None = None,
|
||||
) -> list[dict[str, Any]]:
|
||||
"""
|
||||
Получить препринты из arXiv.
|
||||
@@ -55,6 +56,7 @@ class ArxivParser(BaseParser):
|
||||
limit: Максимальное количество документов
|
||||
categories: Список категорий arXiv (cs.AI, math.ST и т.д.)
|
||||
year_from: Год публикации от
|
||||
progress_cb: см. base.ProgressCallback
|
||||
|
||||
Returns:
|
||||
Список сырых словарей
|
||||
@@ -94,6 +96,10 @@ class ArxivParser(BaseParser):
|
||||
results.extend(entries)
|
||||
start += len(entries)
|
||||
|
||||
if progress_cb and not progress_cb(min(len(results), limit)):
|
||||
logger.info(f"arXiv: выборка остановлена по запросу (получено {len(results)})")
|
||||
break
|
||||
|
||||
if len(entries) < max_results:
|
||||
break
|
||||
|
||||
|
||||
Reference in New Issue
Block a user