diskann-platform 0.51.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

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

use std::{
    io,
    sync::{Mutex, MutexGuard},
};

use windows_sys::Win32::Foundation::HANDLE;

/// `ThreadSafeHandle` struct that wraps a native Windows `HANDLE` object with a mutex to ensure thread safety.
pub struct ThreadSafeHandle(Mutex<HANDLE>);

/// Implement `Send` and `Sync` for `ThreadSafeHandle` to allow it to be shared across threads.
unsafe impl Send for ThreadSafeHandle {}
unsafe impl Sync for ThreadSafeHandle {}

impl ThreadSafeHandle {
    /// Lock the mutex and return a guard to the handle.
    pub fn lock(&self) -> io::Result<MutexGuard<'_, HANDLE>> {
        self.0.lock().map_err(|_| {
            io::Error::new(
                io::ErrorKind::WouldBlock,
                "Unable to acquire lock on ThreadSafeHandle.",
            )
        })
    }

    /// Create a new `ThreadSafeHandle` from a native Windows `HANDLE`.
    pub fn new(handle: HANDLE) -> Self {
        Self(Mutex::new(handle))
    }
}