hd-crane-gs/backend/app/api/v1/module_application/job/model.py

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

from sqlalchemy import Boolean, String, Integer, Text, ForeignKey
from sqlalchemy.orm import Mapped, mapped_column, relationship

from app.core.base_model import ModelMixin, UserMixin


class JobModel(ModelMixin, UserMixin):
    """
    定时任务调度表
    - 0: 运行中
    - 1: 暂停中
    """
    __tablename__: str = 'app_job'
    __table_args__: dict[str, str] = ({'comment': '定时任务调度表'})
    __loader_options__: list[str] = ["job_logs", "created_by", "updated_by"]

    name: Mapped[str | None] = mapped_column(String(64), nullable=True, default='', comment='任务名称')
    jobstore: Mapped[str | None] = mapped_column(String(64), nullable=True, default='default', comment='存储器')
    executor: Mapped[str | None] = mapped_column(String(64), nullable=True, default='default', comment='执行器:将运行此作业的执行程序的名称')
    trigger: Mapped[str] = mapped_column(String(64), nullable=False, comment='触发器:控制此作业计划的 trigger 对象')
    trigger_args: Mapped[str | None] = mapped_column(Text, nullable=True, comment='触发器参数')
    func: Mapped[str] = mapped_column(Text, nullable=False, comment='任务函数')
    args: Mapped[str | None] = mapped_column(Text, nullable=True, comment='位置参数')
    kwargs: Mapped[str | None] = mapped_column(Text, nullable=True, comment='关键字参数')
    coalesce: Mapped[bool] = mapped_column(Boolean, nullable=True, default=False, comment='是否合并运行:是否在多个运行时间到期时仅运行作业一次')
    max_instances: Mapped[int] = mapped_column(Integer, nullable=True, default=1, comment='最大实例数:允许的最大并发执行实例数')
    start_date: Mapped[str | None] = mapped_column(String(64), nullable=True, comment='开始时间')
    end_date: Mapped[str | None] = mapped_column(String(64), nullable=True, comment='结束时间')
    
    # 关联关系
    job_logs: Mapped[list['JobLogModel'] | None] = relationship(
        back_populates="job", 
        lazy="selectin"
    )


class JobLogModel(ModelMixin):
    """
    定时任务调度日志表
    """
    __tablename__: str = 'app_job_log'
    __table_args__: dict[str, str] = ({'comment': '定时任务调度日志表'})
    __loader_options__: list[str] = ["job"]

    job_name: Mapped[str] = mapped_column(String(64), nullable=False, comment='任务名称')
    job_group: Mapped[str] = mapped_column(String(64), nullable=False, comment='任务组名')
    job_executor: Mapped[str] = mapped_column(String(64), nullable=False, comment='任务执行器')
    invoke_target: Mapped[str] = mapped_column(String(500), nullable=False, comment='调用目标字符串')
    job_args: Mapped[str | None] = mapped_column(String(255), nullable=True, default='', comment='位置参数')
    job_kwargs: Mapped[str | None] = mapped_column(String(255), nullable=True, default='', comment='关键字参数')
    job_trigger: Mapped[str | None] = mapped_column(String(255), nullable=True, default='', comment='任务触发器')
    job_message: Mapped[str | None] = mapped_column(String(500), nullable=True, default='', comment='日志信息')
    exception_info: Mapped[str | None] = mapped_column(String(2000), nullable=True, default='', comment='异常信息')
    
    # 任务关联
    job_id: Mapped[int | None] = mapped_column(
        ForeignKey('app_job.id', ondelete="CASCADE"), 
        nullable=True,
        index=True,
        comment='任务ID'
    )
    
    job: Mapped["JobModel | None"] = relationship(
        back_populates="job_logs", 
        lazy="selectin"
    )