Kris
16455cb303
新增 Nostr 渠道扩展，支持在 Yuxi 平台中集成 Nostr 去中心化社交协议。 包含以下功能模块： - bus: 事件总线与中继通信 - account: 账户管理 - key_utils: 密钥工具 - gateway: SSE/WebSocket 网关接入 - outbound: 外发消息管理 - gift_wrap: Gift Wrap 加密 - nip44: NIP-44 加密协议 - config_schema: 配置模式 - defaults: 默认配置 - profile_core: 用户资料核心 - profile_publisher: 资料发布 - event_utils: 事件工具 - deletion: 事件删除 - reactions: 表情反应 - metrics: 指标监控 - seen_tracker: 已读追踪 - session_route: 会话路由 - state_store: 状态存储
50 lines
1.5 KiB
Python
50 lines
1.5 KiB
Python
import time
|
|
from collections import OrderedDict
|
|
|
|
from .defaults import SEEN_MAX_ENTRIES, SEEN_TTL_MS, SEEN_PRUNE_INTERVAL_S
|
|
|
|
|
|
class SeenTracker:
|
|
def __init__(self, max_entries: int = SEEN_MAX_ENTRIES, ttl_ms: int = SEEN_TTL_MS):
|
|
self.max_entries = max_entries
|
|
self.ttl_s = ttl_ms / 1000
|
|
self._store: OrderedDict[str, float] = OrderedDict()
|
|
self._last_prune = time.monotonic()
|
|
|
|
def peek(self, event_id: str) -> bool:
|
|
self._maybe_prune()
|
|
return event_id in self._store
|
|
|
|
def has(self, event_id: str) -> bool:
|
|
self._maybe_prune()
|
|
if event_id in self._store:
|
|
self._store.move_to_end(event_id)
|
|
return True
|
|
self._store[event_id] = time.monotonic()
|
|
if len(self._store) > self.max_entries:
|
|
self._store.popitem(last=False)
|
|
return False
|
|
|
|
def seed(self, ids: list[str]):
|
|
now = time.monotonic()
|
|
for eid in ids:
|
|
self._store[eid] = now
|
|
while len(self._store) > self.max_entries:
|
|
self._store.popitem(last=False)
|
|
|
|
@property
|
|
def size(self) -> int:
|
|
return len(self._store)
|
|
|
|
def get_recent_ids(self, limit: int = 5000) -> list[str]:
|
|
return list(self._store.keys())[-limit:]
|
|
|
|
def _maybe_prune(self):
|
|
now = time.monotonic()
|
|
if now - self._last_prune < SEEN_PRUNE_INTERVAL_S:
|
|
return
|
|
self._last_prune = now
|
|
expired = [k for k, ts in self._store.items() if now - ts > self.ttl_s]
|
|
for k in expired:
|
|
del self._store[k]
|