diskann-disk 0.52.0

DiskANN is a fast approximate nearest neighbor search library for high dimensional data
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
51

/*
 * Copyright (c) Microsoft Corporation.
 * Licensed under the MIT license.
 */

use std::sync::Arc;

use diskann::ANNResult;
use diskann_providers::storage::VirtualStorageProvider;
use vfs::{FileSystem, MemoryFS};

use super::{traits::AlignedReaderFactory, StorageProviderAlignedFileReader};

pub struct VirtualAlignedReaderFactory<P: FileSystem = MemoryFS> {
    pub file_path: String,
    // Use Arc instead of reference because async searcher interfaces require 'static bounds
    // for proper lifetime management in async futures
    pub storage_provider: Arc<VirtualStorageProvider<P>>,
}

impl<P: FileSystem> AlignedReaderFactory for VirtualAlignedReaderFactory<P> {
    type AlignedReaderType = StorageProviderAlignedFileReader;

    fn build(&self) -> ANNResult<Self::AlignedReaderType> {
        StorageProviderAlignedFileReader::new(&*self.storage_provider, self.file_path.as_str())
    }
}

impl<P: FileSystem> VirtualAlignedReaderFactory<P> {
    pub fn new(file_path: String, storage_provider: Arc<VirtualStorageProvider<P>>) -> Self {
        Self {
            file_path,
            storage_provider,
        }
    }
}

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

    #[test]
    fn test_virtual_aligned_reader_factory_new() {
        let fs = Arc::new(VirtualStorageProvider::new_memory());
        let path = "/test.bin".to_string();
        let factory = VirtualAlignedReaderFactory::new(path.clone(), fs.clone());

        assert_eq!(factory.file_path, path);
        assert!(Arc::ptr_eq(&factory.storage_provider, &fs));
    }
}