FastapiAdmin/backend/app/api/v1/module_monitor/job/service.py

# -*- coding: utf-8 -*-

from typing import Any, List, Dict, Optional

from app.core.ap_scheduler import SchedulerUtil
from app.core.exceptions import CustomException
from app.utils.cron_util import CronUtil
from app.utils.excel_util import ExcelUtil
from app.api.v1.module_system.auth.schema import AuthSchema
from .schema import JobCreateSchema, JobUpdateSchema, JobOutSchema, JobLogOutSchema
from .param import JobQueryParam, JobLogQueryParam
from .crud import JobCRUD, JobLogCRUD


class JobService:
    """
    定时任务管理模块服务层
    """
    
    @classmethod
    async def get_job_detail_service(cls, auth: AuthSchema, id: int) -> Dict:
        obj = await JobCRUD(auth).get_obj_by_id_crud(id=id)
        return JobOutSchema.model_validate(obj).model_dump()
    
    @classmethod
    async def get_job_list_service(cls, auth: AuthSchema, search: Optional[JobQueryParam] = None, order_by: Optional[List[Dict[str, str]]] = None) -> List[Dict]:
        obj_list = await JobCRUD(auth).get_obj_list_crud(search=search.__dict__, order_by=order_by)
        return [JobOutSchema.model_validate(obj).model_dump() for obj in obj_list]
    
    @classmethod
    async def create_job_service(cls, auth: AuthSchema, data: JobCreateSchema) -> Dict:
        exist_obj = await JobCRUD(auth).get(name=data.name)
        if exist_obj:
            raise CustomException(msg='创建失败，该定时任务已存在')
        if data.trigger == 'cron' and data.trigger_args and not CronUtil.validate_cron_expression(data.trigger_args):
            raise CustomException(msg=f'新增定时任务{data.name}失败, Cron表达式不正确')
        
        obj = await JobCRUD(auth).create_obj_crud(data=data)
        SchedulerUtil().add_job(job_info=obj)
        return JobOutSchema.model_validate(obj).model_dump()
    
    @classmethod
    async def update_job_service(cls, auth: AuthSchema, id:int, data: JobUpdateSchema) -> Dict:
        exist_obj = await JobCRUD(auth).get_obj_by_id_crud(id=id)
        if not exist_obj:
            raise CustomException(msg='更新失败，该定时任务不存在')
        if data.trigger == 'cron' and data.trigger_args and not CronUtil.validate_cron_expression(data.trigger_args):
            raise CustomException(msg=f'新增定时任务{data.name}失败, Cron表达式不正确')
        obj = await JobCRUD(auth).update_obj_crud(id=id, data=data)
        if not obj:
            raise CustomException(msg='更新失败，该数据定时任务不存在')
        SchedulerUtil().modify_job(job_id=obj.id)
        return JobOutSchema.model_validate(obj).model_dump()
    
    @classmethod
    async def delete_job_service(cls, auth: AuthSchema, ids: list[int]) -> None:
        if len(ids) < 1:
            raise CustomException(msg='删除失败，删除对象不能为空')
        for id in ids:
            exist_obj = await JobCRUD(auth).get_obj_by_id_crud(id=id)
            if not exist_obj:
                raise CustomException(msg='删除失败，该数据定时任务不存在')
            SchedulerUtil.remove_job(job_id=id)
        await JobCRUD(auth).delete_obj_crud(ids=ids)
        

    @classmethod
    async def clear_job_service(cls, auth: AuthSchema) -> None:
        SchedulerUtil().clear_jobs()
        await JobCRUD(auth).clear_obj_crud()

    @classmethod
    async def option_job_service(cls, auth: AuthSchema, id: int, option: int) -> None:
        # 1: 暂停 2: 恢复 3: 重启
        obj = await JobCRUD(auth).get_obj_by_id_crud(id=id)
        if not obj:
            raise CustomException(msg='操作失败，该数据定时任务不存在')
        if option == 1:
            SchedulerUtil().pause_job(job_id=id)
            await JobCRUD(auth).set_obj_field_crud(ids=[id], status=False)
        elif option == 2:
            SchedulerUtil().resume_job(job_id=id)
            await JobCRUD(auth).set_obj_field_crud(ids=[id], status=True)
        # elif option == 3:
        #     SchedulerUtil().reschedule_job(job_id=id)
        #     await JobCRUD(auth).set_obj_field_crud(ids=[id], status=False)

    @classmethod
    async def export_job_service(cls, data_list: List[Dict[str, Any]]) -> bytes:
        """导出公告列表"""
        mapping_dict = {
            'id': '编号',
            'name': '任务名称',
            'func': '任务函数',
            'trigger': '触发器',
            'args': '位置参数',
            'kwargs': '关键字参数',
            'coalesce': '是否合并运行',
            'max_instances': '最大实例数',
            'jobstore': '任务存储',
            'executor': '任务执行器',
            'trigger_args': '触发器参数',
            'status': '任务状态',
            'message': '日志信息',
            'description': '备注',
            'created_at': '创建时间',
            'updated_at': '更新时间',
            'creator_id': '创建者ID',
            'creator': '创建者',
        }

        # 复制数据并转换状态
        data = data_list.copy()
        for item in data:
            item['status'] = '已完成' if item['status'] == 0 else '运行中' if item['status'] == 1 else '暂停'
            item['creator'] = item.get('creator', {}).get('name', '未知') if isinstance(item.get('creator'), dict) else '未知'

        return ExcelUtil.export_list2excel(list_data=data, mapping_dict=mapping_dict)


class JobLogService:
    """
    定时任务日志管理模块服务层
    """
    
    @classmethod
    async def get_job_log_detail_service(cls, auth: AuthSchema, id: int) -> Dict:
        """获取定时任务日志详情"""
        obj = await JobLogCRUD(auth).get_obj_log_by_id_crud(id=id)
        return JobLogOutSchema.model_validate(obj).model_dump()
    
    @classmethod
    async def get_job_log_list_service(cls, auth: AuthSchema, search: Optional[JobLogQueryParam] = None, order_by: Optional[List[Dict[str, str]]] = None) -> List[Dict]:
        """获取定时任务日志列表"""
        obj_list = await JobLogCRUD(auth).get_obj_log_list_crud(search=search.__dict__, order_by=order_by)
        return [JobLogOutSchema.model_validate(obj).model_dump() for obj in obj_list]
    
    @classmethod
    async def delete_job_log_service(cls, auth: AuthSchema, ids: list[int]) -> None:
        """删除定时任务日志"""
        if len(ids) < 1:
            raise CustomException(msg='删除失败，删除对象不能为空')
        for id in ids:
            exist_obj = await JobLogCRUD(auth).get_obj_log_by_id_crud(id=id)
            if not exist_obj:
                raise CustomException(msg='删除失败，该定时任务日志不存在')
        await JobLogCRUD(auth).delete_obj_log_crud(ids=ids)
    
    @classmethod
    async def clear_job_log_service(cls, auth: AuthSchema) -> None:
        """清空定时任务日志"""
        # 获取所有日志ID并批量删除
        all_logs = await JobLogCRUD(auth).get_obj_log_list_crud()
        if all_logs:
            ids = [log.id for log in all_logs]
            await JobLogCRUD(auth).delete_obj_log_crud(ids=ids)

    @classmethod
    async def export_job_log_service(cls, data_list: List[Dict[str, Any]]) -> bytes:
        """导出定时任务日志列表"""
        mapping_dict = {
            'id': '编号',
            'job_name': '任务名称',
            'job_group': '任务组名',
            'job_executor': '任务执行器',
            'invoke_target': '调用目标字符串',
            'job_args': '位置参数',
            'job_kwargs': '关键字参数',
            'job_trigger': '任务触发器',
            'job_message': '日志信息',
            'exception_info': '异常信息',
            'status': '执行状态',
            'create_time': '创建时间',
        }

        # 复制数据并转换状态
        data = data_list.copy()
        for item in data:
            item['status'] = '成功' if item.get('status') else '失败'

        return ExcelUtil.export_list2excel(list_data=data, mapping_dict=mapping_dict)