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

use crate::error::CSDError;
use crate::exec::call_ceph;

use std::fmt::Debug;
use std::fs::File;
use std::io::prelude::*;

use serde::de::DeserializeOwned;

// Generic trait to read file to serializable struct
pub trait FromFile<T> {
    fn from_file(path: &str) -> Result<T, CSDError>;
}

impl<T: DeserializeOwned + Debug> FromFile<T> for T {
    fn from_file(path: &str) -> Result<T, CSDError> {
        let mut file = File::open(path)?;
        let mut buffer = String::new();
        file.read_to_string(&mut buffer)?;
        Ok(serde_json::from_str(&buffer)?)
    }
}

// Generic trait to create structs from ceph JSON output. Since most if not all
// of of ceph's commands can be formatted to JSON. For example:
// let pgmap = PGMap::from_ceph("pg dump").unwrap()
pub trait FromCeph<T> {
    fn from_ceph(cmd: &str) -> Result<T, CSDError>;
}

impl<T: DeserializeOwned + Debug> FromCeph<T> for T {
    fn from_ceph(cmd: &str) -> Result<T, CSDError> {
        let ceph_output = call_ceph(cmd)?;
        let serde_res: Result<T, serde_json::Error> = serde_json::from_str(&ceph_output);
        trace!("deserialize ceph: {:?}", serde_res);

        Ok(serde_res?)
    }
}