Add public API for player statistics and information queries
This commit is contained in:
咕谷酱
344
app/router/v1/public_user.py
Normal file
344
app/router/v1/public_user.py
Normal file
@@ -0,0 +1,344 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from typing import Literal
|
||||
|
||||
from app.database.lazer_user import User
|
||||
from app.database.statistics import UserStatistics
|
||||
from app.dependencies.database import Database, get_redis
|
||||
from app.models.v1_user import (
|
||||
GetPlayerInfoResponse,
|
||||
PlayerStatsResponse,
|
||||
PlayerEventsResponse,
|
||||
PlayerInfoResponse,
|
||||
PlayerAllResponse,
|
||||
PlayerStatsHistory,
|
||||
GetPlayerCountResponse,
|
||||
PlayerCountData
|
||||
)
|
||||
from app.models.score import GameMode
|
||||
from app.log import logger
|
||||
from app.router.v1.public_router import public_router, AllStrModel
|
||||
|
||||
from fastapi import HTTPException, Query
|
||||
from sqlmodel import select
|
||||
|
||||
|
||||
async def _create_player_mode_stats(session: Database, user: User, mode: GameMode, user_statistics: list[UserStatistics]):
|
||||
"""创建指定模式的玩家统计数据"""
|
||||
from app.models.v1_user import PlayerModeStats
|
||||
|
||||
# 查找对应模式的统计数据
|
||||
statistics = None
|
||||
for stats in user_statistics:
|
||||
if stats.mode == mode:
|
||||
statistics = stats
|
||||
break
|
||||
|
||||
if not statistics:
|
||||
# 如果没有统计数据，返回默认值
|
||||
return PlayerModeStats(
|
||||
id=user.id,
|
||||
mode=int(mode),
|
||||
tscore=0,
|
||||
rscore=0,
|
||||
pp=0.0,
|
||||
plays=0,
|
||||
playtime=0,
|
||||
acc=0.0,
|
||||
max_combo=0,
|
||||
total_hits=0,
|
||||
replay_views=0,
|
||||
xh_count=0,
|
||||
x_count=0,
|
||||
sh_count=0,
|
||||
s_count=0,
|
||||
a_count=0,
|
||||
level=1,
|
||||
level_progress=0,
|
||||
rank=0,
|
||||
country_rank=0,
|
||||
history=PlayerStatsHistory()
|
||||
)
|
||||
|
||||
return PlayerModeStats(
|
||||
id=user.id,
|
||||
mode=int(mode),
|
||||
tscore=statistics.total_score if statistics.total_score else 0,
|
||||
rscore=statistics.ranked_score if statistics.ranked_score else 0,
|
||||
pp=float(statistics.pp) if statistics.pp else 0.0,
|
||||
plays=statistics.play_count if statistics.play_count else 0,
|
||||
playtime=statistics.play_time if statistics.play_time else 0,
|
||||
acc=float(statistics.hit_accuracy) if statistics.hit_accuracy else 0.0,
|
||||
max_combo=statistics.maximum_combo if statistics.maximum_combo else 0,
|
||||
total_hits=statistics.total_hits if statistics.total_hits else 0,
|
||||
replay_views=statistics.replays_watched_by_others if statistics.replays_watched_by_others else 0,
|
||||
xh_count=statistics.grade_ssh if statistics.grade_ssh else 0,
|
||||
x_count=statistics.grade_ss if statistics.grade_ss else 0,
|
||||
sh_count=statistics.grade_sh if statistics.grade_sh else 0,
|
||||
s_count=statistics.grade_s if statistics.grade_s else 0,
|
||||
a_count=statistics.grade_a if statistics.grade_a else 0,
|
||||
level=int(statistics.level_current) if statistics.level_current else 1,
|
||||
level_progress=0, # TODO: 计算等级进度
|
||||
rank=0, # global_rank需要从RankHistory获取
|
||||
country_rank=0, # country_rank需要从其他地方获取
|
||||
history=PlayerStatsHistory() # TODO: 获取PP历史数据
|
||||
)
|
||||
|
||||
|
||||
async def _create_player_info(user: User):
|
||||
"""创建玩家基本信息"""
|
||||
from app.models.v1_user import PlayerInfo
|
||||
|
||||
return PlayerInfo(
|
||||
id=user.id,
|
||||
name=user.username,
|
||||
safe_name=user.username, # 使用 username 作为 safe_name
|
||||
priv=user.priv if user.priv else 1,
|
||||
country=user.country_code if user.country_code else "",
|
||||
silence_end=int(user.silence_end_at.timestamp()) if user.silence_end_at else 0,
|
||||
donor_end=int(user.donor_end_at.timestamp()) if user.donor_end_at else 0,
|
||||
creation_time=int(user.join_date.timestamp()) if user.join_date else 0,
|
||||
latest_activity=int(user.last_visit.timestamp()) if user.last_visit else 0,
|
||||
clan_id=0, # TODO: 从 user 获取战队信息
|
||||
clan_priv=0,
|
||||
preferred_mode=int(user.playmode) if user.playmode else 0,
|
||||
preferred_type=0,
|
||||
play_style=0, # TODO: 从 user.playstyle 获取游戏风格
|
||||
custom_badge_enabled=0,
|
||||
custom_badge_name="",
|
||||
custom_badge_icon="",
|
||||
custom_badge_color="",
|
||||
userpage_content=user.page["html"] if user.page and "html" in user.page else "",
|
||||
recentFailed=0, # TODO: 获取最近失败次数
|
||||
social_discord=user.discord,
|
||||
social_youtube=None,
|
||||
social_twitter=user.twitter,
|
||||
social_twitch=None,
|
||||
social_github=None,
|
||||
social_osu=None,
|
||||
username_history=user.previous_usernames if user.previous_usernames else []
|
||||
)
|
||||
|
||||
|
||||
async def _get_player_events(session: Database, user_id: int):
|
||||
"""获取玩家事件列表"""
|
||||
# TODO: 实现事件查询逻辑
|
||||
# 这里应该查询 app.database.events 表
|
||||
return []
|
||||
|
||||
|
||||
async def _count_online_users_optimized(redis):
|
||||
"""
|
||||
优化的在线用户计数函数
|
||||
首先尝试使用HyperLogLog近似计数，失败则回退到SCAN
|
||||
"""
|
||||
try:
|
||||
# 方案1: 尝试使用更高效的方法 - 如果项目维护了在线用户集合
|
||||
# 检查是否存在在线用户集合键
|
||||
online_set_key = "metadata:online_users_set"
|
||||
if await redis.exists(online_set_key):
|
||||
# 如果存在维护好的在线用户集合，直接获取数量
|
||||
count = await redis.scard(online_set_key)
|
||||
logger.debug(f"Using online users set, count: {count}")
|
||||
return count
|
||||
|
||||
except Exception as e:
|
||||
logger.debug(f"Online users set not available: {e}")
|
||||
|
||||
# 方案2: 回退到优化的SCAN操作
|
||||
online_count = 0
|
||||
cursor = 0
|
||||
scan_iterations = 0
|
||||
max_iterations = 50 # 进一步减少最大迭代次数
|
||||
batch_size = 10000 # 增加批次大小
|
||||
|
||||
try:
|
||||
while cursor != 0 or scan_iterations == 0:
|
||||
if scan_iterations >= max_iterations:
|
||||
logger.warning(f"Redis SCAN reached max iterations ({max_iterations}), breaking")
|
||||
break
|
||||
|
||||
cursor, keys = await redis.scan(
|
||||
cursor,
|
||||
match="metadata:online:*",
|
||||
count=batch_size
|
||||
)
|
||||
online_count += len(keys)
|
||||
scan_iterations += 1
|
||||
|
||||
# 如果连续几次没有找到键，可能已经扫描完成
|
||||
if len(keys) == 0 and scan_iterations > 2:
|
||||
break
|
||||
|
||||
logger.debug(f"Found {online_count} online users after {scan_iterations} scan iterations")
|
||||
return online_count
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Error counting online users: {e}")
|
||||
# 如果SCAN失败，返回0而不是让整个API失败
|
||||
return 0
|
||||
|
||||
|
||||
@public_router.get(
|
||||
"/get_player_info",
|
||||
response_model=GetPlayerInfoResponse,
|
||||
name="获取玩家信息",
|
||||
description="返回指定玩家的信息。",
|
||||
)
|
||||
async def api_get_player_info(
|
||||
session: Database,
|
||||
scope: Literal["stats", "events", "info", "all"] = Query(..., description="信息范围"),
|
||||
id: int | None = Query(None, ge=3, le=2147483647, description="用户 ID"),
|
||||
name: str | None = Query(None, regex=r"^[\w \[\]-]{2,32}$", description="用户名"),
|
||||
):
|
||||
"""
|
||||
获取指定玩家的信息
|
||||
|
||||
Args:
|
||||
scope: 信息范围 - stats(统计), events(事件), info(基本信息), all(全部)
|
||||
id: 用户 ID (可选)
|
||||
name: 用户名 (可选)
|
||||
"""
|
||||
# 验证参数
|
||||
if not id and not name:
|
||||
raise HTTPException(400, "Must provide either id or name")
|
||||
|
||||
# 查询用户
|
||||
if id:
|
||||
user = await session.get(User, id)
|
||||
else:
|
||||
user = (await session.exec(select(User).where(User.username == name))).first()
|
||||
|
||||
if not user:
|
||||
raise HTTPException(404, "User not found")
|
||||
|
||||
try:
|
||||
if scope == "stats":
|
||||
# 获取所有模式的统计数据
|
||||
user_statistics = list((
|
||||
await session.exec(
|
||||
select(UserStatistics).where(UserStatistics.user_id == user.id)
|
||||
)
|
||||
).all())
|
||||
|
||||
stats_dict = {}
|
||||
# 获取所有游戏模式的统计数据
|
||||
all_modes = [GameMode.OSU, GameMode.TAIKO, GameMode.FRUITS, GameMode.MANIA,
|
||||
GameMode.OSURX, GameMode.OSUAP]
|
||||
|
||||
for mode in all_modes:
|
||||
mode_stats = await _create_player_mode_stats(session, user, mode, user_statistics)
|
||||
stats_dict[str(int(mode))] = mode_stats
|
||||
|
||||
return GetPlayerInfoResponse(
|
||||
player=PlayerStatsResponse(stats=stats_dict)
|
||||
)
|
||||
|
||||
elif scope == "events":
|
||||
# 获取事件数据
|
||||
events = await _get_player_events(session, user.id)
|
||||
return GetPlayerInfoResponse(
|
||||
player=PlayerEventsResponse(events=events)
|
||||
)
|
||||
|
||||
elif scope == "info":
|
||||
# 获取基本信息
|
||||
info = await _create_player_info(user)
|
||||
return GetPlayerInfoResponse(
|
||||
player=PlayerInfoResponse(info=info)
|
||||
)
|
||||
|
||||
elif scope == "all":
|
||||
# 获取所有信息
|
||||
# 统计数据
|
||||
user_statistics = list((
|
||||
await session.exec(
|
||||
select(UserStatistics).where(UserStatistics.user_id == user.id)
|
||||
)
|
||||
).all())
|
||||
|
||||
stats_dict = {}
|
||||
all_modes = [GameMode.OSU, GameMode.TAIKO, GameMode.FRUITS, GameMode.MANIA,
|
||||
GameMode.OSURX, GameMode.OSUAP]
|
||||
|
||||
for mode in all_modes:
|
||||
mode_stats = await _create_player_mode_stats(session, user, mode, user_statistics)
|
||||
stats_dict[str(int(mode))] = mode_stats
|
||||
|
||||
# 基本信息
|
||||
info = await _create_player_info(user)
|
||||
|
||||
# 事件
|
||||
events = await _get_player_events(session, user.id)
|
||||
|
||||
return GetPlayerInfoResponse(
|
||||
player=PlayerAllResponse(
|
||||
info=info,
|
||||
stats=stats_dict,
|
||||
events=events
|
||||
)
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Error processing get_player_info for user {user.id}: {e}")
|
||||
raise HTTPException(500, "Internal server error")
|
||||
|
||||
|
||||
@public_router.get(
|
||||
"/get_player_count",
|
||||
response_model=GetPlayerCountResponse,
|
||||
name="获取玩家数量",
|
||||
description="返回在线和总用户数量。",
|
||||
)
|
||||
async def api_get_player_count(
|
||||
session: Database,
|
||||
):
|
||||
"""
|
||||
获取玩家数量统计
|
||||
|
||||
Returns:
|
||||
包含在线用户数和总用户数的响应
|
||||
"""
|
||||
try:
|
||||
redis = get_redis()
|
||||
|
||||
online_cache_key = "stats:online_users_count"
|
||||
cached_online = await redis.get(online_cache_key)
|
||||
|
||||
if cached_online is not None:
|
||||
online_count = int(cached_online)
|
||||
logger.debug(f"Using cached online user count: {online_count}")
|
||||
else:
|
||||
logger.debug("Cache miss, scanning Redis for online users")
|
||||
online_count = await _count_online_users_optimized(redis)
|
||||
|
||||
await redis.setex(online_cache_key, 30, str(online_count))
|
||||
logger.debug(f"Cached online user count: {online_count} for 30 seconds")
|
||||
|
||||
cache_key = "stats:total_users"
|
||||
cached_total = await redis.get(cache_key)
|
||||
|
||||
if cached_total is not None:
|
||||
total_count = int(cached_total)
|
||||
logger.debug(f"Using cached total user count: {total_count}")
|
||||
else:
|
||||
logger.debug("Cache miss, querying database for total user count")
|
||||
from sqlmodel import func, select
|
||||
total_count_result = await session.exec(
|
||||
select(func.count()).select_from(User)
|
||||
)
|
||||
total_count = total_count_result.one()
|
||||
|
||||
await redis.setex(cache_key, 3600, str(total_count))
|
||||
logger.debug(f"Cached total user count: {total_count} for 1 hour")
|
||||
|
||||
return GetPlayerCountResponse(
|
||||
counts=PlayerCountData(
|
||||
online=online_count,
|
||||
total=max(0, total_count - 1) # 减去1个机器人账户，确保不为负数
|
||||
)
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Error getting player count: {e}")
|
||||
raise HTTPException(500, "Internal server error")
|
||||
Reference in New Issue
Block a user