Skip to main content

nano_wal/
error.rs

1use std::fmt;
2use std::io;
3
4#[derive(Debug)]
5pub enum WalError {
6    Io(io::Error),
7    InvalidConfig(String),
8    CorruptedData(String),
9    HeaderTooLarge { size: usize, max: usize },
10    Shutdown,
11}
12
13impl fmt::Display for WalError {
14    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
15        match self {
16            WalError::Io(e) => write!(f, "I/O error: {}", e),
17            WalError::InvalidConfig(msg) => write!(f, "Invalid configuration: {}", msg),
18            WalError::CorruptedData(msg) => write!(f, "Data corruption: {}", msg),
19            WalError::HeaderTooLarge { size, max } => {
20                write!(f, "Header size {} exceeds maximum {}", size, max)
21            }
22            WalError::Shutdown => write!(f, "WAL has been shut down"),
23        }
24    }
25}
26
27impl std::error::Error for WalError {
28    fn source(&self) -> Option<&(dyn std::error::Error + 'static)> {
29        match self {
30            WalError::Io(e) => Some(e),
31            _ => None,
32        }
33    }
34}
35
36impl From<io::Error> for WalError {
37    fn from(e: io::Error) -> Self {
38        WalError::Io(e)
39    }
40}
41
42pub type Result<T> = std::result::Result<T, WalError>;
43
44#[cfg(test)]
45mod tests {
46    use super::*;
47
48    #[test]
49    fn test_wal_error_display() {
50        let err = WalError::InvalidConfig("bad".into());
51        assert_eq!(format!("{}", err), "Invalid configuration: bad");
52
53        let err = WalError::Shutdown;
54        assert_eq!(format!("{}", err), "WAL has been shut down");
55
56        let err = WalError::HeaderTooLarge { size: 70000, max: 65535 };
57        assert_eq!(format!("{}", err), "Header size 70000 exceeds maximum 65535");
58
59        let err = WalError::CorruptedData("bad magic".into());
60        assert_eq!(format!("{}", err), "Data corruption: bad magic");
61    }
62
63    #[test]
64    fn test_wal_error_from_io() {
65        let io_err = std::io::Error::new(std::io::ErrorKind::NotFound, "gone");
66        let wal_err: WalError = io_err.into();
67        assert!(matches!(wal_err, WalError::Io(_)));
68    }
69}