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}