sciforge 0.0.3

A comprehensive scientific computing library in pure Rust with zero dependencies
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82

use crate::hub::domain::common::errors::HubResult;

/// Task execution status.
#[derive(Debug, Clone, PartialEq, Eq)]
pub enum TaskStatus {
    Pending,
    Running,
    Completed,
    Failed,
}

/// Result of a single task execution.
pub struct TaskResult {
    /// Unique task identifier.
    pub task_id: String,
    /// Final status.
    pub status: TaskStatus,
    /// Output string on success.
    pub output: Option<String>,
    /// Error message on failure.
    pub error: Option<String>,
    /// Wall-clock time in milliseconds.
    pub elapsed_ms: f64,
}

/// Runs tasks with optional retry logic.
pub struct Executor {
    max_retries: usize,
}

impl Executor {
    /// Creates an executor with no retries.
    pub fn new() -> Self {
        Self { max_retries: 0 }
    }

    /// Sets the maximum number of retries on failure.
    pub fn with_retries(mut self, retries: usize) -> Self {
        self.max_retries = retries;
        self
    }

    /// Executes a closure as a named task, retrying on failure.
    pub fn execute<F>(&self, task_id: &str, f: F) -> TaskResult
    where
        F: Fn() -> HubResult<String>,
    {
        let start = std::time::Instant::now();
        let mut last_err = None;

        for _ in 0..=self.max_retries {
            match f() {
                Ok(output) => {
                    return TaskResult {
                        task_id: task_id.to_string(),
                        status: TaskStatus::Completed,
                        output: Some(output),
                        error: None,
                        elapsed_ms: start.elapsed().as_secs_f64() * 1000.0,
                    };
                }
                Err(e) => {
                    last_err = Some(e);
                }
            }
        }

        TaskResult {
            task_id: task_id.to_string(),
            status: TaskStatus::Failed,
            output: None,
            error: last_err.map(|e| e.to_string()),
            elapsed_ms: start.elapsed().as_secs_f64() * 1000.0,
        }
    }
}

impl Default for Executor {
    fn default() -> Self {
        Self::new()
    }
}