40da994ae8
Technical Details: https://blog.mxgame.top/2025/11/22/An-On-Demand-Design-Within-SQLModel/
421 lines
13 KiB
Python
421 lines
13 KiB
Python
from typing import Annotated, Literal
|
||
|
||
from app.config import settings
|
||
from app.database import Team, TeamMember, User, UserStatistics
|
||
from app.database.statistics import UserStatisticsModel
|
||
from app.dependencies.database import Database, get_redis
|
||
from app.dependencies.user import get_current_user
|
||
from app.models.score import GameMode
|
||
from app.service.ranking_cache_service import get_ranking_cache_service
|
||
from app.utils import api_doc
|
||
|
||
from .router import router
|
||
|
||
from fastapi import BackgroundTasks, Path, Query, Security
|
||
from pydantic import BaseModel, Field
|
||
from sqlmodel import col, func, select
|
||
|
||
|
||
class TeamStatistics(BaseModel):
|
||
team_id: int
|
||
ruleset_id: int
|
||
play_count: int
|
||
ranked_score: int
|
||
performance: int
|
||
|
||
team: Team
|
||
member_count: int
|
||
|
||
|
||
class TeamResponse(BaseModel):
|
||
ranking: list[TeamStatistics]
|
||
total: int = Field(0, description="战队总数")
|
||
|
||
|
||
SortType = Literal["performance", "score"]
|
||
|
||
|
||
@router.get(
|
||
"/rankings/{ruleset}/team",
|
||
name="获取战队排行榜",
|
||
description="获取在指定模式下按照 pp 排序的战队排行榜",
|
||
tags=["排行榜"],
|
||
response_model=TeamResponse,
|
||
)
|
||
async def get_team_ranking_pp(
|
||
session: Database,
|
||
background_tasks: BackgroundTasks,
|
||
ruleset: Annotated[GameMode, Path(..., description="指定 ruleset")],
|
||
current_user: Annotated[User, Security(get_current_user, scopes=["public"])],
|
||
page: Annotated[int, Query(ge=1, description="页码")] = 1,
|
||
):
|
||
return await get_team_ranking(session, background_tasks, "performance", ruleset, current_user, page)
|
||
|
||
|
||
@router.get(
|
||
"/rankings/{ruleset}/team/{sort}",
|
||
response_model=TeamResponse,
|
||
name="获取战队排行榜",
|
||
description="获取在指定模式下的战队排行榜",
|
||
tags=["排行榜"],
|
||
)
|
||
async def get_team_ranking(
|
||
session: Database,
|
||
background_tasks: BackgroundTasks,
|
||
sort: Annotated[
|
||
SortType,
|
||
Path(
|
||
...,
|
||
description="排名类型：performance 表现分 / score 计分成绩总分 "
|
||
"**这个参数是本服务器额外添加的，不属于 v2 API 的一部分**",
|
||
),
|
||
],
|
||
ruleset: Annotated[GameMode, Path(..., description="指定 ruleset")],
|
||
current_user: Annotated[User, Security(get_current_user, scopes=["public"])],
|
||
page: Annotated[int, Query(ge=1, description="页码")] = 1,
|
||
):
|
||
# 获取 Redis 连接和缓存服务
|
||
redis = get_redis()
|
||
cache_service = get_ranking_cache_service(redis)
|
||
|
||
# 尝试从缓存获取数据（战队排行榜）
|
||
cached_data = await cache_service.get_cached_team_ranking(ruleset, page)
|
||
cached_stats = await cache_service.get_cached_team_stats(ruleset)
|
||
|
||
if cached_data and cached_stats:
|
||
# 从缓存返回数据
|
||
return TeamResponse(
|
||
ranking=[TeamStatistics.model_validate(item) for item in cached_data],
|
||
total=cached_stats.get("total", 0),
|
||
)
|
||
|
||
# 缓存未命中，从数据库查询
|
||
response = TeamResponse(ranking=[], total=0)
|
||
teams = (await session.exec(select(Team))).all()
|
||
valid_teams = [] # 存储有效的战队统计
|
||
|
||
for team in teams:
|
||
statistics = (
|
||
await session.exec(
|
||
select(UserStatistics).where(
|
||
UserStatistics.mode == ruleset,
|
||
UserStatistics.pp > 0,
|
||
col(UserStatistics.user).has(col(User.team_membership).has(col(TeamMember.team_id) == team.id)),
|
||
~User.is_restricted_query(col(UserStatistics.user_id)),
|
||
)
|
||
)
|
||
).all()
|
||
|
||
if not statistics:
|
||
continue
|
||
|
||
pp = 0
|
||
total_ranked_score = 0
|
||
total_play_count = 0
|
||
member_count = 0
|
||
|
||
for stat in statistics:
|
||
total_ranked_score += stat.ranked_score
|
||
total_play_count += stat.play_count
|
||
pp += stat.pp
|
||
member_count += 1
|
||
|
||
stats = TeamStatistics(
|
||
team_id=team.id,
|
||
ruleset_id=int(ruleset),
|
||
play_count=total_play_count,
|
||
ranked_score=total_ranked_score,
|
||
performance=round(pp),
|
||
team=team,
|
||
member_count=member_count,
|
||
)
|
||
valid_teams.append(stats)
|
||
|
||
# 排序
|
||
if sort == "performance":
|
||
valid_teams.sort(key=lambda x: x.performance, reverse=True)
|
||
else:
|
||
valid_teams.sort(key=lambda x: x.ranked_score, reverse=True)
|
||
|
||
# 计算总数
|
||
total_count = len(valid_teams)
|
||
|
||
# 分页处理
|
||
page_size = 50
|
||
start_idx = (page - 1) * page_size
|
||
end_idx = start_idx + page_size
|
||
|
||
# 获取当前页的数据
|
||
current_page_data = valid_teams[start_idx:end_idx]
|
||
|
||
# 异步缓存数据（不等待完成）
|
||
cache_data = [item.model_dump() for item in current_page_data]
|
||
stats_data = {"total": total_count}
|
||
|
||
# 创建后台任务来缓存数据
|
||
background_tasks.add_task(
|
||
cache_service.cache_team_ranking,
|
||
ruleset,
|
||
cache_data,
|
||
page,
|
||
ttl=settings.ranking_cache_expire_minutes * 60,
|
||
)
|
||
|
||
# 缓存统计信息
|
||
background_tasks.add_task(
|
||
cache_service.cache_team_stats,
|
||
ruleset,
|
||
stats_data,
|
||
ttl=settings.ranking_cache_expire_minutes * 60,
|
||
)
|
||
|
||
# 返回当前页的结果
|
||
response.ranking = current_page_data
|
||
response.total = total_count
|
||
return response
|
||
|
||
|
||
class CountryStatistics(BaseModel):
|
||
code: str
|
||
active_users: int
|
||
play_count: int
|
||
ranked_score: int
|
||
performance: int
|
||
|
||
|
||
class CountryResponse(BaseModel):
|
||
ranking: list[CountryStatistics]
|
||
|
||
|
||
@router.get(
|
||
"/rankings/{ruleset}/country",
|
||
name="获取地区排行榜",
|
||
description="获取在指定模式下按照 pp 排序的地区排行榜",
|
||
tags=["排行榜"],
|
||
response_model=CountryResponse,
|
||
)
|
||
async def get_country_ranking_pp(
|
||
session: Database,
|
||
background_tasks: BackgroundTasks,
|
||
ruleset: Annotated[GameMode, Path(..., description="指定 ruleset")],
|
||
current_user: Annotated[User, Security(get_current_user, scopes=["public"])],
|
||
page: Annotated[int, Query(ge=1, description="页码")] = 1,
|
||
):
|
||
return await get_country_ranking(session, background_tasks, ruleset, "performance", current_user, page)
|
||
|
||
|
||
@router.get(
|
||
"/rankings/{ruleset}/country/{sort}",
|
||
response_model=CountryResponse,
|
||
name="获取地区排行榜",
|
||
description="获取在指定模式下的地区排行榜",
|
||
tags=["排行榜"],
|
||
)
|
||
async def get_country_ranking(
|
||
session: Database,
|
||
background_tasks: BackgroundTasks,
|
||
ruleset: Annotated[GameMode, Path(..., description="指定 ruleset")],
|
||
sort: Annotated[
|
||
SortType,
|
||
Path(
|
||
...,
|
||
description="排名类型：performance 表现分 / score 计分成绩总分 "
|
||
"**这个参数是本服务器额外添加的，不属于 v2 API 的一部分**",
|
||
),
|
||
],
|
||
current_user: Annotated[User, Security(get_current_user, scopes=["public"])],
|
||
page: Annotated[int, Query(ge=1, description="页码")] = 1,
|
||
):
|
||
# 获取 Redis 连接和缓存服务
|
||
redis = get_redis()
|
||
cache_service = get_ranking_cache_service(redis)
|
||
|
||
# 尝试从缓存获取数据
|
||
cached_data = await cache_service.get_cached_country_ranking(ruleset, page)
|
||
|
||
if cached_data:
|
||
# 从缓存返回数据
|
||
return CountryResponse(ranking=[CountryStatistics.model_validate(item) for item in cached_data])
|
||
|
||
# 缓存未命中，从数据库查询
|
||
response = CountryResponse(ranking=[])
|
||
countries = (await session.exec(select(User.country_code).distinct())).all()
|
||
|
||
for country in countries:
|
||
if not country: # 跳过空的国家代码
|
||
continue
|
||
|
||
statistics = (
|
||
await session.exec(
|
||
select(UserStatistics).where(
|
||
UserStatistics.mode == ruleset,
|
||
UserStatistics.pp > 0,
|
||
col(UserStatistics.user).has(country_code=country),
|
||
col(UserStatistics.user).has(is_active=True),
|
||
~User.is_restricted_query(col(UserStatistics.user_id)),
|
||
)
|
||
)
|
||
).all()
|
||
|
||
if not statistics: # 跳过没有数据的国家
|
||
continue
|
||
|
||
pp = 0
|
||
active_users = 0
|
||
total_play_count = 0
|
||
total_ranked_score = 0
|
||
|
||
for stat in statistics:
|
||
active_users += 1
|
||
total_play_count += stat.play_count
|
||
total_ranked_score += stat.ranked_score
|
||
pp += stat.pp
|
||
|
||
country_stats = CountryStatistics(
|
||
code=country,
|
||
active_users=active_users,
|
||
play_count=total_play_count,
|
||
ranked_score=total_ranked_score,
|
||
performance=round(pp),
|
||
)
|
||
response.ranking.append(country_stats)
|
||
|
||
if sort == "performance":
|
||
response.ranking.sort(key=lambda x: x.performance, reverse=True)
|
||
else:
|
||
response.ranking.sort(key=lambda x: x.ranked_score, reverse=True)
|
||
|
||
# 分页处理
|
||
page_size = 50
|
||
start_idx = (page - 1) * page_size
|
||
end_idx = start_idx + page_size
|
||
|
||
# 获取当前页的数据
|
||
current_page_data = response.ranking[start_idx:end_idx]
|
||
|
||
# 异步缓存数据（不等待完成）
|
||
cache_data = [item.model_dump() for item in current_page_data]
|
||
|
||
# 创建后台任务来缓存数据
|
||
background_tasks.add_task(
|
||
cache_service.cache_country_ranking,
|
||
ruleset,
|
||
cache_data,
|
||
page,
|
||
ttl=settings.ranking_cache_expire_minutes * 60,
|
||
)
|
||
|
||
# 返回当前页的结果
|
||
response.ranking = current_page_data
|
||
return response
|
||
|
||
|
||
@router.get(
|
||
"/rankings/{ruleset}/{sort}",
|
||
responses={
|
||
200: api_doc(
|
||
"用户排行榜",
|
||
{"ranking": list[UserStatisticsModel], "total": int},
|
||
["user.country", "user.cover"],
|
||
name="TopUsersResponse",
|
||
)
|
||
},
|
||
name="获取用户排行榜",
|
||
description="获取在指定模式下的用户排行榜",
|
||
tags=["排行榜"],
|
||
)
|
||
async def get_user_ranking(
|
||
session: Database,
|
||
background_tasks: BackgroundTasks,
|
||
ruleset: Annotated[GameMode, Path(..., description="指定 ruleset")],
|
||
sort: Annotated[SortType, Path(..., description="排名类型：performance 表现分 / score 计分成绩总分")],
|
||
current_user: Annotated[User, Security(get_current_user, scopes=["public"])],
|
||
country: Annotated[str | None, Query(description="国家代码")] = None,
|
||
page: Annotated[int, Query(ge=1, description="页码")] = 1,
|
||
):
|
||
# 获取 Redis 连接和缓存服务
|
||
redis = get_redis()
|
||
cache_service = get_ranking_cache_service(redis)
|
||
|
||
# 尝试从缓存获取数据
|
||
cached_data = await cache_service.get_cached_ranking(ruleset, sort, country, page)
|
||
cached_stats = await cache_service.get_cached_stats(ruleset, sort, country)
|
||
|
||
if cached_data and cached_stats:
|
||
# 从缓存返回数据
|
||
return {
|
||
"ranking": cached_data,
|
||
"total": cached_stats.get("total", 0),
|
||
}
|
||
|
||
# 缓存未命中，从数据库查询
|
||
wheres = [
|
||
col(UserStatistics.mode) == ruleset,
|
||
col(UserStatistics.pp) > 0,
|
||
col(UserStatistics.is_ranked),
|
||
]
|
||
include = UserStatistics.RANKING_INCLUDES.copy()
|
||
if sort == "performance":
|
||
order_by = col(UserStatistics.pp).desc()
|
||
include.append("rank_change_since_30_days")
|
||
else:
|
||
order_by = col(UserStatistics.ranked_score).desc()
|
||
if country:
|
||
wheres.append(col(UserStatistics.user).has(country_code=country.upper()))
|
||
include.append("country_rank")
|
||
|
||
# 查询总数
|
||
count_query = select(func.count()).select_from(UserStatistics).where(*wheres)
|
||
total_count_result = await session.exec(count_query)
|
||
total_count = total_count_result.one()
|
||
|
||
statistics_list = await session.exec(
|
||
select(UserStatistics)
|
||
.where(
|
||
*wheres,
|
||
~User.is_restricted_query(col(UserStatistics.user_id)),
|
||
)
|
||
.order_by(order_by)
|
||
.limit(50)
|
||
.offset(50 * (page - 1))
|
||
)
|
||
|
||
# 转换为响应格式
|
||
ranking_data = []
|
||
for statistics in statistics_list:
|
||
user_stats_resp = await UserStatisticsModel.transform(
|
||
statistics, includes=include, user_country=current_user.country_code
|
||
)
|
||
ranking_data.append(user_stats_resp)
|
||
|
||
# 异步缓存数据（不等待完成）
|
||
# 使用配置文件中的TTL设置
|
||
cache_data = ranking_data
|
||
stats_data = {"total": total_count}
|
||
|
||
# 创建后台任务来缓存数据
|
||
background_tasks.add_task(
|
||
cache_service.cache_ranking,
|
||
ruleset,
|
||
sort,
|
||
cache_data,
|
||
country,
|
||
page,
|
||
ttl=settings.ranking_cache_expire_minutes * 60,
|
||
)
|
||
|
||
# 缓存统计信息
|
||
background_tasks.add_task(
|
||
cache_service.cache_stats,
|
||
ruleset,
|
||
sort,
|
||
stats_data,
|
||
country,
|
||
ttl=settings.ranking_cache_expire_minutes * 60,
|
||
)
|
||
|
||
return {
|
||
"ranking": ranking_data,
|
||
"total": total_count,
|
||
}
|