zeppelin-embed 0.6.0

In-process vector, lexical, and hybrid search for macOS and Apple silicon
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

//! Temporary mutation lookup tables and test-only logical work accounting.
use std::collections::HashSet;
use std::hash::Hash;

pub(super) type LookupSet<T> = HashSet<T>;

pub(super) fn set<T: Copy + Eq + Hash>(values: &[T]) -> LookupSet<T> {
    work(values.len());
    values.iter().copied().collect()
}

pub(super) fn contains<T: Eq + Hash>(values: &LookupSet<T>, value: &T) -> bool {
    work(1);
    values.contains(value)
}

#[inline]
pub(super) fn work(amount: usize) {
    #[cfg(test)]
    WORK.with(|count| count.set(count.get() + amount));
    #[cfg(not(test))]
    let _ = amount;
}

#[cfg(test)]
thread_local! {
    static WORK: std::cell::Cell<usize> = const { std::cell::Cell::new(0) };
}

#[cfg(test)]
pub(super) fn take_work() -> usize {
    WORK.with(|count| count.replace(0))
}