fd_lock/sys/unix/
rw_lock.rs

1use rustix::fd::AsFd;
2use rustix::fs::FlockOperation;
3use std::io::{self, Error, ErrorKind};
4
5use super::{compatible_unix_lock, RwLockReadGuard, RwLockWriteGuard};
6
7#[derive(Debug)]
8pub struct RwLock<T: AsFd> {
9    pub(crate) inner: T,
10}
11
12impl<T: AsFd> RwLock<T> {
13    #[inline]
14    pub fn new(inner: T) -> Self {
15        RwLock { inner }
16    }
17
18    #[inline]
19    pub fn write(&mut self) -> io::Result<RwLockWriteGuard<'_, T>> {
20        compatible_unix_lock(self.inner.as_fd(), FlockOperation::LockExclusive)?;
21        Ok(RwLockWriteGuard::new(self))
22    }
23
24    #[inline]
25    pub fn try_write(&mut self) -> Result<RwLockWriteGuard<'_, T>, Error> {
26        compatible_unix_lock(self.inner.as_fd(), FlockOperation::NonBlockingLockExclusive)
27            .map_err(|err| match err.kind() {
28                ErrorKind::AlreadyExists => ErrorKind::WouldBlock.into(),
29                _ => Error::from(err),
30            })?;
31        Ok(RwLockWriteGuard::new(self))
32    }
33
34    #[inline]
35    pub fn read(&self) -> io::Result<RwLockReadGuard<'_, T>> {
36        compatible_unix_lock(self.inner.as_fd(), FlockOperation::LockShared)?;
37        Ok(RwLockReadGuard::new(self))
38    }
39
40    #[inline]
41    pub fn try_read(&self) -> Result<RwLockReadGuard<'_, T>, Error> {
42        compatible_unix_lock(self.inner.as_fd(), FlockOperation::NonBlockingLockShared).map_err(
43            |err| match err.kind() {
44                ErrorKind::AlreadyExists => ErrorKind::WouldBlock.into(),
45                _ => Error::from(err),
46            },
47        )?;
48        Ok(RwLockReadGuard::new(self))
49    }
50
51    #[inline]
52    pub fn into_inner(self) -> T
53    where
54        T: Sized,
55    {
56        self.inner
57    }
58}
fd_lock/sys/unix/rw_lock.rs

fd_lock/sys/unix/
rw_lock.rs
