kannolo 0.3.1

kANNolo is designed for easy prototyping of ANN Search algorithms while ensuring high effectiveness and efficiency over both dense and sparse vectors.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

use std::{fs, io::Error, path::PathBuf};

use serde::{de::DeserializeOwned, Serialize};

pub struct IndexSerializer {
    // We could have a `mode` field here that determines which serializer is used.
}

impl IndexSerializer {
    pub fn save_index<T: Serialize>(filename: &str, index: &T) -> Result<(), Error> {
        let filepath = PathBuf::from(filename);
        let serialized = bincode::serialize(&index).unwrap();
        fs::write(filepath, serialized)
    }

    pub fn load_index<T: DeserializeOwned>(filename: &str) -> T {
        let filepath = PathBuf::from(filename);
        let serialized: Vec<u8> = fs::read(filepath).unwrap();
        bincode::deserialize::<T>(&serialized).unwrap()
    }
}