feat(lps): 워커 루프 + LISTEN/NOTIFY — 큐 소비 파이프라인 가동

큐를 실제로 돌린다: 적재 → 워커 claim → 핸들러 실행 → 결과 저장 → DONE.
API(적재)와 워커(소비)를 분리 프로세스로(코드베이스 공유, 독립 스케일).

- worker/notify: 전용 asyncpg LISTEN 리스너. enqueue 에서 pg_notify → 유휴 워커 즉시 기상(폴링 제거)
- worker/runner: Worker(claim→처리, 처리중 heartbeat 로 lease 갱신, complete/fail) + run_reaper
- worker/handlers: job_type 별 핸들러(주입식). SEARCH=소스 어댑터 검색→정규화 결과
- worker_main: API 분리 워커 진입점(브라우저 무거워 기본 동시성 1)
- job_crud.enqueue: 삽입 시 pg_notify (중복 스킵 시엔 미발생)
- tests: drain→DONE·실패→재시도→dead·reaper 회수 후 재처리·NOTIFY 기상 4건 (전체 20/20)
- 라이브 E2E 확인: 적재→워커가 실제 쿠팡 검색(8건)→DONE

Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
This commit is contained in:
민헌 2026-07-08 16:58:58 +09:00
parent bebb6a71e6
commit abcb58ba05
6 changed files with 294 additions and 0 deletions

View File

@ -16,6 +16,10 @@ from common.database.db_session_manager import DB_SESSION_MNG
from common.enums import DBType, DBWRType, JobStatus
# 잡 적재 시 워커를 즉시 깨우는 LISTEN/NOTIFY 채널(폴링 제거).
JOB_NOTIFY_CHANNEL = "lps_job"
def compute_backoff(attempts: int, base: float = 5.0, cap: float = 600.0) -> float:
"""지수 백오프(초). attempts 회 시도 후 다음 재시도까지 대기 = base * 2^(attempts-1), cap 상한."""
return min(cap, base * (2 ** max(0, attempts - 1)))
@ -53,6 +57,9 @@ class JobQueue:
"t": job_type, "p": priority, "payload": json.dumps(payload),
"dk": dedupe_key, "ma": max_attempts,
})).first()
if row:
# 커밋 시 전달됨 → LISTEN 중인 유휴 워커를 즉시 깨운다(중복 스킵 시엔 알림 안 함).
await s.execute(text("SELECT pg_notify(:ch, '')"), {"ch": JOB_NOTIFY_CHANNEL})
return str(row[0]) if row else None
return await self._tx(run)

69
lps/tests/test_worker.py Normal file
View File

@ -0,0 +1,69 @@
"""워커 루프 테스트 — drain→DONE, 실패→재시도→dead, reaper 회수 후 재처리, NOTIFY 깨움.
핸들러는 fake(브라우저 없이) — 워커 로직만 결정론적으로 검증한다."""
import asyncio
import pytest_asyncio
from sqlalchemy import text
from common.enums import JobStatus, JobType
from crud.job_crud import JobQueue
from worker.notify import JobListener
from worker.runner import Worker
@pytest_asyncio.fixture
async def q(db_engine):
async with db_engine.begin() as conn:
await conn.execute(text("TRUNCATE job"))
return JobQueue()
async def test_worker_drains_all_to_done(q):
for i in range(3):
await q.enqueue(JobType.SEARCH.value, {"product_name": f"item{i}"})
async def handler(job):
return {"ok": True, "q": job["payload"]["product_name"]}
processed = await Worker("w1", q, handler).drain()
assert processed == 3
assert (await q.counts())["DONE"] == 3
async def test_worker_failure_retries_then_dead(q):
await q.enqueue(JobType.SEARCH.value, {"product_name": "x"}, max_attempts=2)
async def boom(job):
raise RuntimeError("nope")
w = Worker("w1", q, boom, backoff_fn=lambda a: 0) # 백오프 0 → 즉시 재시도 가능
assert await w.process_one() is True # 1/2 실패 → PENDING
assert (await q.counts())["PENDING"] == 1
assert await w.process_one() is True # 2/2 실패 → DEAD
counts = await q.counts()
assert counts["DEAD"] == 1 and counts["PENDING"] == 0
assert await w.process_one() is False # DEAD 는 claim 대상 아님
async def test_reaper_reclaims_then_worker_reprocesses(q):
jid = await q.enqueue(JobType.SEARCH.value, {"product_name": "x"})
await q.claim("dead-worker", lease_sec=1) # 점유 후 사망 흉내
await asyncio.sleep(1.3)
assert jid in await q.reap() # 회수 → PENDING
async def handler(job):
return {"ok": True}
assert await Worker("w2", q, handler).process_one() is True
assert (await q.counts())["DONE"] == 1
async def test_enqueue_notifies_listener(q):
listener = JobListener()
await listener.start()
try:
await q.enqueue(JobType.SEARCH.value, {"product_name": "x"}) # pg_notify 발생
assert await listener.wait(3.0) is True # 즉시 깨어남
finally:
await listener.close()

35
lps/worker/handlers.py Normal file
View File

@ -0,0 +1,35 @@
"""잡 핸들러 — job_type 별 처리. 현재는 SEARCH(검색)만.
검색 핸들러는 소스 어댑터로 검색해 정규화 결과를 반환한다.
필터/이상치/AI 유사도(코어 파이프라인)는 다음 단계에서 이 핸들러 안에 결합한다.
"""
from common.enums import JobType
from services.search.contract import SearchAdapter
def build_search_handler(adapters: dict[str, SearchAdapter], default_source: str = "coupang", limit: int = 40):
"""검색 핸들러 생성. adapters = {source: SearchAdapter}."""
async def handler(job: dict) -> dict:
if job["job_type"] != JobType.SEARCH.value:
raise ValueError(f"unsupported job_type: {job['job_type']}")
payload = job.get("payload") or {}
query = (payload.get("product_name") or "").strip()
if not query:
raise ValueError("empty product_name")
adapter = adapters.get(default_source)
if adapter is None:
raise ValueError(f"no adapter for source: {default_source}")
products = await adapter.search(query, limit=limit)
return {
"source": adapter.source,
"query": query,
"count": len(products),
"products": [p.model_dump() for p in products],
}
return handler

47
lps/worker/notify.py Normal file
View File

@ -0,0 +1,47 @@
"""LISTEN/NOTIFY 리스너 — 잡 적재 시 워커를 즉시 깨운다(폴링 낭비 제거).
전용 asyncpg 연결로 LISTEN 한다(SQLAlchemy 풀과 분리). 알림이 오면 이벤트를 세팅하고,
워커는 claim 이 비었을 때 wait()로 알림 또는 짧은 타임아웃(안전망/reaper)까지 대기한다.
"""
import asyncio
import asyncpg
from config.server_configs import main_db_config
from crud.job_crud import JOB_NOTIFY_CHANNEL
def _dsn() -> str:
c = main_db_config
pw = f":{c.write_pw}" if c.write_pw else ""
return f"postgresql://{c.write_id}{pw}@{c.write_host}:{c.write_port}/{c.name}"
class JobListener:
def __init__(self, channel: str = JOB_NOTIFY_CHANNEL):
self._channel = channel
self._conn: asyncpg.Connection | None = None
self._event = asyncio.Event()
async def start(self):
self._conn = await asyncpg.connect(_dsn())
await self._conn.add_listener(self._channel, self._on_notify)
def _on_notify(self, *_args):
self._event.set()
async def wait(self, timeout: float) -> bool:
"""알림이 오거나 timeout 까지 대기. 알림으로 깨면 True, 타임아웃이면 False."""
try:
await asyncio.wait_for(self._event.wait(), timeout)
return True
except asyncio.TimeoutError:
return False
finally:
self._event.clear()
async def close(self):
if self._conn is not None:
await self._conn.close()
self._conn = None

84
lps/worker/runner.py Normal file
View File

@ -0,0 +1,84 @@
"""워커 루프 + reaper.
워커는 큐에서 잡을 원자적으로 claim → 핸들러 실행 → complete/fail 한다.
- 처리 중 heartbeat 로 lease 를 갱신(긴 잡이 reaper 에 회수되지 않게).
- claim 이 비면 LISTEN 알림 또는 짧은 타임아웃까지 대기(폴링 최소화).
- 핸들러는 주입식(async def(job)->dict) — 프로덕션은 검색 파이프라인, 테스트는 fake.
"""
import asyncio
from common.enums import JobStatus
from common.logger import LOG
from crud.job_crud import JobQueue, compute_backoff
class Worker:
def __init__(self, worker_id: str, queue: JobQueue, handler, lease_sec: int = 120, backoff_fn=compute_backoff):
self.worker_id = worker_id
self.queue = queue
self.handler = handler
self.lease_sec = lease_sec
self.backoff_fn = backoff_fn
async def process_one(self) -> bool:
"""대기 잡 1건을 claim·처리. 처리했으면 True, 없으면 False."""
job = await self.queue.claim(self.worker_id, self.lease_sec)
if not job:
return False
await self._process(job)
return True
async def drain(self) -> int:
"""큐가 빌 때까지 처리(테스트/일회성 배치용). 처리한 잡 수 반환."""
n = 0
while await self.process_one():
n += 1
return n
async def run(self, listener=None, stop: asyncio.Event | None = None, idle_timeout: float = 5.0):
"""상시 루프. stop 이 설정될 때까지 처리하고, 유휴 시 알림/타임아웃까지 대기."""
stop = stop or asyncio.Event()
while not stop.is_set():
worked = await self.process_one()
if not worked:
if listener is not None:
await listener.wait(idle_timeout)
else:
await asyncio.sleep(idle_timeout)
async def _process(self, job: dict):
jid = job["job_id"]
hb = asyncio.create_task(self._heartbeat(jid))
try:
result = await self.handler(job)
await self.queue.complete(jid, self.worker_id, result)
LOG.d(f"[{self.worker_id}] done {jid}")
except Exception as ex:
backoff = self.backoff_fn(job["attempts"])
st = await self.queue.fail(jid, self.worker_id, f"{type(ex).__name__}: {ex}", backoff)
LOG.w(f"[{self.worker_id}] fail {jid} → {JobStatus(st).name if st else '?'} ({type(ex).__name__}: {ex})")
finally:
hb.cancel()
try:
await hb
except asyncio.CancelledError:
pass
async def _heartbeat(self, jid: str):
interval = max(1, self.lease_sec // 3)
while True:
await asyncio.sleep(interval)
await self.queue.renew_lease(jid, self.worker_id, self.lease_sec)
async def run_reaper(queue: JobQueue, stop: asyncio.Event, interval: float = 30.0):
"""만료 lease(워커 사망) 잡을 주기적으로 회수. 재시도 남으면 재큐, 소진되면 DEAD."""
while not stop.is_set():
reclaimed = await queue.reap()
if reclaimed:
LOG.w(f"[reaper] reclaimed {len(reclaimed)} stale job(s)")
try:
await asyncio.wait_for(stop.wait(), interval)
except asyncio.TimeoutError:
pass

52
lps/worker_main.py Normal file
View File

@ -0,0 +1,52 @@
# LPS 워커 프로세스 진입점 (API 와 분리 실행 — 코드베이스 공유, 독립 스케일).
# python worker_main.py
# WORKER_CONCURRENCY=2 python worker_main.py
#
# 쿠팡 검색은 브라우저(Chrome)라 무겁고 컨텍스트당 직렬이므로 기본 동시성은 1.
# 여러 브라우저로 늘리려면 CoupangAdapter 인스턴스를 워커마다 따로 준다.
import asyncio
import os
from common.logger import LOG
from config.server_configs import web_server_config
from crud.job_crud import JobQueue
from services.search.coupang.adapter import CoupangAdapter
from worker.handlers import build_search_handler
from worker.notify import JobListener
from worker.runner import Worker, run_reaper
LOG.SetPrefix(f"{web_server_config.server_name}-worker")
async def main(concurrency: int = 1):
queue = JobQueue()
adapters = {"coupang": CoupangAdapter(headless=False)} # 워커들이 공유(내부 직렬화)
handler = build_search_handler(adapters)
stop = asyncio.Event()
listeners: list[JobListener] = []
tasks: list[asyncio.Task] = []
for i in range(concurrency):
listener = JobListener()
await listener.start()
listeners.append(listener)
worker = Worker(f"worker-{i}", queue, handler)
tasks.append(asyncio.create_task(worker.run(listener, stop)))
tasks.append(asyncio.create_task(run_reaper(queue, stop)))
LOG.i(f"LPS 워커 {concurrency}개 + reaper 기동")
try:
await asyncio.gather(*tasks)
finally:
stop.set()
for listener in listeners:
await listener.close()
for adapter in adapters.values():
await adapter.close()
if __name__ == "__main__":
asyncio.run(main(int(os.environ.get("WORKER_CONCURRENCY", "1"))))