56 lines
2.2 KiB
Python
56 lines
2.2 KiB
Python
"""NAS realestate 내부 계약 — X-Internal-Key + retry."""
|
|
from __future__ import annotations
|
|
|
|
import asyncio
|
|
import logging
|
|
|
|
import httpx
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
_RETRY_STATUSES = {429, 500, 502, 503, 504}
|
|
_TARGETS_TIMEOUT = 15.0
|
|
_INGEST_TIMEOUT = 180.0 # 첫 실행 bulk upsert/매칭/알림이 느림 (초 단위 수십)
|
|
|
|
|
|
class NASClient:
|
|
def __init__(self, base_url: str, api_key: str):
|
|
self._base_url = base_url.rstrip("/")
|
|
self._api_key = api_key
|
|
self._client = httpx.AsyncClient()
|
|
|
|
async def close(self) -> None:
|
|
await self._client.aclose()
|
|
|
|
async def get_targets(self) -> dict:
|
|
return await self._request("GET", "/api/internal/realestate/targets",
|
|
timeout=_TARGETS_TIMEOUT, max_attempts=3)
|
|
|
|
async def post_ingest(self, fetched_at: str, batches: list[dict]) -> dict:
|
|
# ingest는 재시도 안 함(max_attempts=1) — 타임아웃 후 재전송 시
|
|
# NAS가 같은 batch를 재처리/재알림할 위험. 실패 시 다음 사이클에 재시도.
|
|
return await self._request(
|
|
"POST", "/api/internal/realestate/listings-ingest",
|
|
json={"fetched_at": fetched_at, "batches": batches},
|
|
timeout=_INGEST_TIMEOUT, max_attempts=1)
|
|
|
|
async def _request(self, method: str, path: str, *, timeout: float,
|
|
max_attempts: int, **kwargs) -> dict:
|
|
url = f"{self._base_url}{path}"
|
|
headers = {"X-Internal-Key": self._api_key}
|
|
for attempt in range(max_attempts):
|
|
try:
|
|
resp = await self._client.request(method, url, headers=headers,
|
|
timeout=timeout, **kwargs)
|
|
if resp.status_code in _RETRY_STATUSES and attempt < max_attempts - 1:
|
|
await asyncio.sleep(2 ** attempt)
|
|
continue
|
|
resp.raise_for_status()
|
|
return resp.json()
|
|
except httpx.TimeoutException:
|
|
if attempt < max_attempts - 1:
|
|
await asyncio.sleep(2 ** attempt)
|
|
continue
|
|
raise
|
|
raise RuntimeError("retry exhausted")
|