57 lines
1.7 KiB
Python
57 lines
1.7 KiB
Python
# aider-watch-client buffer | 2026-04-20 @ Asia/Taipei
|
||
"""JSONL buffer 當 awoooi API 不可達時緩衝 events；launchd 定期呼 flush()。"""
|
||
from __future__ import annotations
|
||
import json
|
||
from pathlib import Path
|
||
from typing import Callable
|
||
|
||
# Late-import to allow tests to monkey-patch
|
||
import aider_watch_client.config as _config
|
||
BUFFER_DIR: Path = _config.BUFFER_DIR
|
||
|
||
from ulid import ULID
|
||
|
||
|
||
def _ensure():
|
||
_config.ensure_dirs()
|
||
|
||
|
||
def _get_buffer_dir() -> Path:
|
||
# test 可 monkey-patch 這個 module-level BUFFER_DIR
|
||
import aider_watch_client.buffer as _self
|
||
return _self.BUFFER_DIR
|
||
|
||
|
||
def write(events: list[dict]) -> Path:
|
||
"""批次寫一個 pending file。"""
|
||
bd = _get_buffer_dir()
|
||
bd.mkdir(parents=True, exist_ok=True)
|
||
fp = bd / f"pending_{ULID()}.jsonl"
|
||
with fp.open("w") as f:
|
||
for ev in events:
|
||
f.write(json.dumps(ev, ensure_ascii=False, default=str) + "\n")
|
||
return fp
|
||
|
||
|
||
def flush(post_fn: Callable[[list[dict]], bool]) -> int:
|
||
"""把 buffer 下所有 pending file 重送；成功刪檔，失敗保留。回傳成功筆數。"""
|
||
bd = _get_buffer_dir()
|
||
bd.mkdir(parents=True, exist_ok=True)
|
||
total = 0
|
||
for fp in list(bd.glob("pending_*.jsonl")):
|
||
lines = [l for l in fp.read_text().splitlines() if l.strip()]
|
||
if not lines:
|
||
fp.unlink(); continue
|
||
batch = [json.loads(l) for l in lines]
|
||
ok_all = True
|
||
# API max 50 per request
|
||
for i in range(0, len(batch), 50):
|
||
chunk = batch[i:i+50]
|
||
if not post_fn(chunk):
|
||
ok_all = False
|
||
break
|
||
total += len(chunk)
|
||
if ok_all:
|
||
fp.unlink()
|
||
return total
|