larryh/corso
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add logs

Browse Source
This commit is contained in:
Abhishek Pandey 2023-11-08 22:11:50 -08:00
parent 4df81e7fd7
commit f10a7bb7b7
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/internal/common/limiters/sliding_window.go
View File

@ -89,8 +89,13 @@ func (s *slidingWindow) Wait(ctx context.Context) error {
return s.WaitN(ctx, 1) return s.WaitN(ctx, 1)
} }
// Wait blocks a request until N tokens are available or the context gets // WaitN blocks a request until n tokens are available or the context gets
// cancelled. // cancelled. WaitN should be called with n <= capacity otherwise it will block
// forever.
//
// TODO(pandeyabs): Enforce n <= capacity check. Not adding it right now because
// we are relying on capacity = 0 for ctx cancellation test, which would need
// some refactoring.
func (s *slidingWindow) WaitN(ctx context.Context, n int) error { func (s *slidingWindow) WaitN(ctx context.Context, n int) error {
for i := 0; i < n; i++ { for i := 0; i < n; i++ {
select { select {