Reorganization

1 files changed, 60 insertions, 0 deletions

diff --git a/src/rwlock/write_lock.rs b/src/rwlock/write_lock.rs
new file mode 100644
index 0000000..dd204f5
--- /dev/null
+++ b/src/rwlock/write_lock.rs
@@ -0,0 +1,60 @@
+use std::fmt::Debug;
+
+use lock_api::RawRwLock;
+
+use crate::key::Keyable;
+
+use super::{RwLock, RwLockWriteGuard, RwLockWriteRef, WriteLock};
+
+impl<'a, T: ?Sized, R> Debug for WriteLock<'a, T, R> {
+	fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+		f.write_str(&format!("WriteLock<{}>", std::any::type_name::<T>()))
+	}
+}
+
+impl<'a, T: ?Sized, R> From<&'a RwLock<T, R>> for WriteLock<'a, T, R> {
+	fn from(value: &'a RwLock<T, R>) -> Self {
+		Self::new(value)
+	}
+}
+
+impl<'a, T: ?Sized, R> AsRef<RwLock<T, R>> for WriteLock<'a, T, R> {
+	fn as_ref(&self) -> &RwLock<T, R> {
+		self.0
+	}
+}
+
+impl<'a, T: ?Sized, R> WriteLock<'a, T, R> {
+	#[must_use]
+	pub const fn new(rwlock: &'a RwLock<T, R>) -> Self {
+		Self(rwlock)
+	}
+}
+
+impl<'a, T: ?Sized, R: RawRwLock> WriteLock<'a, T, R> {
+	pub fn lock<'s, 'key: 's, Key: Keyable + 'key>(
+		&'s self,
+		key: Key,
+	) -> RwLockWriteGuard<'_, 'key, T, Key, R> {
+		self.0.write(key)
+	}
+
+	pub(crate) unsafe fn lock_no_key(&self) -> RwLockWriteRef<'_, T, R> {
+		self.0.write_no_key()
+	}
+
+	pub fn try_lock<'s, 'key: 's, Key: Keyable + 'key>(
+		&'s self,
+		key: Key,
+	) -> Option<RwLockWriteGuard<'_, 'key, T, Key, R>> {
+		self.0.try_write(key)
+	}
+
+	pub(crate) unsafe fn try_lock_no_key(&self) -> Option<RwLockWriteRef<'_, T, R>> {
+		self.0.try_write_no_key()
+	}
+
+	pub fn unlock<'key, Key: Keyable + 'key>(guard: RwLockWriteGuard<'_, 'key, T, Key, R>) -> Key {
+		RwLock::unlock_write(guard)
+	}
+}