79 lines
3.4 KiB
Python
79 lines
3.4 KiB
Python
from typing import Any
|
|
|
|
import jwt
|
|
from fastapi import Request
|
|
|
|
from common.enums import RedisInitKeyConfig
|
|
from common.vo import CrudResponseModel
|
|
from config.env import AppConfig, JwtConfig
|
|
from exceptions.exception import ServiceException
|
|
from module_admin.entity.vo.online_vo import DeleteOnlineModel, OnlineQueryModel
|
|
from utils.common_util import CamelCaseUtil
|
|
|
|
|
|
class OnlineService:
|
|
"""
|
|
在线用户管理模块服务层
|
|
"""
|
|
|
|
@classmethod
|
|
async def get_online_list_services(cls, request: Request, query_object: OnlineQueryModel) -> list[dict[str, Any]]:
|
|
"""
|
|
获取在线用户表信息service
|
|
|
|
:param request: Request对象
|
|
:param query_object: 查询参数对象
|
|
:return: 在线用户列表信息
|
|
"""
|
|
access_token_keys = await request.app.state.redis.keys(f'{RedisInitKeyConfig.ACCESS_TOKEN.key}*')
|
|
if not access_token_keys:
|
|
access_token_keys = []
|
|
access_token_values_list = [await request.app.state.redis.get(key) for key in access_token_keys]
|
|
online_info_list = []
|
|
for item in access_token_values_list:
|
|
payload = jwt.decode(item, JwtConfig.jwt_secret_key, algorithms=[JwtConfig.jwt_algorithm])
|
|
online_dict = {
|
|
'token_id': payload.get('session_id') if AppConfig.app_same_time_login else payload.get('user_id'),
|
|
'user_name': payload.get('user_name'),
|
|
'dept_name': payload.get('dept_name'),
|
|
'ipaddr': payload.get('login_info').get('ipaddr'),
|
|
'login_location': payload.get('login_info').get('loginLocation'),
|
|
'browser': payload.get('login_info').get('browser'),
|
|
'os': payload.get('login_info').get('os'),
|
|
'login_time': payload.get('login_info').get('loginTime'),
|
|
}
|
|
if query_object.user_name and not query_object.ipaddr:
|
|
if query_object.user_name == payload.get('user_name'):
|
|
online_info_list = [online_dict]
|
|
break
|
|
elif not query_object.user_name and query_object.ipaddr:
|
|
if query_object.ipaddr == payload.get('login_info').get('ipaddr'):
|
|
online_info_list = [online_dict]
|
|
break
|
|
elif query_object.user_name and query_object.ipaddr:
|
|
if query_object.user_name == payload.get('user_name') and query_object.ipaddr == payload.get(
|
|
'login_info'
|
|
).get('ipaddr'):
|
|
online_info_list = [online_dict]
|
|
break
|
|
else:
|
|
online_info_list.append(online_dict)
|
|
|
|
return CamelCaseUtil.transform_result(online_info_list)
|
|
|
|
@classmethod
|
|
async def delete_online_services(cls, request: Request, page_object: DeleteOnlineModel) -> CrudResponseModel:
|
|
"""
|
|
强退在线用户信息service
|
|
|
|
:param request: Request对象
|
|
:param page_object: 强退在线用户对象
|
|
:return: 强退在线用户校验结果
|
|
"""
|
|
if page_object.token_ids:
|
|
token_id_list = page_object.token_ids.split(',')
|
|
for token_id in token_id_list:
|
|
await request.app.state.redis.delete(f'{RedisInitKeyConfig.ACCESS_TOKEN.key}:{token_id}')
|
|
return CrudResponseModel(is_success=True, message='强退成功')
|
|
raise ServiceException(message='传入session_id为空')
|