Fix improper handling of time.Duration type

adamkpickering · Feb 4, 2023 · 80cd78b · 80cd78b
1 parent c846b43
commit 80cd78b
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/internal/scheduler/two_review.go b/internal/scheduler/two_review.go
@@ -51,17 +51,17 @@ func (scheduler *TwoReviewScheduler) GetNextReview(card *models.Card) (time.Time
 	}
 
 	if reviews[0].Result == models.Failed {
-		interval := scheduler.config.FailedReviewInterval * uint(time.Hour)
-		return reviews[0].Datetime.Add(time.Duration(interval)), nil
+		interval := time.Duration(scheduler.config.FailedReviewInterval) * time.Hour
+		return reviews[0].Datetime.Add(interval), nil
 	}
 
 	if reviewsLength == 1 || (reviewsLength == 2 && reviews[1].Result == models.Failed) {
 		intervalInHours, err := getSecondReviewIntervalFor(reviews[0].Result, scheduler.config)
 		if err != nil {
 			return time.Time{}, fmt.Errorf("failed to get second review interval: %w", err)
 		}
-		interval := intervalInHours * uint(time.Hour)
-		return reviews[0].Datetime.Add(time.Duration(interval)), nil
+		interval := time.Duration(intervalInHours) * time.Hour
+		return reviews[0].Datetime.Add(interval), nil
 	}
 
 	lastReview := reviews[0]