diff --git a/system/locker.go b/system/locker.go index eab2a15..b7937e5 100644 --- a/system/locker.go +++ b/system/locker.go @@ -51,7 +51,9 @@ func (l *Locker) TryAcquire(ctx context.Context) error { return nil case <-ctx.Done(): if err := ctx.Err(); err != nil { - return err + if errors.Is(err, context.DeadlineExceeded) || errors.Is(err, context.Canceled) { + return ErrLockerLocked + } } return nil }