add Semaphore

osspkg · Aug 2, 2024 · e28926c · e28926c
1 parent 383548f
commit e28926c
Showing 1 changed file with 25 additions and 0 deletions.
diff --git a/control/semaphore.go b/control/semaphore.go
@@ -0,0 +1,25 @@
+package control
+
+type (
+	Semaphore interface {
+		Acquire()
+		Release()
+	}
+	_semaphore struct {
+		c chan struct{}
+	}
+)
+
+func NewSemaphore(count uint64) Semaphore {
+	return &_semaphore{
+		c: make(chan struct{}, count),
+	}
+}
+
+func (s *_semaphore) Acquire() {
+	s.c <- struct{}{}
+}
+
+func (s *_semaphore) Release() {
+	<-s.c
+}