Compare commits

...

4 Commits

Author SHA1 Message Date
qananasikq
eba3750f15 speed up cars count query 2026-04-17 23:53:54 +03:00
qananasikq
49de1d6f53 log create_tables errors 2026-04-17 23:53:53 +03:00
qananasikq
7f74793812 add shard page cap 2026-04-17 23:53:53 +03:00
qananasikq
53eb323287 fix batch pool leak 2026-04-17 23:53:10 +03:00
3 changed files with 59 additions and 9 deletions

View File

@@ -26,21 +26,28 @@ def list_cars(
# Список автомобилей с пагинацией и фильтрами
with persistence.session_scope() as session:
query = select(Car)
count_query = select(func.count(Car.id))
if brand:
escaped_brand = brand.replace("%", r"\%").replace("_", r"\_")
query = query.where(Car.brand.ilike(f"%{escaped_brand}%", escape="\\"))
cond = Car.brand.ilike(f"%{escaped_brand}%", escape="\\")
query = query.where(cond)
count_query = count_query.where(cond)
if model:
escaped_model = model.replace("%", r"\%").replace("_", r"\_")
query = query.where(Car.model.ilike(f"%{escaped_model}%", escape="\\"))
cond = Car.model.ilike(f"%{escaped_model}%", escape="\\")
query = query.where(cond)
count_query = count_query.where(cond)
if year_min is not None:
query = query.where(Car.year >= year_min)
count_query = count_query.where(Car.year >= year_min)
if year_max is not None:
query = query.where(Car.year <= year_max)
count_query = count_query.where(Car.year <= year_max)
if is_sold is not None:
query = query.where(Car.is_sold == is_sold)
count_query = count_query.where(Car.is_sold == is_sold)
count_query = select(func.count()).select_from(query.subquery())
total = session.execute(count_query).scalar() or 0
offset = (page - 1) * per_page

View File

@@ -919,6 +919,41 @@ class EncarScraper:
excluded_brands: set[str] | None = None,
runtime_filters: FiltersConfig | None = None,
probe_all_photos: bool = False,
) -> dict[str, Any]:
"""Публичная обёртка: гарантирует закрытие HTTP pool даже при ошибках."""
try:
return self._sync_listing_impl(
limit=limit,
filters=filters,
lane=lane,
only_new=only_new,
batch_size=batch_size,
redis_client=redis_client,
allowed_brands=allowed_brands,
excluded_brands=excluded_brands,
runtime_filters=runtime_filters,
probe_all_photos=probe_all_photos,
)
finally:
if self._batch_pool is not None:
try:
self._batch_pool.close()
except Exception:
logger.debug("Failed to close batch pool", exc_info=True)
self._batch_pool = None
def _sync_listing_impl(
self,
limit: int | None = None,
filters: EncarFilters | None = None,
lane: str = "encar",
only_new: bool = False,
batch_size: int = 1000,
redis_client: Any | None = None,
allowed_brands: set[str] | None = None,
excluded_brands: set[str] | None = None,
runtime_filters: FiltersConfig | None = None,
probe_all_photos: bool = False,
) -> dict[str, Any]:
"""Полная синхронизация листинга Encar.
@@ -988,11 +1023,21 @@ class EncarScraper:
max_consecutive_errors = 5
consecutive_zero_new = 0
max_consecutive_zero_new = 5
# Защита от бесконечной пагинации: Encar API отдаёт максимум
# ~10k записей на query, шарды строятся ≤9500 → не более ~10
# страниц при page_size=1000. Жёсткий cap в 50 — паранойя.
max_pages_per_shard = 50
shard_query = shard_filters.build_query()
shard_collected = 0
while True:
if page >= max_pages_per_shard:
logger.warning(
"Shard %d hit hard page cap (%d), moving to next",
shard_idx, max_pages_per_shard,
)
break
offset = page * page_size
try:
response = self._fetch_listing_page(
@@ -1152,11 +1197,6 @@ class EncarScraper:
self._clear_checkpoint(redis_client)
# Закрываем pool batch API
if self._batch_pool is not None:
self._batch_pool.close()
self._batch_pool = None
logger.info(
"Full sync complete: %d shards, %d collected, %d synced, %d failed, %d marked sold",
len(shards), items_collected, synced, failed, marked_sold,

View File

@@ -46,7 +46,10 @@ class PersistenceService:
try:
Base.metadata.create_all(self.engine)
except Exception:
logger.debug("create_tables skipped (schema already exists)")
# Не глотаем: таблицы могут уже существовать (ок), либо есть проблема
# доступа — пусть вышестоящий код видит её при первой операции, но
# сигнализируем в warning чтобы упростить диагностику.
logger.warning("create_tables failed (continuing — schema may already exist)", exc_info=True)
@contextmanager
def session_scope(self) -> Iterator[Session]: