add check context cancel

Signed-off-by: Kosuke Morimoto <[email protected]>
vdaas · Sep 5, 2024 · e8bcb40 · e8bcb40
1 parent b17cc73
commit e8bcb40
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/internal/backoff/backoff.go b/internal/backoff/backoff.go
@@ -187,7 +187,22 @@ func (b *backoff) Do(
 					jdur = b.addJitter(dur)
 				}
 			}
+
+		}
+	}
+	select {
+	case <-dctx.Done():
+		switch dctx.Err() {
+		case context.DeadlineExceeded:
+			log.Debugf("[backoff]\tfor: "+name+",\tDeadline Exceeded\terror: %v", err.Error())
+			return nil, errors.ErrBackoffTimeout(err)
+		case context.Canceled:
+			log.Debugf("[backoff]\tfor: "+name+",\tCanceled\terror: %v", err.Error())
+			return nil, err
+		default:
+			return nil, errors.Join(dctx.Err(), err)
 		}
+	default:
 	}
 	return res, err
 }