From 7756df1a471e501d4771540269332a92e421baca Mon Sep 17 00:00:00 2001
From: Dmytro Nozdrin <nozdrin.dmytro@gmail.com>
Date: Wed, 9 Aug 2023 19:29:06 +0300
Subject: [PATCH] Avoid memory leaks in semaphore

---
 semaphore/semaphore.go | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/semaphore/semaphore.go b/semaphore/semaphore.go
index d4aaa04..b3c1b93 100644
--- a/semaphore/semaphore.go
+++ b/semaphore/semaphore.go
@@ -29,10 +29,15 @@ func New(tickets int, timeout time.Duration) *Semaphore {
 // If it cannot after "timeout" amount of time, it returns ErrNoTickets. It is
 // safe to call Acquire concurrently on a single Semaphore.
 func (s *Semaphore) Acquire() error {
+	timer := time.NewTimer(s.timeout)
 	select {
 	case s.sem <- struct{}{}:
+		if !timer.Stop() {
+			<-timer.C
+		}
+
 		return nil
-	case <-time.After(s.timeout):
+	case <-timer.C:
 		return ErrNoTickets
 	}
 }