e1464fd317
Fixes https://github.com/docker/distribution/issues/3097 Signed-off-by: Yong Wen Chua <lawliet89@users.noreply.github.com>
25 lines
500 B
Go
25 lines
500 B
Go
package aws
|
|
|
|
import (
|
|
"time"
|
|
)
|
|
|
|
// SleepWithContext will wait for the timer duration to expire, or the context
|
|
// is canceled. Which ever happens first. If the context is canceled the Context's
|
|
// error will be returned.
|
|
//
|
|
// Expects Context to always return a non-nil error if the Done channel is closed.
|
|
func SleepWithContext(ctx Context, dur time.Duration) error {
|
|
t := time.NewTimer(dur)
|
|
defer t.Stop()
|
|
|
|
select {
|
|
case <-t.C:
|
|
break
|
|
case <-ctx.Done():
|
|
return ctx.Err()
|
|
}
|
|
|
|
return nil
|
|
}
|