From 56fb681279b2f2221eef933617d521469c6e6d83 Mon Sep 17 00:00:00 2001
From: metamuffin <metamuffin@disroot.org>
Date: Thu, 6 Apr 2023 20:23:00 +0200
Subject: apply limits

---
 src/limiter.rs | 37 +++++++++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)
 create mode 100644 src/limiter.rs

(limited to 'src/limiter.rs')

diff --git a/src/limiter.rs b/src/limiter.rs
new file mode 100644
index 0000000..97bdb5c
--- /dev/null
+++ b/src/limiter.rs
@@ -0,0 +1,37 @@
+use std::sync::{atomic::AtomicUsize, Arc};
+
+pub struct Limiter {
+    limit: usize,
+    counter: Arc<AtomicUsize>,
+}
+
+impl Limiter {
+    pub fn new(limit: usize) -> Self {
+        Limiter {
+            counter: Default::default(),
+            limit,
+        }
+    }
+    pub fn obtain(&self) -> Option<LimitLock> {
+        let c = self
+            .counter
+            .fetch_add(1, std::sync::atomic::Ordering::Relaxed);
+        if c < self.limit {
+            Some(LimitLock {
+                counter: self.counter.clone(),
+            })
+        } else {
+            None
+        }
+    }
+}
+
+pub struct LimitLock {
+    counter: Arc<AtomicUsize>,
+}
+impl Drop for LimitLock {
+    fn drop(&mut self) {
+        self.counter
+            .fetch_sub(1, std::sync::atomic::Ordering::Relaxed);
+    }
+}
-- 
cgit v1.2.3-70-g09d2