Kris
a6fa7245e5
新增飞书机器人适配器全套功能，包括： - 基础适配器入口与工具导出 - 消息格式化、卡片渲染、回复调度逻辑 - 会话ID生成、模型覆盖策略 - 消息发送缓存、顺序队列管理 - 飞书签名验证、加解密webhook请求 - 审批权限校验、机器人菜单事件处理 - 文档评论、钉消息、语音转码处理 - 静态/动态目录管理、子代理生命周期管理 - 各类工具集：聊天、云盘、文档、知识库API封装
49 lines
1.5 KiB
Python
49 lines
1.5 KiB
Python
from __future__ import annotations
|
|
|
|
import time
|
|
|
|
|
|
|
|
class ThreadBindingManager:
|
|
DEFAULT_TTL_S = 3600
|
|
|
|
def __init__(self, ttl_s: int = DEFAULT_TTL_S):
|
|
self._bindings: dict[str, str] = {}
|
|
self._reverse: dict[str, str] = {}
|
|
self._timestamps: dict[str, float] = {}
|
|
self._ttl_s = ttl_s
|
|
|
|
def bind(self, thread_id: str, session_id: str) -> None:
|
|
self._bindings[thread_id] = session_id
|
|
self._reverse[session_id] = thread_id
|
|
self._timestamps[thread_id] = time.monotonic()
|
|
|
|
def unbind_thread(self, thread_id: str) -> None:
|
|
session_id = self._bindings.pop(thread_id, None)
|
|
if session_id:
|
|
self._reverse.pop(session_id, None)
|
|
self._timestamps.pop(thread_id, None)
|
|
|
|
def unbind_session(self, session_id: str) -> None:
|
|
thread_id = self._reverse.pop(session_id, None)
|
|
if thread_id:
|
|
self._bindings.pop(thread_id, None)
|
|
self._timestamps.pop(thread_id, None)
|
|
|
|
def get_session(self, thread_id: str) -> str | None:
|
|
session_id = self._bindings.get(thread_id)
|
|
if session_id:
|
|
ts = self._timestamps.get(thread_id, 0)
|
|
if time.monotonic() - ts > self._ttl_s:
|
|
self.unbind_thread(thread_id)
|
|
return None
|
|
return session_id
|
|
|
|
def get_thread(self, session_id: str) -> str | None:
|
|
return self._reverse.get(session_id)
|
|
|
|
def clear(self) -> None:
|
|
self._bindings.clear()
|
|
self._reverse.clear()
|
|
self._timestamps.clear()
|