use serde::{Deserialize, Serialize};
use std::fmt;
use super::error::DomainError;
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, Serialize, Deserialize)]
#[serde(rename_all = "snake_case")]
pub enum TransferErrorKind {
Transient,
Permanent,
}
impl TransferErrorKind {
pub fn as_str(&self) -> &'static str {
match self {
Self::Transient => "transient",
Self::Permanent => "permanent",
}
}
}
impl fmt::Display for TransferErrorKind {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.write_str(self.as_str())
}
}
impl std::str::FromStr for TransferErrorKind {
type Err = DomainError;
fn from_str(s: &str) -> Result<Self, Self::Err> {
match s {
"transient" => Ok(Self::Transient),
"permanent" => Ok(Self::Permanent),
other => Err(DomainError::Validation {
field: "transfer_error_kind".into(),
reason: format!("unknown value: {other}"),
}),
}
}
}
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub struct RetryPolicy {
max_attempts: u32,
}
impl RetryPolicy {
pub const DEFAULT_MAX_ATTEMPTS: u32 = 3;
pub fn new(max_attempts: u32) -> Self {
Self {
max_attempts: max_attempts.max(1),
}
}
pub fn max_attempts(&self) -> u32 {
self.max_attempts
}
}
impl Default for RetryPolicy {
fn default() -> Self {
Self::new(Self::DEFAULT_MAX_ATTEMPTS)
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn default_policy() {
let p = RetryPolicy::default();
assert_eq!(p.max_attempts(), 3);
}
#[test]
fn custom_policy() {
let p = RetryPolicy::new(5);
assert_eq!(p.max_attempts(), 5);
}
#[test]
fn zero_clamped_to_one() {
let p = RetryPolicy::new(0);
assert_eq!(p.max_attempts(), 1);
}
#[test]
fn error_kind_roundtrip() {
for kind in [TransferErrorKind::Transient, TransferErrorKind::Permanent] {
let s = kind.as_str();
let parsed: TransferErrorKind = s.parse().unwrap();
assert_eq!(parsed, kind);
}
}
#[test]
fn error_kind_invalid() {
let result: Result<TransferErrorKind, _> = "unknown".parse();
assert!(result.is_err());
}
}