io/ratelimit: allow new limiter with duration

5aaee9 · Jul 6, 2023 · ef08fd0 · ef08fd0
1 parent 5289808
commit ef08fd0
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/pkgs/io/ratelimit/limiter.go b/pkgs/io/ratelimit/limiter.go
@@ -24,6 +24,16 @@ func NewLimiter(size uint64) *Limiter {
 	}
 }
 
+// NewLimiter returns new limiter
+func NewLimiterWithDuration(size uint64, duration time.Duration) *Limiter {
+	return &Limiter{
+		size:     size,
+		now:      0,
+		duration: duration,
+		last:     time.Now(),
+	}
+}
+
 // Wait allow you take data
 func (l *Limiter) Wait(size uint64) {
 	if l.size == 0 {