Browse Source

check Retry configuration in providers

zyx 6 years ago
parent
commit
9f91d90fc5
3 changed files with 9 additions and 0 deletions
  1. 3 0
      worker/cmd_provider.go
  2. 3 0
      worker/rsync_provider.go
  3. 3 0
      worker/two_stage_rsync_provider.go

+ 3 - 0
worker/cmd_provider.go

@@ -24,6 +24,9 @@ type cmdProvider struct {
 
 
 func newCmdProvider(c cmdConfig) (*cmdProvider, error) {
 func newCmdProvider(c cmdConfig) (*cmdProvider, error) {
 	// TODO: check config options
 	// TODO: check config options
+	if c.retry == 0 {
+		c.retry = defaultMaxRetry
+	}
 	provider := &cmdProvider{
 	provider := &cmdProvider{
 		baseProvider: baseProvider{
 		baseProvider: baseProvider{
 			name:     c.name,
 			name:     c.name,

+ 3 - 0
worker/rsync_provider.go

@@ -32,6 +32,9 @@ func newRsyncProvider(c rsyncConfig) (*rsyncProvider, error) {
 	if !strings.HasSuffix(c.upstreamURL, "/") {
 	if !strings.HasSuffix(c.upstreamURL, "/") {
 		return nil, errors.New("rsync upstream URL should ends with /")
 		return nil, errors.New("rsync upstream URL should ends with /")
 	}
 	}
+	if c.retry == 0 {
+		c.retry = defaultMaxRetry
+	}
 	provider := &rsyncProvider{
 	provider := &rsyncProvider{
 		baseProvider: baseProvider{
 		baseProvider: baseProvider{
 			name:     c.name,
 			name:     c.name,

+ 3 - 0
worker/two_stage_rsync_provider.go

@@ -43,6 +43,9 @@ func newTwoStageRsyncProvider(c twoStageRsyncConfig) (*twoStageRsyncProvider, er
 	if !strings.HasSuffix(c.upstreamURL, "/") {
 	if !strings.HasSuffix(c.upstreamURL, "/") {
 		return nil, errors.New("rsync upstream URL should ends with /")
 		return nil, errors.New("rsync upstream URL should ends with /")
 	}
 	}
+	if c.retry == 0 {
+		c.retry = defaultMaxRetry
+	}
 
 
 	provider := &twoStageRsyncProvider{
 	provider := &twoStageRsyncProvider{
 		baseProvider: baseProvider{
 		baseProvider: baseProvider{