diskann-inmem 0.60.0

DiskANN3 is a composable library for bringing scalable, accurate and cost-effective vector indexing to multiple databases.
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
52
53
54
55
56
57
58
59
60
61
62
63
64

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

macro_rules! representation {
    ($T:ty) => {
        $crate::repr::internal::macros::representation!({} $T where);
    };
    ({ $($generics:ident),* $(,)? } $T:ty where $($where:tt)*) => {
        impl<$($generics),*> $crate::repr::Representation for $T
        where
            $($where)*
        {
            fn max_degree(&self) -> MaxDegree {
                self.store.neighbors().max_degree()
            }

            fn retire(&self, i: u32) -> ANNResult<()> {
                Ok(self.store.retire(i.into_usize())?)
            }

            fn is_readable(&self, i: u32) -> Option<bool> {
                self.store.can_read_approximate(i.into_usize())
            }

            fn id_limit(&self) -> IdLimit {
                self.store.id_limit()
            }

            fn capacity(&self) -> Capacity {
                self.store.capacity()
            }
        }
    };
}

macro_rules! set_guard {
    ($(#[$doc:meta])* for<$lt:lifetime> $exclusive:ty) => {
        $(#[$doc])*
        #[derive(Debug)]
        pub struct Guard<$lt> {
            slot: $crate::store::Exclusive<$lt, $exclusive>,
        }

        impl<$lt> Guard<$lt> {
            fn new(slot: $crate::store::Exclusive<$lt, $exclusive>) -> Self {
                Self { slot }
            }
        }

        impl $crate::repr::Guard for Guard<'_> {
            fn publish(self) {
                self.slot.publish();
            }
            fn id(&self) -> u32 {
                self.slot.slot()
            }
        }
    }
}

pub(in crate::repr) use representation;
pub(in crate::repr) use set_guard;