Kris
87a8931db3
新增微信客服、微信公众号、微信支付通知三个渠道扩展。 微信客服渠道扩展功能模块： - account: 账户管理 - config: 渠道配置管理 - gateway: SSE/WebSocket 网关接入 - webhook: Webhook 事件处理 - outbound: 外发消息管理 - streaming: 流式消息处理 - pairing: 用户配对与绑定 - security: 安全校验 - crypto: 加解密处理 - dedupe: 消息去重 - customer: 客户管理 - servicer: 客服管理 - session: 会话管理 - status: 会话状态管理 - media: 媒体资源处理 - statistics: 统计功能 - sync: 数据同步 - upgrade: 升级处理 微信公众号渠道扩展功能模块： - config: 渠道配置管理 - gateway: SSE/WebSocket 网关接入 - webhook: Webhook 事件处理 - outbound: 外发消息管理 - streaming: 流式消息处理 - pairing: 用户配对与绑定 - security: 安全校验 - crypto: 加解密处理 - dedupe: 消息去重 - passive_reply: 被动回复 - message: 消息处理 - broadcast: 群发消息 - template: 模板消息 - menu: 菜单管理 - qrcode: 二维码管理 - user: 用户管理 - media: 媒体资源处理 - status: 会话状态管理 微信支付通知渠道扩展功能模块： - config: 渠道配置管理 - webhook: Webhook 事件处理 - crypto: 加解密与签名校验 - cert_manager: 证书管理 - event_router: 事件路由 - dedupe: 消息去重 - pay_repo: 支付数据仓库 - query_client: 查询客户端 - arq_tasks: 异步任务 - callback_compensator: 回调补偿
82 lines
2.4 KiB
Python
82 lines
2.4 KiB
Python
from __future__ import annotations
|
|
|
|
import logging
|
|
import os
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
PAY_DEDUP_TTL_SECONDS = 259200
|
|
|
|
|
|
class PayNotifyDeduplicator:
|
|
def __init__(self, redis_client):
|
|
self._redis = redis_client
|
|
|
|
@staticmethod
|
|
def _key(notification_id: str) -> str:
|
|
return f"wp:dedup:{notification_id}"
|
|
|
|
async def try_claim(self, notification_id: str) -> bool:
|
|
key = self._key(notification_id)
|
|
try:
|
|
result = await self._redis.set(key, "1", nx=True, ex=PAY_DEDUP_TTL_SECONDS)
|
|
return bool(result)
|
|
except Exception:
|
|
logger.exception("PayNotifyDeduplicator Redis error, allowing through")
|
|
return True
|
|
|
|
async def get_status(self, notification_id: str) -> str | None:
|
|
key = self._key(notification_id)
|
|
try:
|
|
val = await self._redis.get(key)
|
|
return val.decode("utf-8") if val else None
|
|
except Exception:
|
|
return None
|
|
|
|
|
|
class InMemoryPayDeduplicator:
|
|
def __init__(self, max_size: int = 2000):
|
|
from collections import OrderedDict
|
|
|
|
self._store: OrderedDict[str, float] = OrderedDict()
|
|
self._max_size = max_size
|
|
|
|
async def try_claim(self, notification_id: str) -> bool:
|
|
import time
|
|
|
|
now = time.monotonic()
|
|
|
|
if notification_id in self._store:
|
|
return False
|
|
|
|
self._store[notification_id] = now
|
|
while len(self._store) > self._max_size:
|
|
self._store.popitem(last=False)
|
|
return True
|
|
|
|
|
|
_DEDUP_INSTANCE = None
|
|
|
|
|
|
def get_pay_deduplicator():
|
|
global _DEDUP_INSTANCE
|
|
if _DEDUP_INSTANCE is not None:
|
|
return _DEDUP_INSTANCE
|
|
|
|
redis_url = os.environ.get("YUXI_REDIS_URL", os.environ.get("REDIS_URL", ""))
|
|
|
|
if redis_url:
|
|
try:
|
|
import redis.asyncio as aioredis
|
|
|
|
client = aioredis.from_url(redis_url, decode_responses=False)
|
|
_DEDUP_INSTANCE = PayNotifyDeduplicator(client)
|
|
logger.info("PayNotifyDeduplicator: using Redis backend (TTL=%ds)", PAY_DEDUP_TTL_SECONDS)
|
|
return _DEDUP_INSTANCE
|
|
except Exception:
|
|
logger.warning("PayNotifyDeduplicator: Redis unavailable, falling back to in-memory")
|
|
|
|
_DEDUP_INSTANCE = InMemoryPayDeduplicator()
|
|
logger.warning("PayNotifyDeduplicator: using in-memory backend (NOT production-ready)")
|
|
return _DEDUP_INSTANCE
|