mscheduler 0.3.1

use mongodb::bson::DateTime;
use serde::{Deserialize, Serialize};

/// 记录worker的一次执行, 便于记录失败次数和简化状态流转
/// 约束: 同一个worker不能有两个RUNNING状态的info, 用running_id区分和更新不同的info
/// running_id由worker确保唯一,不在其他地方记录
#[derive(Deserialize, Serialize)]
pub struct TaskWorkerRunningInfo<K> {
    // use to identify running info
    pub running_id: Option<String>,
    // worker identifier
    pub worker_id: String,
    // when can other workers accept this task
    pub ping_expire_time: Option<DateTime>,
    // when did this task success
    pub success_time: Option<DateTime>,
    // when did this task fail
    pub fail_time: Option<DateTime>,
    // fail reason
    pub fail_reason: Option<String>,
    // task return values
    pub returns: Option<K>,
}

/// record task state info, including all worker states and some generic task info
#[derive(Deserialize, Serialize)]
pub struct TaskState<K> {
    // when did this task created
    pub create_time: DateTime,
    // when should this task run
    pub start_time: DateTime,
    // worker ids of which are running this task
    pub worker_states: Vec<TaskWorkerRunningInfo<K>>,
}

/// record how task sender specify how to run this task
#[derive(Deserialize, Serialize)]
pub struct TaskOption {
    // priority
    pub priority: u32,
    // how many workers are allowed to run this task concurrently
    pub concurrent_worker_cnt: u32,
    // the interval to update worker running state
    pub ping_interval_ms: u32,
    // after which time other worker can replace timeout worker
    pub worker_timeout_ms: u32,
    // only allow these workers to run this task
    pub specific_worker_ids: Vec<String>,
    // how many unexpected error are allowed to occur
    pub max_unexpected_retries: u32,
    // retry delay when unexpected error is occurred
    pub unexpected_retry_delay_ms: u32,
}

#[derive(Deserialize, Serialize)]
pub struct Task<T, K> {
    // identifier
    pub key: String,
    // record task running state
    pub task_state: TaskState<K>,
    // task running options
    pub task_option: TaskOption,
    // task parameters
    pub params: Option<T>,
}

/// use this struct to find problematic tasks and mark them as failed
#[derive(Deserialize, Serialize)]
pub struct TaskBasicInfo<K> {
    pub key: String,
    pub task_option: Option<TaskOption>,
    pub task_state: Option<TaskState<K>>,
}