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

use redis::{ErrorKind, RedisError};
use sea_streamer_types::{StreamErr, StreamResult};
use thiserror::Error;

#[derive(Error, Debug, Clone)]
/// Different types of Redis errors.
pub enum RedisErr {
    #[error("Failed to parse message ID: {0}")]
    MessageId(String),
    #[error("Failed to parse StreamReadReply: {0:?}")]
    StreamReadReply(String),
    #[error("The Producer task died")]
    ProducerDied,
    #[error("Consumer died with unrecoverable error. Check the log for details.")]
    ConsumerDied,
    #[error("The server generated an invalid response: {0}")]
    ResponseError(String),
    #[error("The authentication with the server failed: {0}")]
    AuthenticationFailed(String),
    #[error("Operation failed because of a type mismatch: {0}")]
    TypeError(String),
    #[error("A script execution was aborted: {0}")]
    ExecAbortError(String),
    #[error("The server cannot response because it's loading a dump: {0}")]
    BusyLoadingError(String),
    #[error("A script that was requested does not actually exist: {0}")]
    NoScriptError(String),
    #[error("An error that was caused because the parameter to the client were wrong: {0}")]
    InvalidClientConfig(String),
    #[error("Raised if a key moved to a different node: {0}")]
    Moved(String),
    #[error("Raised if a key moved to a different node but we need to ask: {0}")]
    Ask(String),
    #[error("Raised if a request needs to be retried: {0}")]
    TryAgain(String),
    #[error("Raised if a redis cluster is down: {0}")]
    ClusterDown(String),
    #[error("A request spans multiple slots: {0}")]
    CrossSlot(String),
    #[error("A cluster master is unavailable: {0}")]
    MasterDown(String),
    #[error("IO error: {0}")]
    IoError(String),
    #[error("An error raised that was identified on the client before execution: {0}")]
    ClientError(String),
    #[error("Extension error: {0}")]
    ExtensionError(String),
    #[error("Attempt to write to a read-only server: {0}")]
    ReadOnly(String),
    #[error("Unknown error: {0}")]
    Unknown(String),
}

/// A type alias for convenience.
pub type RedisResult<T> = StreamResult<T, RedisErr>;

pub(crate) fn map_err(err: RedisError) -> StreamErr<RedisErr> {
    let e = format!("{err}");
    StreamErr::Backend(match err.kind() {
        ErrorKind::ResponseError => RedisErr::ResponseError(e),
        ErrorKind::AuthenticationFailed => RedisErr::AuthenticationFailed(e),
        ErrorKind::TypeError => RedisErr::TypeError(e),
        ErrorKind::ExecAbortError => RedisErr::ExecAbortError(e),
        ErrorKind::BusyLoadingError => RedisErr::BusyLoadingError(e),
        ErrorKind::NoScriptError => RedisErr::NoScriptError(e),
        ErrorKind::InvalidClientConfig => RedisErr::InvalidClientConfig(e),
        ErrorKind::Moved => RedisErr::Moved(e),
        ErrorKind::Ask => RedisErr::Ask(e),
        ErrorKind::TryAgain => RedisErr::TryAgain(e),
        ErrorKind::ClusterDown => RedisErr::ClusterDown(e),
        ErrorKind::CrossSlot => RedisErr::CrossSlot(e),
        ErrorKind::MasterDown => RedisErr::MasterDown(e),
        ErrorKind::IoError => RedisErr::IoError(e),
        ErrorKind::ClientError => RedisErr::ClientError(e),
        ErrorKind::ExtensionError => RedisErr::ExtensionError(e),
        ErrorKind::ReadOnly => RedisErr::ReadOnly(e),
        _ => RedisErr::Unknown(e),
    })
}