dataflow-rs 4.0.1

//! How a run and each workflow in it ended.

use crate::base::{Error, WorkflowIx};
use serde::ser::SerializeStruct;
use serde::{Serialize, Serializer};

/// How a run ended.
#[derive(Debug)]
#[non_exhaustive]
pub enum RunStatus {
    /// Every considered workflow finished and the message holds no errors.
    Completed,
    /// Every considered workflow finished, but the message holds error records.
    CompletedWithErrors,
    /// A workflow stopped the run with this error. A function's `Err` of
    /// kind `TimedOut` or `Cancelled`, such as a child run's propagated,
    /// ends the run here when its task's policies stop the run and it
    /// returned before the run's own deadline passed or its token was
    /// cancelled: only the run's own limits give the two statuses below.
    Stopped(Error),
    /// The run's deadline passed: at a check between ops or at a sweep, or
    /// while a call ran that it then cut short or that returned `Err`
    /// after it.
    TimedOut,
    /// The run's token, or one of its ancestors, was cancelled: before a
    /// check, or while a call ran that it then cut short or that returned
    /// `Err` after it.
    Cancelled,
}

/// Why a workflow did or did not run.
#[derive(Debug)]
#[non_exhaustive]
pub enum Selection {
    /// The workflow was selected and ran.
    Ran,
    /// Its condition was false.
    SkippedByCondition,
    /// Its condition failed to evaluate, which counts as false.
    ConditionFailed(Error),
    /// The message's routing bucket is outside its rollout.
    SkippedByRollout,
}

/// How a workflow that ran ended.
#[derive(Clone, Copy, Debug, PartialEq, Eq)]
#[non_exhaustive]
pub enum WorkflowExit {
    /// It ran to its end.
    Completed,
    /// A task or a halt ended it early, on purpose.
    Halted,
    /// A task failed and `on_failure: continue` stepped past it.
    Failed,
    /// It stopped the run.
    Stopped,
}

/// One workflow's part in a run.
#[derive(Debug)]
#[non_exhaustive]
pub struct WorkflowRecord<'r> {
    /// The workflow's id.
    pub id: &'r str,
    /// The workflow's position in the program.
    pub index: WorkflowIx,
    /// Whether it was selected.
    pub selection: &'r Selection,
    /// The sweeps begun: one for a workflow without a loop that ran, and for
    /// a loop each sweep its condition admitted.
    pub sweeps: u32,
    /// How it ended, if it ran.
    pub exit: Option<WorkflowExit>,
}

/// Writes the `status` string of a run's JSON: `completed`,
/// `completed_with_errors`, `stopped`, `timed_out` or `cancelled`. The
/// `error` of a stopped run is a key of the run's object, which
/// [`Run`](crate::Run) writes.
impl Serialize for RunStatus {
    fn serialize<S: Serializer>(&self, s: S) -> Result<S::Ok, S::Error> {
        s.serialize_str(match self {
            Self::Completed => "completed",
            Self::CompletedWithErrors => "completed_with_errors",
            Self::Stopped(_) => "stopped",
            Self::TimedOut => "timed_out",
            Self::Cancelled => "cancelled",
        })
    }
}

/// Writes the `selection` string of a workflow's record: `ran`,
/// `skipped_by_condition`, `condition_failed` or `skipped_by_rollout`. The
/// error of a failed condition is a key of the record, which
/// [`WorkflowRecord`] writes.
impl Serialize for Selection {
    fn serialize<S: Serializer>(&self, s: S) -> Result<S::Ok, S::Error> {
        s.serialize_str(match self {
            Self::Ran => "ran",
            Self::SkippedByCondition => "skipped_by_condition",
            Self::ConditionFailed(_) => "condition_failed",
            Self::SkippedByRollout => "skipped_by_rollout",
        })
    }
}

/// Writes how a workflow ended as a lowercase string: `completed`,
/// `halted`, `failed` or `stopped`.
impl Serialize for WorkflowExit {
    fn serialize<S: Serializer>(&self, s: S) -> Result<S::Ok, S::Error> {
        s.serialize_str(match self {
            Self::Completed => "completed",
            Self::Halted => "halted",
            Self::Failed => "failed",
            Self::Stopped => "stopped",
        })
    }
}

/// Writes a workflow's record as `{"id", "index", "selection", "error"?,
/// "sweeps", "exit"}`: `error` only for a condition that failed to
/// evaluate, and `exit` `null` for a workflow that did not run.
impl Serialize for WorkflowRecord<'_> {
    fn serialize<S: Serializer>(&self, s: S) -> Result<S::Ok, S::Error> {
        let error = match self.selection {
            Selection::ConditionFailed(e) => Some(e),
            _ => None,
        };
        let mut o = s.serialize_struct("WorkflowRecord", 6)?;
        o.serialize_field("id", self.id)?;
        o.serialize_field("index", &self.index)?;
        o.serialize_field("selection", self.selection)?;
        match error {
            Some(e) => o.serialize_field("error", e)?,
            None => o.skip_field("error")?,
        }
        o.serialize_field("sweeps", &self.sweeps)?;
        o.serialize_field("exit", &self.exit)?;
        o.end()
    }
}

#[cfg(feature = "ts")]
crate::base::ts::ts_decl!(
    RunStatus,
    "RunStatus",
    [" How a run ended: a stopped run's error is its `error`."],
    |_cfg| crate::base::ts::literals([
        "completed",
        "completed_with_errors",
        "stopped",
        "timed_out",
        "cancelled",
    ])
);

#[cfg(feature = "ts")]
crate::base::ts::ts_decl!(
    Selection,
    "Selection",
    [
        " Why a workflow did or did not run: a failed condition's error is its",
        " record's `error`.",
    ],
    |_cfg| crate::base::ts::literals([
        "ran",
        "skipped_by_condition",
        "condition_failed",
        "skipped_by_rollout",
    ])
);

#[cfg(feature = "ts")]
crate::base::ts::ts_decl!(
    WorkflowExit,
    "WorkflowExit",
    [" How a workflow that ran ended."],
    |_cfg| crate::base::ts::literals(["completed", "halted", "failed", "stopped"])
);

#[cfg(feature = "ts")]
crate::base::ts::ts_decl!(
    WorkflowRecord<'_>,
    "WorkflowRecord",
    [" One workflow's part in a run."],
    |cfg| {
        use crate::base::ts::{Prop, object};
        object(&[
            Prop::new("id", "string", &[" The workflow's id."]),
            Prop::new(
                "index",
                WorkflowIx::name(cfg),
                &[" The workflow's position in the program."],
            ),
            Prop::new(
                "selection",
                Selection::name(cfg),
                &[" Whether it was selected."],
            ),
            Prop::optional(
                "error",
                Error::name(cfg),
                &[
                    " Why its condition failed to evaluate; present only when `selection` is",
                    " `condition_failed`.",
                ],
            ),
            Prop::new(
                "sweeps",
                "number",
                &[
                    " The sweeps begun: one for a workflow without a loop that ran, and for",
                    " a loop each sweep its condition admitted.",
                ],
            ),
            Prop::new(
                "exit",
                Option::<WorkflowExit>::name(cfg),
                &[" How it ended; `null` when it did not run."],
            ),
        ])
    }
);