-
-
Notifications
You must be signed in to change notification settings - Fork 15
/
serial_file_lock.rs
76 lines (67 loc) · 1.82 KB
/
serial_file_lock.rs
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
65
66
67
68
69
70
71
72
73
74
75
76
use crate::file_lock::make_lock_for_name_and_path;
#[doc(hidden)]
pub fn fs_serial_core(name: &str, path: Option<&str>, function: fn()) {
let mut lock = make_lock_for_name_and_path(name, path);
lock.start_serial();
function();
lock.end_serial();
}
#[doc(hidden)]
pub fn fs_serial_core_with_return<E>(
name: &str,
path: Option<&str>,
function: fn() -> Result<(), E>,
) -> Result<(), E> {
let mut lock = make_lock_for_name_and_path(name, path);
lock.start_serial();
let ret = function();
lock.end_serial();
ret
}
#[doc(hidden)]
#[cfg(feature = "async")]
pub async fn fs_async_serial_core_with_return<E>(
name: &str,
path: Option<&str>,
fut: impl std::future::Future<Output = Result<(), E>>,
) -> Result<(), E> {
let mut lock = make_lock_for_name_and_path(name, path);
lock.start_serial();
let ret = fut.await;
lock.end_serial();
ret
}
#[doc(hidden)]
#[cfg(feature = "async")]
pub async fn fs_async_serial_core(
name: &str,
path: Option<&str>,
fut: impl std::future::Future<Output = ()>,
) {
let mut lock = make_lock_for_name_and_path(name, path);
lock.start_serial();
fut.await;
lock.end_serial();
}
#[cfg(test)]
mod tests {
use std::panic;
use fslock::LockFile;
use super::fs_serial_core;
use crate::file_lock::path_for_name;
#[test]
fn test_serial() {
fs_serial_core("test", None, || {});
}
#[test]
fn unlock_on_assert_sync_without_return() {
let lock_path = path_for_name("unlock_on_assert_sync_without_return");
let _ = panic::catch_unwind(|| {
fs_serial_core("foo", Some(&lock_path), || {
assert!(false);
})
});
let mut lockfile = LockFile::open(&lock_path).unwrap();
assert!(lockfile.try_lock().unwrap());
}
}