123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197 |
- // Copyright 2021 The Gitea Authors. All rights reserved.
- // Use of this source code is governed by a MIT-style
- // license that can be found in the LICENSE file.
-
- package base
-
- import (
- "context"
- "time"
- )
-
- var (
- _ Downloader = &RetryDownloader{}
- )
-
- // RetryDownloader retry the downloads
- type RetryDownloader struct {
- Downloader
- ctx context.Context
- RetryTimes int // the total execute times
- RetryDelay int // time to delay seconds
- }
-
- // NewRetryDownloader creates a retry downloader
- func NewRetryDownloader(ctx context.Context, downloader Downloader, retryTimes, retryDelay int) *RetryDownloader {
- return &RetryDownloader{
- Downloader: downloader,
- ctx: ctx,
- RetryTimes: retryTimes,
- RetryDelay: retryDelay,
- }
- }
-
- func (d *RetryDownloader) retry(work func() error) error {
- var (
- times = d.RetryTimes
- err error
- )
- for ; times > 0; times-- {
- if err = work(); err == nil {
- return nil
- }
- if IsErrNotSupported(err) {
- return err
- }
- select {
- case <-d.ctx.Done():
- return d.ctx.Err()
- case <-time.After(time.Second * time.Duration(d.RetryDelay)):
- }
- }
- return err
- }
-
- // SetContext set context
- func (d *RetryDownloader) SetContext(ctx context.Context) {
- d.ctx = ctx
- d.Downloader.SetContext(ctx)
- }
-
- // GetRepoInfo returns a repository information with retry
- func (d *RetryDownloader) GetRepoInfo() (*Repository, error) {
- var (
- repo *Repository
- err error
- )
-
- err = d.retry(func() error {
- repo, err = d.Downloader.GetRepoInfo()
- return err
- })
-
- return repo, err
- }
-
- // GetTopics returns a repository's topics with retry
- func (d *RetryDownloader) GetTopics() ([]string, error) {
- var (
- topics []string
- err error
- )
-
- err = d.retry(func() error {
- topics, err = d.Downloader.GetTopics()
- return err
- })
-
- return topics, err
- }
-
- // GetMilestones returns a repository's milestones with retry
- func (d *RetryDownloader) GetMilestones() ([]*Milestone, error) {
- var (
- milestones []*Milestone
- err error
- )
-
- err = d.retry(func() error {
- milestones, err = d.Downloader.GetMilestones()
- return err
- })
-
- return milestones, err
- }
-
- // GetReleases returns a repository's releases with retry
- func (d *RetryDownloader) GetReleases() ([]*Release, error) {
- var (
- releases []*Release
- err error
- )
-
- err = d.retry(func() error {
- releases, err = d.Downloader.GetReleases()
- return err
- })
-
- return releases, err
- }
-
- // GetLabels returns a repository's labels with retry
- func (d *RetryDownloader) GetLabels() ([]*Label, error) {
- var (
- labels []*Label
- err error
- )
-
- err = d.retry(func() error {
- labels, err = d.Downloader.GetLabels()
- return err
- })
-
- return labels, err
- }
-
- // GetIssues returns a repository's issues with retry
- func (d *RetryDownloader) GetIssues(page, perPage int) ([]*Issue, bool, error) {
- var (
- issues []*Issue
- isEnd bool
- err error
- )
-
- err = d.retry(func() error {
- issues, isEnd, err = d.Downloader.GetIssues(page, perPage)
- return err
- })
-
- return issues, isEnd, err
- }
-
- // GetComments returns a repository's comments with retry
- func (d *RetryDownloader) GetComments(opts GetCommentOptions) ([]*Comment, bool, error) {
- var (
- comments []*Comment
- isEnd bool
- err error
- )
-
- err = d.retry(func() error {
- comments, isEnd, err = d.Downloader.GetComments(opts)
- return err
- })
-
- return comments, isEnd, err
- }
-
- // GetPullRequests returns a repository's pull requests with retry
- func (d *RetryDownloader) GetPullRequests(page, perPage int) ([]*PullRequest, bool, error) {
- var (
- prs []*PullRequest
- err error
- isEnd bool
- )
-
- err = d.retry(func() error {
- prs, isEnd, err = d.Downloader.GetPullRequests(page, perPage)
- return err
- })
-
- return prs, isEnd, err
- }
-
- // GetReviews returns pull requests reviews
- func (d *RetryDownloader) GetReviews(pullRequestContext IssueContext) ([]*Review, error) {
- var (
- reviews []*Review
- err error
- )
-
- err = d.retry(func() error {
- reviews, err = d.Downloader.GetReviews(pullRequestContext)
- return err
- })
-
- return reviews, err
- }
|