foundationdb/recipes/ranked_register/
errors.rs

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
// Copyright 2024 foundationdb-rs developers
//
// Licensed under the Apache License, Version 2.0, <LICENSE-APACHE or
// http://apache.org/licenses/LICENSE-2.0> or the MIT license <LICENSE-MIT or
// http://opensource.org/licenses/MIT>, at your option. This file may not be
// copied, modified, or distributed except according to those terms.

//! Error types for the ranked register

use crate::tuple::PackError;
use crate::{FdbBindingError, FdbError, RetryableError};
use std::fmt;

/// Ranked register specific errors
#[derive(Debug)]
pub enum RankedRegisterError {
    /// Database error
    Fdb(FdbError),
    /// Retry loop error
    Binding(FdbBindingError),
    /// Serialization error
    PackError(PackError),
    /// Invalid state encountered in the register
    InvalidState(String),
}

impl fmt::Display for RankedRegisterError {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        match self {
            Self::Fdb(e) => write!(f, "Database error: {e}"),
            Self::Binding(e) => write!(f, "Retry loop error: {e}"),
            Self::PackError(e) => write!(f, "Pack error: {e:?}"),
            Self::InvalidState(msg) => write!(f, "Invalid state: {msg}"),
        }
    }
}

impl std::error::Error for RankedRegisterError {
    fn source(&self) -> Option<&(dyn std::error::Error + 'static)> {
        match self {
            Self::Fdb(e) => Some(e),
            Self::Binding(e) => Some(e),
            Self::PackError(e) => Some(e),
            Self::InvalidState(_) => None,
        }
    }
}

impl From<FdbError> for RankedRegisterError {
    fn from(error: FdbError) -> Self {
        Self::Fdb(error)
    }
}

impl From<FdbBindingError> for RankedRegisterError {
    fn from(error: FdbBindingError) -> Self {
        Self::Binding(error)
    }
}

impl From<PackError> for RankedRegisterError {
    fn from(error: PackError) -> Self {
        Self::PackError(error)
    }
}

/// The `source()` chain exposes the wrapped `FdbError`, so the default
/// `retry_decision` makes this error retry-transparent in `db.run`.
impl RetryableError for RankedRegisterError {}

/// Result type for ranked register operations
pub type Result<T> = std::result::Result<T, RankedRegisterError>;