persistent_scheduler/core/
retry.rs

1use serde::{Deserialize, Serialize};
2
3#[derive(Clone, Copy, Debug, Eq, PartialEq, Deserialize, Serialize)]
4/// Retry strategy for jobs
5pub enum RetryStrategy {
6    /// Linear backoff in seconds
7    ///
8    /// For example, `RetryStrategy::Linear(5)` will retry a failed job 5 seconds after the previous attempt
9    Linear { interval: u32 },
10
11    /// Exponential backoff with a base interval in seconds
12    ///
13    /// For example, `RetryStrategy::Exponential(2)` will retry a failed job 2 seconds after the first failure,
14    /// 4 seconds after the second failure, and so on
15    Exponential { base: u32 },
16}
17
18#[derive(Clone, Copy, Debug, Eq, PartialEq, serde::Deserialize, serde::Serialize)]
19/// Retry policy to define how and when jobs should be retried
20pub struct RetryPolicy {
21    /// Strategy for retrying failed jobs
22    pub strategy: RetryStrategy,
23
24    /// Maximum number of retries or retry indefinitely
25    pub max_retries: Option<u32>, // `None` means infinite retries
26}
27
28impl RetryPolicy {
29    /// Create a new linear retry policy with a maximum number of retries
30    pub const fn linear(interval: u32, max_retries: Option<u32>) -> Self {
31        RetryPolicy {
32            strategy: RetryStrategy::Linear { interval },
33            max_retries,
34        }
35    }
36
37    /// Create a new exponential retry policy with a maximum number of retries
38    pub const fn exponential(base: u32, max_retries: Option<u32>) -> Self {
39        RetryPolicy {
40            strategy: RetryStrategy::Exponential { base },
41            max_retries,
42        }
43    }
44
45    pub fn wait_time(&self, retry_count: u32) -> u32 {
46        match self.strategy {
47            RetryStrategy::Linear { interval } => interval,
48            RetryStrategy::Exponential { base } => base.saturating_pow(retry_count),
49        }
50    }
51}
persistent_scheduler/core/retry.rs

persistent_scheduler/core/
retry.rs
