mirror of
				https://github.com/go-gitea/gitea.git
				synced 2025-11-03 20:36:07 +01:00 
			
		
		
		
	Add retry for migration http/https requests (#9019)
* Add retry for migration http/https requests * give the more suitable name for retry configuraion items * fix docs and lint * Only use retryDownloader when setting > 1
This commit is contained in:
		@@ -5,7 +5,11 @@
 | 
			
		||||
 | 
			
		||||
package base
 | 
			
		||||
 | 
			
		||||
import "code.gitea.io/gitea/modules/structs"
 | 
			
		||||
import (
 | 
			
		||||
	"time"
 | 
			
		||||
 | 
			
		||||
	"code.gitea.io/gitea/modules/structs"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// Downloader downloads the site repo informations
 | 
			
		||||
type Downloader interface {
 | 
			
		||||
@@ -25,3 +29,148 @@ type DownloaderFactory interface {
 | 
			
		||||
	New(opts MigrateOptions) (Downloader, error)
 | 
			
		||||
	GitServiceType() structs.GitServiceType
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// RetryDownloader retry the downloads
 | 
			
		||||
type RetryDownloader struct {
 | 
			
		||||
	Downloader
 | 
			
		||||
	RetryTimes int // the total execute times
 | 
			
		||||
	RetryDelay int // time to delay seconds
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// NewRetryDownloader creates a retry downloader
 | 
			
		||||
func NewRetryDownloader(downloader Downloader, retryTimes, retryDelay int) *RetryDownloader {
 | 
			
		||||
	return &RetryDownloader{
 | 
			
		||||
		Downloader: downloader,
 | 
			
		||||
		RetryTimes: retryTimes,
 | 
			
		||||
		RetryDelay: retryDelay,
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// GetRepoInfo returns a repository information with retry
 | 
			
		||||
func (d *RetryDownloader) GetRepoInfo() (*Repository, error) {
 | 
			
		||||
	var (
 | 
			
		||||
		times = d.RetryTimes
 | 
			
		||||
		repo  *Repository
 | 
			
		||||
		err   error
 | 
			
		||||
	)
 | 
			
		||||
	for ; times > 0; times-- {
 | 
			
		||||
		if repo, err = d.Downloader.GetRepoInfo(); err == nil {
 | 
			
		||||
			return repo, nil
 | 
			
		||||
		}
 | 
			
		||||
		time.Sleep(time.Second * time.Duration(d.RetryDelay))
 | 
			
		||||
	}
 | 
			
		||||
	return nil, err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// GetTopics returns a repository's topics with retry
 | 
			
		||||
func (d *RetryDownloader) GetTopics() ([]string, error) {
 | 
			
		||||
	var (
 | 
			
		||||
		times  = d.RetryTimes
 | 
			
		||||
		topics []string
 | 
			
		||||
		err    error
 | 
			
		||||
	)
 | 
			
		||||
	for ; times > 0; times-- {
 | 
			
		||||
		if topics, err = d.Downloader.GetTopics(); err == nil {
 | 
			
		||||
			return topics, nil
 | 
			
		||||
		}
 | 
			
		||||
		time.Sleep(time.Second * time.Duration(d.RetryDelay))
 | 
			
		||||
	}
 | 
			
		||||
	return nil, err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// GetMilestones returns a repository's milestones with retry
 | 
			
		||||
func (d *RetryDownloader) GetMilestones() ([]*Milestone, error) {
 | 
			
		||||
	var (
 | 
			
		||||
		times      = d.RetryTimes
 | 
			
		||||
		milestones []*Milestone
 | 
			
		||||
		err        error
 | 
			
		||||
	)
 | 
			
		||||
	for ; times > 0; times-- {
 | 
			
		||||
		if milestones, err = d.Downloader.GetMilestones(); err == nil {
 | 
			
		||||
			return milestones, nil
 | 
			
		||||
		}
 | 
			
		||||
		time.Sleep(time.Second * time.Duration(d.RetryDelay))
 | 
			
		||||
	}
 | 
			
		||||
	return nil, err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// GetReleases returns a repository's releases with retry
 | 
			
		||||
func (d *RetryDownloader) GetReleases() ([]*Release, error) {
 | 
			
		||||
	var (
 | 
			
		||||
		times    = d.RetryTimes
 | 
			
		||||
		releases []*Release
 | 
			
		||||
		err      error
 | 
			
		||||
	)
 | 
			
		||||
	for ; times > 0; times-- {
 | 
			
		||||
		if releases, err = d.Downloader.GetReleases(); err == nil {
 | 
			
		||||
			return releases, nil
 | 
			
		||||
		}
 | 
			
		||||
		time.Sleep(time.Second * time.Duration(d.RetryDelay))
 | 
			
		||||
	}
 | 
			
		||||
	return nil, err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// GetLabels returns a repository's labels with retry
 | 
			
		||||
func (d *RetryDownloader) GetLabels() ([]*Label, error) {
 | 
			
		||||
	var (
 | 
			
		||||
		times  = d.RetryTimes
 | 
			
		||||
		labels []*Label
 | 
			
		||||
		err    error
 | 
			
		||||
	)
 | 
			
		||||
	for ; times > 0; times-- {
 | 
			
		||||
		if labels, err = d.Downloader.GetLabels(); err == nil {
 | 
			
		||||
			return labels, nil
 | 
			
		||||
		}
 | 
			
		||||
		time.Sleep(time.Second * time.Duration(d.RetryDelay))
 | 
			
		||||
	}
 | 
			
		||||
	return nil, err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// GetIssues returns a repository's issues with retry
 | 
			
		||||
func (d *RetryDownloader) GetIssues(page, perPage int) ([]*Issue, bool, error) {
 | 
			
		||||
	var (
 | 
			
		||||
		times  = d.RetryTimes
 | 
			
		||||
		issues []*Issue
 | 
			
		||||
		isEnd  bool
 | 
			
		||||
		err    error
 | 
			
		||||
	)
 | 
			
		||||
	for ; times > 0; times-- {
 | 
			
		||||
		if issues, isEnd, err = d.Downloader.GetIssues(page, perPage); err == nil {
 | 
			
		||||
			return issues, isEnd, nil
 | 
			
		||||
		}
 | 
			
		||||
		time.Sleep(time.Second * time.Duration(d.RetryDelay))
 | 
			
		||||
	}
 | 
			
		||||
	return nil, false, err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// GetComments returns a repository's comments with retry
 | 
			
		||||
func (d *RetryDownloader) GetComments(issueNumber int64) ([]*Comment, error) {
 | 
			
		||||
	var (
 | 
			
		||||
		times    = d.RetryTimes
 | 
			
		||||
		comments []*Comment
 | 
			
		||||
		err      error
 | 
			
		||||
	)
 | 
			
		||||
	for ; times > 0; times-- {
 | 
			
		||||
		if comments, err = d.Downloader.GetComments(issueNumber); err == nil {
 | 
			
		||||
			return comments, nil
 | 
			
		||||
		}
 | 
			
		||||
		time.Sleep(time.Second * time.Duration(d.RetryDelay))
 | 
			
		||||
	}
 | 
			
		||||
	return nil, err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// GetPullRequests returns a repository's pull requests with retry
 | 
			
		||||
func (d *RetryDownloader) GetPullRequests(page, perPage int) ([]*PullRequest, error) {
 | 
			
		||||
	var (
 | 
			
		||||
		times = d.RetryTimes
 | 
			
		||||
		prs   []*PullRequest
 | 
			
		||||
		err   error
 | 
			
		||||
	)
 | 
			
		||||
	for ; times > 0; times-- {
 | 
			
		||||
		if prs, err = d.Downloader.GetPullRequests(page, perPage); err == nil {
 | 
			
		||||
			return prs, nil
 | 
			
		||||
		}
 | 
			
		||||
		time.Sleep(time.Second * time.Duration(d.RetryDelay))
 | 
			
		||||
	}
 | 
			
		||||
	return nil, err
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user