zhenxun_plugin_farm/core/player/playerPool.py

import threading
import time
from typing import Any

from zhenxun.services.log import logger


class CPlayerPool:
    """
    用户池管理类
    管理用户对象的生命周期，支持自动清理超时用户
    """

    def __init__(self, timeoutSeconds: int = 300, cleanupInterval: int = 3600):
        """
        初始化用户池

        Args:
            timeoutSeconds: 用户超时时间（秒），默认5分钟
            cleanupInterval: 清理间隔（秒），默认1小时
        """
        self._players: dict[str, dict[str, Any]] = {}
        self._lock = threading.RLock()
        self.timeoutSeconds = timeoutSeconds
        self.cleanupInterval = cleanupInterval

        # 启动后台清理线程
        self._cleanupThread = threading.Thread(target=self._cleanupWorker, daemon=True)
        self._running = True
        self._cleanupThread.start()

        logger.debug(
            f"用户池初始化完成，超时时间: {timeoutSeconds}秒, 清理间隔: {cleanupInterval}秒"
        )

    def createUser(self, uid: str, userObj: Any) -> bool:
        """
        创建并管理用户对象

        Args:
            uid: 用户ID
            userObj: 用户对象

        Returns:
            bool: 是否创建成功
        """
        with self._lock:
            if uid in self._players:
                logger.debug(f"用户 {uid} 已存在，正在覆盖")
                # 可以选择返回False或者覆盖，这里选择覆盖
                # return False

            self._players[uid] = {
                "object": userObj,
                "lastActive": time.time(),
                "activeCount": 0,
            }
            logger.debug(f"用户 {uid} 创建并开始管理")
            return True

    def getUser(self, uid: str) -> Any | None:
        """
        获取用户对象并刷新活跃时间

        Args:
            uid: 用户ID

        Returns:
            Optional[Any]: 用户对象，如果不存在或已超时则返回None
        """
        with self._lock:
            userData = self._players.get(uid)

            if not userData:
                logger.debug(f"用户 {uid} 不存在")
                return None

            # 检查是否已超时（防御性检查）
            currentTime = time.time()
            if currentTime - userData["lastActive"] > self.timeoutSeconds:
                logger.debug(f"用户 {uid} 在获取操作期间已超时")
                self._removeUser(uid)
                return None

            # 刷新活跃时间
            userData["lastActive"] = currentTime
            userData["activeCount"] += 1

            logger.debug(f"用户 {uid} 获取成功，活跃次数: {userData['activeCount']}")
            return userData["object"]

    def updateUser(self, uid: str, userObj: Any) -> bool:
        """
        更新用户对象

        Args:
            uid: 用户ID
            userObj: 新的用户对象

        Returns:
            bool: 是否更新成功
        """
        with self._lock:
            if uid not in self._players:
                logger.debug(f"用户 {uid} 不存在，无法更新")
                return False

            self._players[uid]["object"] = userObj
            self._players[uid]["lastActive"] = time.time()
            logger.debug(f"用户 {uid} 更新成功")
            return True

    def removeUser(self, uid: str) -> bool:
        """
        主动移除用户

        Args:
            uid: 用户ID

        Returns:
            bool: 是否移除成功
        """
        with self._lock:
            return self._removeUser(uid)

    def _removeUser(self, uid: str) -> bool:
        """内部移除用户方法"""
        if uid in self._players:
            userData = self._players.pop(uid)
            # 如果需要清理资源，可以在这里处理
            if hasattr(userData["object"], "close"):
                try:
                    userData["object"].close()
                except Exception as e:
                    logger.debug(f"关闭用户 {uid} 时出错: {e}")

            logger.debug(f"用户 {uid} 已移除，总活跃次数: {userData['activeCount']}")
            return True
        return False

    def _cleanupWorker(self):
        """后台清理线程的工作函数"""
        while self._running:
            try:
                self._cleanupExpiredUsers()
            except Exception as e:
                logger.debug(f"清理工作线程出错: {e}")

            # 休眠指定间隔
            time.sleep(self.cleanupInterval)

    def _cleanupExpiredUsers(self):
        """清理超时用户"""
        currentTime = time.time()
        expiredUsers = []

        # 首先收集过期的用户ID，避免在迭代中修改字典
        with self._lock:
            for uid, userData in self._players.items():
                if currentTime - userData["lastActive"] > self.timeoutSeconds:
                    expiredUsers.append(uid)

        # 移除过期用户
        for uid in expiredUsers:
            with self._lock:
                # 再次检查，防止在收集和移除之间用户被更新
                if (
                    uid in self._players
                    and currentTime - self._players[uid]["lastActive"]
                    > self.timeoutSeconds
                ):
                    self._removeUser(uid)

        if expiredUsers:
            logger.debug(f"已清理 {len(expiredUsers)} 个过期用户: {expiredUsers}")

    def getActiveUsers(self) -> dict[str, dict[str, Any]]:
        """
        获取当前活跃用户信息

        Returns:
            Dict: 用户信息字典
        """
        with self._lock:
            # 返回副本避免外部修改
            return {
                uid: {
                    "lastActive": data["lastActive"],
                    "activeCount": data["activeCount"],
                    "timeRemaining": self.timeoutSeconds
                    - (time.time() - data["lastActive"]),
                }
                for uid, data in self._players.items()
            }

    def userCount(self) -> int:
        """获取当前用户数量"""
        with self._lock:
            return len(self._players)

    def shutdown(self):
        """关闭用户池，清理资源"""
        self._running = False
        if self._cleanupThread.is_alive():
            self._cleanupThread.join(timeout=5)

        # 清理所有用户
        with self._lock:
            uids = list(self._players.keys())
            for uid in uids:
                self._removeUser(uid)

        logger.debug("用户池关闭完成")


g_pUserPool = CPlayerPool()