123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176 |
- // Copyright 2019 The Gitea Authors. All rights reserved.
- // Copyright 2018 Jonas Franz. 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 (
- "time"
-
- "code.gitea.io/gitea/modules/structs"
- )
-
- // Downloader downloads the site repo informations
- type Downloader interface {
- GetRepoInfo() (*Repository, error)
- GetTopics() ([]string, error)
- GetMilestones() ([]*Milestone, error)
- GetReleases() ([]*Release, error)
- GetLabels() ([]*Label, error)
- GetIssues(page, perPage int) ([]*Issue, bool, error)
- GetComments(issueNumber int64) ([]*Comment, error)
- GetPullRequests(page, perPage int) ([]*PullRequest, error)
- }
-
- // DownloaderFactory defines an interface to match a downloader implementation and create a downloader
- type DownloaderFactory interface {
- Match(opts MigrateOptions) (bool, error)
- 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
- }
|