From 6a4178648f795e211d2a2b37cd0b20ede4f2cd7a Mon Sep 17 00:00:00 2001
From: Dane Everitt <dane@daneeveritt.com>
Date: Mon, 31 Jan 2022 18:39:41 -0500
Subject: [PATCH] Return context cancelations as a locker locked error

---
 system/locker.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

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
 	}