io/ratelimit: add shared init

5aaee9 · Jul 2, 2023 · a4599c8 · a4599c8
1 parent eca21c6
commit a4599c8
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/pkgs/io/ratelimit/readwrite.go b/pkgs/io/ratelimit/readwrite.go
@@ -16,6 +16,13 @@ func NewRateLimitReaderWriterCloser(underlay io.ReadWriteCloser, size uint64) io
 	}
 }
 
+func NewRateLimitReaderWriterCloserShared(underlay io.ReadWriteCloser, l *Limiter) io.ReadWriteCloser {
+	return &RateLimitReadWriteCloser{
+		limiter:  l,
+		underlay: underlay,
+	}
+}
+
 func (r *RateLimitReadWriteCloser) Close() error {
 	return r.underlay.Close()
 }