Kris
05abecc02b
该提交实现了支持企业微信、微信公众号、个人微信桥接三种模式的完整微信渠道适配器，包含以下核心模块： 1. 基础认证与配置相关：auth_adapter、config_reload、setup_contract等 2. 消息处理与格式转换：format、attachment_adapter、outbound_adapter等 3. 多模式客户端支持：wecom/mp子模块，包含加解密、消息收发能力 4. 辅助能力：限速器、防抖、会话绑定、事件映射、模板渲染等 5. 扩展能力：二维码登录、消息读取、特权用户、心跳监控等 实现了完整的微信生态对接能力，支持消息收发、事件处理、API调用限流、配置热重载等功能。
58 lines
1.8 KiB
Python
58 lines
1.8 KiB
Python
from __future__ import annotations
|
|
|
|
import asyncio
|
|
from collections.abc import Callable, Awaitable
|
|
from typing import Any
|
|
|
|
from yuxi.channels.models import ChannelMessage
|
|
|
|
QUEUE_DEBOUNCE_MS = 500
|
|
|
|
|
|
class WeChatQueueDebounce:
|
|
def __init__(self, debounce_ms: int = QUEUE_DEBOUNCE_MS):
|
|
self._debounce_s = debounce_ms / 1000.0
|
|
self._pending: dict[str, asyncio.Task] = {}
|
|
self._queues: dict[str, list[ChannelMessage]] = {}
|
|
self._lock = asyncio.Lock()
|
|
|
|
async def enqueue(
|
|
self,
|
|
session_key: str,
|
|
message: ChannelMessage,
|
|
handler: Callable[[list[ChannelMessage]], Awaitable[Any]],
|
|
) -> None:
|
|
async with self._lock:
|
|
if session_key not in self._queues:
|
|
self._queues[session_key] = []
|
|
self._queues[session_key].append(message)
|
|
|
|
if session_key in self._pending:
|
|
self._pending[session_key].cancel()
|
|
self._pending[session_key] = asyncio.create_task(self._flush_after_delay(session_key, handler))
|
|
|
|
async def _flush_after_delay(
|
|
self,
|
|
session_key: str,
|
|
handler: Callable[[list[ChannelMessage]], Awaitable[Any]],
|
|
) -> None:
|
|
await asyncio.sleep(self._debounce_s)
|
|
async with self._lock:
|
|
messages = self._queues.pop(session_key, [])
|
|
self._pending.pop(session_key, None)
|
|
if messages:
|
|
await handler(messages)
|
|
|
|
async def flush_all(self) -> None:
|
|
async with self._lock:
|
|
keys = list(self._queues.keys())
|
|
for key in keys:
|
|
async with self._lock:
|
|
if key in self._pending:
|
|
self._pending[key].cancel()
|
|
self._pending.pop(key, None)
|
|
|
|
@property
|
|
def debounce_ms(self) -> int:
|
|
return int(self._debounce_s * 1000)
|