saku-storage 0.1.0

Shared storage infrastructure for the saku ecosystem
Documentation 
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

use std::path::Path;

use crate::error::IoError;

/// Atomically write content to a file by writing to a temp file first, then renaming.
pub fn atomic_write(path: &Path, content: &str) -> Result<(), IoError> {
    let unique_temp = format!("{}.tmp.{}", path.display(), uuid::Uuid::new_v4());
    let temp_path = std::path::PathBuf::from(&unique_temp);

    std::fs::write(&temp_path, content).map_err(|e| IoError::WriteFailed {
        path: temp_path.clone(),
        source: e,
    })?;

    std::fs::rename(&temp_path, path).map_err(|e| IoError::RenameFailed {
        from: temp_path,
        to: path.to_path_buf(),
        source: e,
    })?;

    Ok(())
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn writes_content_atomically() {
        let tmp = tempfile::tempdir().unwrap();
        let file_path = tmp.path().join("test.json");

        atomic_write(&file_path, r#"{"hello": "world"}"#).unwrap();

        let content = std::fs::read_to_string(&file_path).unwrap();
        assert_eq!(content, r#"{"hello": "world"}"#);
    }

    #[test]
    fn overwrites_existing_file() {
        let tmp = tempfile::tempdir().unwrap();
        let file_path = tmp.path().join("test.json");

        atomic_write(&file_path, "first").unwrap();
        atomic_write(&file_path, "second").unwrap();

        let content = std::fs::read_to_string(&file_path).unwrap();
        assert_eq!(content, "second");
    }
}