545 lines
20 KiB
Python
545 lines
20 KiB
Python
from fastapi import APIRouter, Depends
|
||
from fastapi.responses import JSONResponse
|
||
from sqlalchemy import select, update
|
||
from sqlalchemy.ext.asyncio import AsyncSession
|
||
|
||
from loveace.database.auth.user import ACEUser
|
||
from loveace.database.creator import get_db_session
|
||
from loveace.database.isim.room import RoomBind
|
||
from loveace.router.dependencies.auth import get_user_by_token
|
||
from loveace.router.endpoint.isim.model.protect_router import ISIMRouterErrorToCode
|
||
from loveace.router.endpoint.isim.model.room import (
|
||
BindRoomRequest,
|
||
BindRoomResponse,
|
||
BuildingInfo,
|
||
BuildingListResponse,
|
||
CacheRoomsData,
|
||
CurrentRoomResponse,
|
||
FloorRoomsResponse,
|
||
ForceRefreshResponse,
|
||
RoomDetailResponse,
|
||
)
|
||
from loveace.router.endpoint.isim.utils.isim import ISIMClient, get_isim_client
|
||
from loveace.router.endpoint.isim.utils.lock_manager import get_refresh_lock_manager
|
||
from loveace.router.schemas.uniresponse import UniResponseModel
|
||
|
||
isim_room_router = APIRouter(
|
||
prefix="/room",
|
||
responses=ISIMRouterErrorToCode.gen_code_table(),
|
||
)
|
||
|
||
|
||
@isim_room_router.get(
|
||
"/list",
|
||
summary="[完整数据] 获取所有楼栋、楼层、房间的完整树形结构",
|
||
response_model=UniResponseModel[CacheRoomsData],
|
||
)
|
||
async def get_rooms(
|
||
isim_conn: ISIMClient = Depends(get_isim_client),
|
||
) -> UniResponseModel[CacheRoomsData] | JSONResponse:
|
||
"""
|
||
获取完整的寝室列表（所有楼栋、楼层、房间的树形结构）
|
||
|
||
⚠️ 数据量大：包含所有楼栋的完整数据，适合需要完整数据的场景
|
||
💡 建议：移动端或需要部分数据的场景，请使用其他精细化查询接口
|
||
"""
|
||
try:
|
||
rooms = await isim_conn.get_cached_rooms()
|
||
if not rooms:
|
||
return ISIMRouterErrorToCode().null_response.to_json_response(
|
||
isim_conn.client.logger.trace_id
|
||
)
|
||
return UniResponseModel[CacheRoomsData](
|
||
success=True,
|
||
data=rooms,
|
||
message="获取寝室列表成功",
|
||
error=None,
|
||
)
|
||
except Exception as e:
|
||
isim_conn.client.logger.error(f"获取寝室列表异常: {str(e)}")
|
||
return ISIMRouterErrorToCode().server_error.to_json_response(
|
||
isim_conn.client.logger.trace_id
|
||
)
|
||
|
||
|
||
@isim_room_router.get(
|
||
"/list/buildings",
|
||
summary="[轻量级] 获取所有楼栋列表（仅楼栋信息，不含楼层和房间）",
|
||
response_model=UniResponseModel[BuildingListResponse],
|
||
)
|
||
async def get_all_buildings(
|
||
isim_conn: ISIMClient = Depends(get_isim_client),
|
||
) -> UniResponseModel[BuildingListResponse] | JSONResponse:
|
||
"""
|
||
获取所有楼栋列表（仅楼栋的代码和名称）
|
||
|
||
✅ 数据量小：只返回楼栋列表，不包含楼层和房间
|
||
💡 使用场景：
|
||
- 楼栋选择器
|
||
- 第一级导航菜单
|
||
- 需要快速获取楼栋列表的场景
|
||
"""
|
||
logger = isim_conn.client.logger
|
||
try:
|
||
# 从Hash缓存获取完整数据
|
||
full_data = await isim_conn.get_cached_rooms()
|
||
|
||
if not full_data or not full_data.data:
|
||
logger.warning("楼栋数据不存在")
|
||
return ISIMRouterErrorToCode().null_response.to_json_response(
|
||
logger.trace_id
|
||
)
|
||
|
||
# 提取楼栋信息
|
||
buildings = [
|
||
{"code": building.code, "name": building.name}
|
||
for building in full_data.data
|
||
]
|
||
|
||
result = BuildingListResponse(
|
||
buildings=[BuildingInfo(**b) for b in buildings],
|
||
building_count=len(buildings),
|
||
datetime=full_data.datetime,
|
||
)
|
||
|
||
logger.info(f"成功获取楼栋列表，共 {len(buildings)} 个楼栋")
|
||
return UniResponseModel[BuildingListResponse](
|
||
success=True,
|
||
data=result,
|
||
message=f"获取楼栋列表成功，共 {len(buildings)} 个楼栋",
|
||
error=None,
|
||
)
|
||
|
||
except Exception as e:
|
||
logger.error(f"获取楼栋列表异常: {str(e)}")
|
||
return ISIMRouterErrorToCode().server_error.to_json_response(logger.trace_id)
|
||
|
||
|
||
@isim_room_router.get(
|
||
"/list/building/{building_code}",
|
||
summary="[按楼栋查询] 获取指定楼栋的所有楼层和房间",
|
||
response_model=UniResponseModel[CacheRoomsData],
|
||
)
|
||
async def get_building_rooms(
|
||
building_code: str, isim_conn: ISIMClient = Depends(get_isim_client)
|
||
) -> UniResponseModel[CacheRoomsData] | JSONResponse:
|
||
"""
|
||
获取指定楼栋及其所有楼层和房间的完整数据
|
||
|
||
✅ 数据量适中：只返回单个楼栋的数据，比完整列表小90%+
|
||
💡 使用场景：
|
||
- 用户选择楼栋后，展示该楼栋的所有楼层和房间
|
||
- 楼栋详情页
|
||
- 减少移动端流量消耗
|
||
|
||
Args:
|
||
building_code: 楼栋代码（如：01, 02, 11等）
|
||
"""
|
||
logger = isim_conn.client.logger
|
||
try:
|
||
# 使用Hash精细化查询，只获取指定楼栋
|
||
building_data = await isim_conn.get_building_with_floors(building_code)
|
||
|
||
if not building_data:
|
||
logger.warning(f"楼栋 {building_code} 不存在或无数据")
|
||
return ISIMRouterErrorToCode().null_response.to_json_response(
|
||
logger.trace_id
|
||
)
|
||
|
||
# 构造响应数据
|
||
import datetime
|
||
|
||
result = CacheRoomsData(
|
||
datetime=datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
|
||
data=[building_data],
|
||
)
|
||
|
||
logger.info(
|
||
f"成功获取楼栋 {building_code} 信息，"
|
||
f"楼层数: {len(building_data.floors)}, "
|
||
f"房间数: {sum(len(f.rooms) for f in building_data.floors)}"
|
||
)
|
||
return UniResponseModel[CacheRoomsData](
|
||
success=True,
|
||
data=result,
|
||
message=f"获取楼栋 {building_code} 信息成功",
|
||
error=None,
|
||
)
|
||
|
||
except Exception as e:
|
||
logger.error(f"获取楼栋 {building_code} 信息异常: {str(e)}")
|
||
return ISIMRouterErrorToCode().server_error.to_json_response(logger.trace_id)
|
||
|
||
|
||
@isim_room_router.get(
|
||
"/list/floor/{floor_code}",
|
||
summary="[按楼层查询] 获取指定楼层的所有房间列表",
|
||
response_model=UniResponseModel[FloorRoomsResponse],
|
||
)
|
||
async def get_floor_rooms(
|
||
floor_code: str, isim_conn: ISIMClient = Depends(get_isim_client)
|
||
) -> UniResponseModel[FloorRoomsResponse] | JSONResponse:
|
||
"""
|
||
获取指定楼层的所有房间信息
|
||
|
||
✅ 数据量最小：只返回单个楼层的房间列表，极小数据量
|
||
💡 使用场景：
|
||
- 用户选择楼层后，展示该楼层的所有房间
|
||
- 房间选择器的第三级
|
||
- 移动端分页加载
|
||
- 需要最快响应速度的场景
|
||
|
||
Args:
|
||
floor_code: 楼层代码（如：0101, 0102, 1101等）
|
||
"""
|
||
logger = isim_conn.client.logger
|
||
try:
|
||
# 获取楼层信息
|
||
floor_info = await isim_conn.get_floor_info(floor_code)
|
||
|
||
if not floor_info:
|
||
logger.warning(f"楼层 {floor_code} 不存在")
|
||
return ISIMRouterErrorToCode().null_response.to_json_response(
|
||
logger.trace_id
|
||
)
|
||
|
||
# 获取房间列表（从Hash直接查询，非常快速）
|
||
rooms = await isim_conn.get_rooms_by_floor(floor_code)
|
||
|
||
# 从楼层代码提取楼栋代码（前2位）
|
||
building_code = floor_code[:2] if len(floor_code) >= 2 else ""
|
||
|
||
result = FloorRoomsResponse(
|
||
floor_code=floor_info.code,
|
||
floor_name=floor_info.name,
|
||
building_code=building_code,
|
||
rooms=rooms,
|
||
room_count=len(rooms),
|
||
)
|
||
|
||
logger.info(
|
||
f"成功获取楼层 {floor_code} ({floor_info.name}) 的房间信息，共 {len(rooms)} 个房间"
|
||
)
|
||
return UniResponseModel[FloorRoomsResponse](
|
||
success=True,
|
||
data=result,
|
||
message=f"获取楼层 {floor_code} 的房间信息成功，共 {len(rooms)} 个房间",
|
||
error=None,
|
||
)
|
||
|
||
except Exception as e:
|
||
logger.error(f"获取楼层 {floor_code} 房间信息异常: {str(e)}")
|
||
return ISIMRouterErrorToCode().server_error.to_json_response(logger.trace_id)
|
||
|
||
|
||
@isim_room_router.get(
|
||
"/info/{room_code}",
|
||
summary="[房间详情] 获取单个房间的完整层级信息",
|
||
response_model=UniResponseModel[RoomDetailResponse],
|
||
)
|
||
async def get_room_info(
|
||
room_code: str, isim_conn: ISIMClient = Depends(get_isim_client)
|
||
) -> UniResponseModel[RoomDetailResponse] | JSONResponse:
|
||
"""
|
||
获取指定房间的完整信息（包括楼栋、楼层、房间的完整层级结构）
|
||
|
||
✅ 功能强大：一次性返回房间的完整上下文信息
|
||
💡 使用场景：
|
||
- 房间详情页展示
|
||
- 显示完整的 "楼栋/楼层/房间" 路径
|
||
- 房间搜索结果展示
|
||
- 需要房间完整信息的场景
|
||
|
||
Args:
|
||
room_code: 房间代码（如：010101, 110627等）
|
||
"""
|
||
logger = isim_conn.client.logger
|
||
try:
|
||
# 提取层级代码
|
||
if len(room_code) < 4:
|
||
logger.warning(f"房间代码 {room_code} 格式错误")
|
||
return ISIMRouterErrorToCode().null_response.to_json_response(
|
||
logger.trace_id
|
||
)
|
||
|
||
building_code = room_code[:2]
|
||
floor_code = room_code[:4]
|
||
|
||
# 并发获取所有需要的信息
|
||
import asyncio
|
||
|
||
building_info, floor_info, room_info = await asyncio.gather(
|
||
isim_conn.get_building_info(building_code),
|
||
isim_conn.get_floor_info(floor_code),
|
||
isim_conn.query_room_info_fast(room_code),
|
||
)
|
||
|
||
if not room_info:
|
||
logger.warning(f"房间 {room_code} 不存在")
|
||
return ISIMRouterErrorToCode().null_response.to_json_response(
|
||
logger.trace_id
|
||
)
|
||
|
||
# 构造显示文本
|
||
building_name = building_info.name if building_info else "未知楼栋"
|
||
floor_name = floor_info.name if floor_info else "未知楼层"
|
||
display_text = f"{building_name}/{floor_name}/{room_info.name}"
|
||
|
||
result = RoomDetailResponse(
|
||
room_code=room_info.code,
|
||
room_name=room_info.name,
|
||
floor_code=floor_code,
|
||
floor_name=floor_name,
|
||
building_code=building_code,
|
||
building_name=building_name,
|
||
display_text=display_text,
|
||
)
|
||
|
||
logger.info(f"成功获取房间 {room_code} 的详细信息: {display_text}")
|
||
return UniResponseModel[RoomDetailResponse](
|
||
success=True,
|
||
data=result,
|
||
message=f"获取房间 {room_code} 的详细信息成功",
|
||
error=None,
|
||
)
|
||
|
||
except Exception as e:
|
||
logger.error(f"获取房间 {room_code} 详细信息异常: {str(e)}")
|
||
return ISIMRouterErrorToCode().server_error.to_json_response(logger.trace_id)
|
||
|
||
|
||
@isim_room_router.post(
|
||
"/bind",
|
||
summary="[用户操作] 绑定寝室到当前用户",
|
||
response_model=UniResponseModel[BindRoomResponse],
|
||
)
|
||
async def bind_room(
|
||
bind_request: BindRoomRequest,
|
||
isim_conn: ISIMClient = Depends(get_isim_client),
|
||
db: AsyncSession = Depends(get_db_session),
|
||
) -> UniResponseModel[BindRoomResponse] | JSONResponse:
|
||
"""
|
||
绑定寝室到当前用户（存在即更新）
|
||
|
||
💡 使用场景：
|
||
- 用户首次绑定寝室
|
||
- 用户更换寝室
|
||
- 修改绑定信息
|
||
"""
|
||
logger = isim_conn.client.logger
|
||
try:
|
||
exist = await db.execute(
|
||
select(RoomBind).where(RoomBind.user_id == isim_conn.client.userid)
|
||
)
|
||
exist = exist.scalars().first()
|
||
if exist:
|
||
if exist.roomid == bind_request.room_id:
|
||
return UniResponseModel[BindRoomResponse](
|
||
success=True,
|
||
data=BindRoomResponse(success=True),
|
||
message="宿舍绑定成功",
|
||
error=None,
|
||
)
|
||
else:
|
||
# 使用快速查询方法（从Hash直接获取，无需遍历完整树）
|
||
room_info = await isim_conn.query_room_info_fast(bind_request.room_id)
|
||
roomtext = room_info.name if room_info else None
|
||
|
||
# 如果Hash中没有，回退到完整查询
|
||
if not roomtext:
|
||
roomtext = await isim_conn.query_room_name(bind_request.room_id)
|
||
|
||
await db.execute(
|
||
update(RoomBind)
|
||
.where(RoomBind.user_id == isim_conn.client.userid)
|
||
.values(roomid=bind_request.room_id, roomtext=roomtext)
|
||
)
|
||
await db.commit()
|
||
logger.info(f"更新寝室绑定成功: {roomtext}({bind_request.room_id})")
|
||
return UniResponseModel[BindRoomResponse](
|
||
success=True,
|
||
data=BindRoomResponse(success=True),
|
||
message="宿舍绑定成功",
|
||
error=None,
|
||
)
|
||
else:
|
||
# 使用快速查询方法（从Hash直接获取，无需遍历完整树）
|
||
room_info = await isim_conn.query_room_info_fast(bind_request.room_id)
|
||
roomtext = room_info.name if room_info else None
|
||
|
||
# 如果Hash中没有，回退到完整查询
|
||
if not roomtext:
|
||
roomtext = await isim_conn.query_room_name(bind_request.room_id)
|
||
|
||
new_bind = RoomBind(
|
||
user_id=isim_conn.client.userid,
|
||
roomid=bind_request.room_id,
|
||
roomtext=roomtext,
|
||
)
|
||
db.add(new_bind)
|
||
await db.commit()
|
||
logger.info(f"新增寝室绑定成功: {roomtext}({bind_request.room_id})")
|
||
return UniResponseModel[BindRoomResponse](
|
||
success=True,
|
||
data=BindRoomResponse(success=True),
|
||
message="宿舍绑定成功",
|
||
error=None,
|
||
)
|
||
except Exception as e:
|
||
logger.error(f"绑定寝室异常: {str(e)}")
|
||
return ISIMRouterErrorToCode().server_error.to_json_response(
|
||
isim_conn.client.logger.trace_id
|
||
)
|
||
|
||
|
||
@isim_room_router.get(
|
||
"/current",
|
||
summary="[用户查询] 获取当前用户绑定的宿舍信息",
|
||
response_model=UniResponseModel[CurrentRoomResponse],
|
||
)
|
||
async def get_current_room(
|
||
user: ACEUser = Depends(get_user_by_token),
|
||
isim_conn: ISIMClient = Depends(get_isim_client),
|
||
db: AsyncSession = Depends(get_db_session),
|
||
) -> UniResponseModel[CurrentRoomResponse] | JSONResponse:
|
||
"""
|
||
获取当前用户绑定的宿舍信息，返回 room_code 和 display_text
|
||
|
||
💡 使用场景：
|
||
- 个人中心显示已绑定宿舍
|
||
- 查询当前用户的寝室信息
|
||
- 验证用户是否已绑定寝室
|
||
"""
|
||
logger = isim_conn.client.logger
|
||
try:
|
||
# 查询用户绑定的房间
|
||
result = await db.execute(
|
||
select(RoomBind).where(RoomBind.user_id == user.userid)
|
||
)
|
||
room_bind = result.scalars().first()
|
||
|
||
if not room_bind:
|
||
logger.warning(f"用户 {user.userid} 未绑定宿舍")
|
||
return UniResponseModel[CurrentRoomResponse](
|
||
success=True,
|
||
data=CurrentRoomResponse(
|
||
room_code="",
|
||
display_text="",
|
||
),
|
||
message="获取宿舍信息成功，用户未绑定宿舍",
|
||
error=None,
|
||
)
|
||
|
||
# 优先从Hash缓存快速获取房间显示文本
|
||
display_text = await isim_conn.get_room_display_text(room_bind.roomid)
|
||
if not display_text:
|
||
# 如果缓存中没有，使用数据库中存储的文本
|
||
display_text = room_bind.roomtext
|
||
logger.debug(
|
||
f"Hash缓存中未找到房间 {room_bind.roomid}，使用数据库存储的文本"
|
||
)
|
||
|
||
logger.info(f"成功获取用户 {user.userid} 的宿舍信息: {display_text}")
|
||
return UniResponseModel[CurrentRoomResponse](
|
||
success=True,
|
||
data=CurrentRoomResponse(
|
||
room_code=room_bind.roomid,
|
||
display_text=display_text,
|
||
),
|
||
message="获取宿舍信息成功",
|
||
error=None,
|
||
)
|
||
|
||
except Exception as e:
|
||
logger.error(f"获取当前宿舍异常: {str(e)}")
|
||
return ISIMRouterErrorToCode().server_error.to_json_response(logger.trace_id)
|
||
|
||
|
||
@isim_room_router.post(
|
||
"/refresh",
|
||
summary="[管理操作] 强制刷新房间列表缓存",
|
||
response_model=UniResponseModel[ForceRefreshResponse],
|
||
)
|
||
async def force_refresh_rooms(
|
||
isim_conn: ISIMClient = Depends(get_isim_client),
|
||
) -> UniResponseModel[ForceRefreshResponse] | JSONResponse:
|
||
"""
|
||
强制刷新房间列表缓存（从ISIM系统重新获取数据）
|
||
|
||
⚠️ 限制：
|
||
- 使用全局锁确保同一时间只有一个请求在执行刷新操作
|
||
- 刷新完成后有30分钟的冷却时间
|
||
|
||
💡 使用场景：
|
||
- 发现数据不准确时手动刷新
|
||
- 管理员更新缓存数据
|
||
- 调试和测试
|
||
"""
|
||
logger = isim_conn.client.logger
|
||
lock_manager = get_refresh_lock_manager()
|
||
|
||
try:
|
||
# 尝试获取锁
|
||
acquired, remaining_cooldown = await lock_manager.try_acquire()
|
||
|
||
if not acquired:
|
||
if remaining_cooldown is not None:
|
||
# 在冷却期内
|
||
minutes = int(remaining_cooldown // 60)
|
||
seconds = int(remaining_cooldown % 60)
|
||
message = f"刷新操作冷却中，请在 {minutes} 分 {seconds} 秒后重试"
|
||
logger.warning(f"刷新请求被拒绝: {message}")
|
||
return UniResponseModel[ForceRefreshResponse](
|
||
success=False,
|
||
data=ForceRefreshResponse(
|
||
success=False,
|
||
message=message,
|
||
remaining_cooldown=remaining_cooldown,
|
||
),
|
||
message=message,
|
||
error=None,
|
||
)
|
||
else:
|
||
# 有其他人正在刷新
|
||
message = "其他用户正在刷新房间列表，请稍后再试"
|
||
logger.warning(message)
|
||
return UniResponseModel[ForceRefreshResponse](
|
||
success=False,
|
||
data=ForceRefreshResponse(
|
||
success=False,
|
||
message=message,
|
||
remaining_cooldown=0.0,
|
||
),
|
||
message=message,
|
||
error=None,
|
||
)
|
||
|
||
# 成功获取锁，执行刷新操作
|
||
try:
|
||
logger.info("开始强制刷新房间列表缓存")
|
||
await isim_conn.force_refresh_room_cache()
|
||
logger.info("房间列表缓存刷新完成")
|
||
|
||
return UniResponseModel[ForceRefreshResponse](
|
||
success=True,
|
||
data=ForceRefreshResponse(
|
||
success=True,
|
||
message="房间列表刷新成功",
|
||
remaining_cooldown=0.0,
|
||
),
|
||
message="房间列表刷新成功",
|
||
error=None,
|
||
)
|
||
|
||
finally:
|
||
# 释放锁并设置冷却时间
|
||
lock_manager.release()
|
||
logger.info("刷新锁已释放，冷却时间已设置")
|
||
|
||
except Exception as e:
|
||
logger.error(f"强制刷新房间列表异常: {str(e)}")
|
||
# 确保异常时也释放锁
|
||
if lock_manager.is_refreshing():
|
||
lock_manager.release()
|
||
return ISIMRouterErrorToCode().server_error.to_json_response(logger.trace_id)
|