238 lines
8.2 KiB
Python
238 lines
8.2 KiB
Python
from __future__ import annotations
|
||
|
||
import asyncio
|
||
from datetime import datetime, timedelta
|
||
import json
|
||
from typing import Any
|
||
from concurrent.futures import ThreadPoolExecutor
|
||
|
||
from app.dependencies.database import get_redis, get_redis_message
|
||
from app.log import logger
|
||
|
||
from .router import router
|
||
|
||
from fastapi import APIRouter
|
||
from pydantic import BaseModel
|
||
|
||
# Redis key constants
|
||
REDIS_ONLINE_USERS_KEY = "server:online_users"
|
||
REDIS_PLAYING_USERS_KEY = "server:playing_users"
|
||
REDIS_REGISTERED_USERS_KEY = "server:registered_users"
|
||
REDIS_ONLINE_HISTORY_KEY = "server:online_history"
|
||
|
||
# 线程池用于同步Redis操作
|
||
_executor = ThreadPoolExecutor(max_workers=2)
|
||
|
||
async def _redis_exec(func, *args, **kwargs):
|
||
"""在线程池中执行同步Redis操作"""
|
||
loop = asyncio.get_event_loop()
|
||
return await loop.run_in_executor(_executor, func, *args, **kwargs)
|
||
|
||
class ServerStats(BaseModel):
|
||
"""服务器统计信息响应模型"""
|
||
registered_users: int
|
||
online_users: int
|
||
playing_users: int
|
||
timestamp: datetime
|
||
|
||
class OnlineHistoryPoint(BaseModel):
|
||
"""在线历史数据点"""
|
||
timestamp: datetime
|
||
online_count: int
|
||
playing_count: int
|
||
|
||
class OnlineHistoryResponse(BaseModel):
|
||
"""24小时在线历史响应模型"""
|
||
history: list[OnlineHistoryPoint]
|
||
current_stats: ServerStats
|
||
|
||
@router.get("/stats", response_model=ServerStats, tags=["统计"])
|
||
async def get_server_stats() -> ServerStats:
|
||
"""
|
||
获取服务器实时统计信息
|
||
|
||
返回服务器注册用户数、在线用户数、正在游玩用户数等实时统计信息
|
||
"""
|
||
redis = get_redis()
|
||
|
||
try:
|
||
# 并行获取所有统计数据
|
||
registered_count, online_count, playing_count = await asyncio.gather(
|
||
_get_registered_users_count(redis),
|
||
_get_online_users_count(redis),
|
||
_get_playing_users_count(redis)
|
||
)
|
||
|
||
return ServerStats(
|
||
registered_users=registered_count,
|
||
online_users=online_count,
|
||
playing_users=playing_count,
|
||
timestamp=datetime.utcnow()
|
||
)
|
||
except Exception as e:
|
||
logger.error(f"Error getting server stats: {e}")
|
||
# 返回默认值
|
||
return ServerStats(
|
||
registered_users=0,
|
||
online_users=0,
|
||
playing_users=0,
|
||
timestamp=datetime.utcnow()
|
||
)
|
||
|
||
@router.get("/stats/history", response_model=OnlineHistoryResponse, tags=["统计"])
|
||
async def get_online_history() -> OnlineHistoryResponse:
|
||
"""
|
||
获取最近24小时在线统计历史
|
||
|
||
返回过去24小时内每小时的在线用户数和游玩用户数统计
|
||
"""
|
||
redis = get_redis()
|
||
|
||
try:
|
||
# 获取历史数据 - 使用同步Redis客户端
|
||
redis_sync = get_redis_message()
|
||
history_data = await _redis_exec(redis_sync.lrange, REDIS_ONLINE_HISTORY_KEY, 0, -1)
|
||
history_points = []
|
||
|
||
# 处理历史数据
|
||
for data in history_data:
|
||
try:
|
||
point_data = json.loads(data)
|
||
history_points.append(OnlineHistoryPoint(
|
||
timestamp=datetime.fromisoformat(point_data["timestamp"]),
|
||
online_count=point_data["online_count"],
|
||
playing_count=point_data["playing_count"]
|
||
))
|
||
except (json.JSONDecodeError, KeyError, ValueError) as e:
|
||
logger.warning(f"Invalid history data point: {data}, error: {e}")
|
||
continue
|
||
|
||
# 按时间排序（最新的在前）
|
||
history_points.sort(key=lambda x: x.timestamp, reverse=True)
|
||
|
||
# 获取当前统计信息
|
||
current_stats = await get_server_stats()
|
||
|
||
return OnlineHistoryResponse(
|
||
history=history_points,
|
||
current_stats=current_stats
|
||
)
|
||
except Exception as e:
|
||
logger.error(f"Error getting online history: {e}")
|
||
# 返回空历史和当前状态
|
||
current_stats = await get_server_stats()
|
||
return OnlineHistoryResponse(
|
||
history=[],
|
||
current_stats=current_stats
|
||
)
|
||
|
||
async def _get_registered_users_count(redis) -> int:
|
||
"""获取注册用户总数（从缓存）"""
|
||
try:
|
||
count = await redis.get(REDIS_REGISTERED_USERS_KEY)
|
||
return int(count) if count else 0
|
||
except Exception as e:
|
||
logger.error(f"Error getting registered users count: {e}")
|
||
return 0
|
||
|
||
async def _get_online_users_count(redis) -> int:
|
||
"""获取当前在线用户数"""
|
||
try:
|
||
count = await redis.scard(REDIS_ONLINE_USERS_KEY)
|
||
return count
|
||
except Exception as e:
|
||
logger.error(f"Error getting online users count: {e}")
|
||
return 0
|
||
|
||
async def _get_playing_users_count(redis) -> int:
|
||
"""获取当前游玩用户数"""
|
||
try:
|
||
count = await redis.scard(REDIS_PLAYING_USERS_KEY)
|
||
return count
|
||
except Exception as e:
|
||
logger.error(f"Error getting playing users count: {e}")
|
||
return 0
|
||
|
||
# 统计更新功能
|
||
async def update_registered_users_count() -> None:
|
||
"""更新注册用户数缓存"""
|
||
from app.dependencies.database import with_db
|
||
from app.database import User
|
||
from app.const import BANCHOBOT_ID
|
||
from sqlmodel import select, func
|
||
|
||
redis = get_redis()
|
||
try:
|
||
async with with_db() as db:
|
||
# 排除机器人用户（BANCHOBOT_ID）
|
||
result = await db.exec(
|
||
select(func.count()).select_from(User).where(User.id != BANCHOBOT_ID)
|
||
)
|
||
count = result.first()
|
||
await redis.set(REDIS_REGISTERED_USERS_KEY, count or 0, ex=300) # 5分钟过期
|
||
logger.debug(f"Updated registered users count: {count}")
|
||
except Exception as e:
|
||
logger.error(f"Error updating registered users count: {e}")
|
||
|
||
async def add_online_user(user_id: int) -> None:
|
||
"""添加在线用户"""
|
||
redis_sync = get_redis_message()
|
||
redis_async = get_redis()
|
||
try:
|
||
await _redis_exec(redis_sync.sadd, REDIS_ONLINE_USERS_KEY, str(user_id))
|
||
await redis_async.expire(REDIS_ONLINE_USERS_KEY, 3600) # 1小时过期
|
||
except Exception as e:
|
||
logger.error(f"Error adding online user {user_id}: {e}")
|
||
|
||
async def remove_online_user(user_id: int) -> None:
|
||
"""移除在线用户"""
|
||
redis_sync = get_redis_message()
|
||
try:
|
||
await _redis_exec(redis_sync.srem, REDIS_ONLINE_USERS_KEY, str(user_id))
|
||
await _redis_exec(redis_sync.srem, REDIS_PLAYING_USERS_KEY, str(user_id))
|
||
except Exception as e:
|
||
logger.error(f"Error removing online user {user_id}: {e}")
|
||
|
||
async def add_playing_user(user_id: int) -> None:
|
||
"""添加游玩用户"""
|
||
redis_sync = get_redis_message()
|
||
redis_async = get_redis()
|
||
try:
|
||
await _redis_exec(redis_sync.sadd, REDIS_PLAYING_USERS_KEY, str(user_id))
|
||
await redis_async.expire(REDIS_PLAYING_USERS_KEY, 3600) # 1小时过期
|
||
except Exception as e:
|
||
logger.error(f"Error adding playing user {user_id}: {e}")
|
||
|
||
async def remove_playing_user(user_id: int) -> None:
|
||
"""移除游玩用户"""
|
||
redis_sync = get_redis_message()
|
||
try:
|
||
await _redis_exec(redis_sync.srem, REDIS_PLAYING_USERS_KEY, str(user_id))
|
||
except Exception as e:
|
||
logger.error(f"Error removing playing user {user_id}: {e}")
|
||
|
||
async def record_hourly_stats() -> None:
|
||
"""记录每小时统计数据"""
|
||
redis_sync = get_redis_message()
|
||
redis_async = get_redis()
|
||
try:
|
||
online_count = await _get_online_users_count(redis_async)
|
||
playing_count = await _get_playing_users_count(redis_async)
|
||
|
||
history_point = {
|
||
"timestamp": datetime.utcnow().isoformat(),
|
||
"online_count": online_count,
|
||
"playing_count": playing_count
|
||
}
|
||
|
||
# 添加到历史记录
|
||
await _redis_exec(redis_sync.lpush, REDIS_ONLINE_HISTORY_KEY, json.dumps(history_point))
|
||
# 只保留48个数据点
|
||
await _redis_exec(redis_sync.ltrim, REDIS_ONLINE_HISTORY_KEY, 0, 47)
|
||
# 设置过期时间为25小时
|
||
await redis_async.expire(REDIS_ONLINE_HISTORY_KEY, 25 * 3600)
|
||
|
||
logger.debug(f"Recorded hourly stats: online={online_count}, playing={playing_count}")
|
||
except Exception as e:
|
||
logger.error(f"Error recording hourly stats: {e}")
|