PayrollManagementSystem/backend/app/models/tplus_automation.py

"""自动采集的周期、重试和进程心跳；不承载业务核算状态。"""

from datetime import datetime

from sqlalchemy import DateTime, Integer, String, UniqueConstraint
from sqlalchemy.orm import Mapped, mapped_column

from app.models.base import BIGINT_UNSIGNED, TINYINT_TYPE, Base, TimestampMixin, mysql_table_args


class TPlusAutomationConfig(Base, TimestampMixin):
    """页面保存的源级开关，与长任务心跳表分开，切换不争用采集状态行。"""

    __tablename__ = "tplus_automation_configs"
    __table_args__ = mysql_table_args(
        "T+自动采集设置（不含连接凭据）",
        UniqueConstraint("source_key", name="uk_tplus_automation_configs_source_key"),
    )
    id: Mapped[int] = mapped_column(BIGINT_UNSIGNED, primary_key=True, autoincrement=True, comment="主键ID")
    source_key: Mapped[str] = mapped_column(String(64), comment="源连接身份摘要")
    source_database: Mapped[str] = mapped_column(String(128), comment="源账套")
    enabled: Mapped[int] = mapped_column(TINYINT_TYPE, comment="自动采集是否开启：1开启0关闭")
    version: Mapped[int] = mapped_column(Integer, comment="设置版本号，防止并发覆盖")
    updated_by: Mapped[int] = mapped_column(BIGINT_UNSIGNED, comment="最近修改账号ID")


class TPlusRuntimeControl(Base, TimestampMixin):
    """单部署的服务期望状态与托管进程报告，默认不启动。"""

    __tablename__ = "tplus_runtime_controls"
    __table_args__ = mysql_table_args("T+读取服务运行控制")
    id: Mapped[int] = mapped_column(BIGINT_UNSIGNED, primary_key=True, autoincrement=False, comment="主键ID")
    desired_state: Mapped[str] = mapped_column(String(16), server_default="STOPPED", comment="期望状态：RUNNING/STOPPED")
    actual_state: Mapped[str] = mapped_column(String(16), server_default="STOPPED", comment="实际状态：RUNNING/STOPPED/STOPPING/DEGRADED/OFFLINE")
    worker_phase: Mapped[str] = mapped_column(String(16), server_default="STOPPED", comment="读取服务阶段")
    heartbeat_at: Mapped[datetime | None] = mapped_column(DateTime, comment="最近心跳时间")
    last_error: Mapped[str | None] = mapped_column(String(255), comment="最近受控错误")
    host: Mapped[str | None] = mapped_column(String(128), comment="运行主机")
    platform: Mapped[str | None] = mapped_column(String(32), comment="运行平台")
    owner: Mapped[str | None] = mapped_column(String(36), comment="当前托管进程身份")
    version: Mapped[int] = mapped_column(Integer, server_default="0", comment="状态版本号")
    updated_by: Mapped[int | None] = mapped_column(BIGINT_UNSIGNED, comment="最近修改账号ID")


class TPlusAutomationState(Base, TimestampMixin):
    __tablename__ = "tplus_automation_states"
    __table_args__ = mysql_table_args("T+自动采集周期与进程状态（不自动导入正式业务）")
    id: Mapped[int] = mapped_column(BIGINT_UNSIGNED, primary_key=True, autoincrement=True, comment="主键ID")
    source_key: Mapped[str] = mapped_column(String(64), unique=True, comment="源连接身份摘要（不含凭据）")
    source_database: Mapped[str] = mapped_column(String(128), comment="源账套")
    scheduled_period: Mapped[str | None] = mapped_column(String(16), comment="最近已排队ISO周")
    root_batch_no: Mapped[str | None] = mapped_column(String(32), comment="本周期原始采集批次")
    current_batch_no: Mapped[str | None] = mapped_column(String(32), comment="当前原始或重试批次")
    retry_attempts: Mapped[int] = mapped_column(Integer, server_default="0", comment="本周期自动重试次数")
    next_retry_at: Mapped[datetime | None] = mapped_column(DateTime, comment="下一次自动重试时间")
    last_success_at: Mapped[datetime | None] = mapped_column(DateTime, comment="最近成功采集时间")
    maintenance_at: Mapped[datetime | None] = mapped_column(DateTime, comment="最近周期检查时间")
    heartbeat_at: Mapped[datetime | None] = mapped_column(DateTime, comment="独立进程心跳时间")
    worker_owner: Mapped[str | None] = mapped_column(String(36), comment="当前持锁进程身份")
    worker_phase: Mapped[str] = mapped_column(String(16), server_default="STOPPED", comment="进程状态：IDLE/RUNNING/STOPPED")