satoridb 0.1.2

Embedded vector database for approximate nearest neighbor search (experimental).
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

use futures::lock::Mutex;
use parking_lot::Mutex as MapMutex;
use std::collections::HashMap;
use std::sync::Arc;

/// Shared per-bucket locks used to serialize worker writes and rebalancer splits/merges.
pub struct BucketLocks {
    map: MapMutex<HashMap<u64, Arc<Mutex<()>>>>,
}

impl BucketLocks {
    pub fn new() -> Self {
        Self {
            map: MapMutex::new(HashMap::new()),
        }
    }

    /// Get (or create) the lock for a bucket id.
    pub fn lock_for(&self, bucket_id: u64) -> Arc<Mutex<()>> {
        let mut guard = self.map.lock();
        guard
            .entry(bucket_id)
            .or_insert_with(|| Arc::new(Mutex::new(())))
            .clone()
    }
}

impl Default for BucketLocks {
    fn default() -> Self {
        Self::new()
    }
}