From 32f972a26a0066291873445088718deec3ed4233 Mon Sep 17 00:00:00 2001
From: Mica White <botahamec@gmail.com>
Date: Tue, 16 Jul 2024 17:18:33 -0400
Subject: Impl Lock for LockCollection

---
 src/collection/owned.rs | 40 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 40 insertions(+)

(limited to 'src/collection/owned.rs')

diff --git a/src/collection/owned.rs b/src/collection/owned.rs
index d180ed2..2b6e974 100644
--- a/src/collection/owned.rs
+++ b/src/collection/owned.rs
@@ -11,6 +11,46 @@ fn get_locks<L: Lockable>(data: &L) -> Vec<&dyn RawLock> {
 	locks
 }
 
+unsafe impl<L: Lockable + Send + Sync> RawLock for OwnedLockCollection<L> {
+	unsafe fn lock(&self) {
+		let locks = get_locks(&self.data);
+		for lock in locks {
+			lock.lock();
+		}
+	}
+
+	unsafe fn try_lock(&self) -> bool {
+		let locks = get_locks(&self.data);
+		utils::ordered_try_lock(&locks)
+	}
+
+	unsafe fn unlock(&self) {
+		let locks = get_locks(&self.data);
+		for lock in locks {
+			lock.unlock();
+		}
+	}
+
+	unsafe fn read(&self) {
+		let locks = get_locks(&self.data);
+		for lock in locks {
+			lock.read();
+		}
+	}
+
+	unsafe fn try_read(&self) -> bool {
+		let locks = get_locks(&self.data);
+		utils::ordered_try_read(&locks)
+	}
+
+	unsafe fn unlock_read(&self) {
+		let locks = get_locks(&self.data);
+		for lock in locks {
+			lock.unlock_read();
+		}
+	}
+}
+
 unsafe impl<L: Lockable> Lockable for OwnedLockCollection<L> {
 	type Guard<'g> = L::Guard<'g> where Self: 'g;
 
-- 
cgit v1.2.3