[Fix] RetryDownloader (add GetReviews()) (go-gitea#11093)

* add GetReviews() to RetryDownloader
ydelafollye · Jul 31, 2020 · f6f42df · f6f42df
1 parent 65e41a9
commit f6f42df
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/modules/migrations/base/downloader.go b/modules/migrations/base/downloader.go
@@ -186,3 +186,19 @@ func (d *RetryDownloader) GetPullRequests(page, perPage int) ([]*PullRequest, er
  }
  return nil, err
 }
+
+// GetReviews returns pull requests reviews
+func (d *RetryDownloader) GetReviews(pullRequestNumber int64) ([]*Review, error) {
+ var (
+ times = d.RetryTimes
+ reviews []*Review
+ err error
+ )
+ for ; times > 0; times-- {
+ if reviews, err = d.Downloader.GetReviews(pullRequestNumber); err == nil {
+ return reviews, nil
+ }
+ time.Sleep(time.Second * time.Duration(d.RetryDelay))
+ }
+ return nil, err
+}