RateLimitedSafeWaitGroup must not be copied after first use.
type RateLimitedSafeWaitGroup struct {
// contains filtered or unexported fields
}
func (wg *RateLimitedSafeWaitGroup) Add(delta int) error
Add adds delta, which may be negative, similar to sync.WaitGroup. If Add with a positive delta happens after Wait, it will return error, which prevent unsafe Add.
func (wg *RateLimitedSafeWaitGroup) Done()
Done decrements the WaitGroup counter, rate limiting is applied only when the wait group is in waiting mode.
func (wg *RateLimitedSafeWaitGroup) Wait(limiterFactory RateLimiterFactoryFunc) (int, int, error)
Wait blocks until the WaitGroup counter is zero or a hard limit has elapsed. It returns the number of active request(s) accounted for at the time Wait has been invoked, number of request(s) that have drianed (done using the wait group immediately before Wait returns). Ideally, the both numbers returned should be equal, to indicate that all request(s) using the wait group have released their lock.
RateLimiter abstracts the rate limiter used by RateLimitedSafeWaitGroup. The implementation must be thread-safe.
type RateLimiter interface {
Wait(ctx context.Context) error
}
RateLimiterFactoryFunc is used by the RateLimitedSafeWaitGroup to create a new instance of a RateLimiter that will be used to rate limit the return rate of the active number of request(s). 'count' is the number of requests in flight that are expected to invoke 'Done' on this wait group.
type RateLimiterFactoryFunc func(count int) (RateLimiter, context.Context, context.CancelFunc)
SafeWaitGroup must not be copied after first use.
type SafeWaitGroup struct {
// contains filtered or unexported fields
}
func (wg *SafeWaitGroup) Add(delta int) error
Add adds delta, which may be negative, similar to sync.WaitGroup. If Add with a positive delta happens after Wait, it will return error, which prevent unsafe Add.
func (wg *SafeWaitGroup) Done()
Done decrements the WaitGroup counter.
func (wg *SafeWaitGroup) Wait()
Wait blocks until the WaitGroup counter is zero.