From 9f91d90fc5ca025b1daffdee423ee31571d2b0a9 Mon Sep 17 00:00:00 2001
From: zyx <zz593141477@gmail.com>
Date: Sat, 13 Apr 2019 11:01:56 +0800
Subject: [PATCH] check Retry configuration in providers

---
 worker/cmd_provider.go             | 3 +++
 worker/rsync_provider.go           | 3 +++
 worker/two_stage_rsync_provider.go | 3 +++
 3 files changed, 9 insertions(+)

diff --git a/worker/cmd_provider.go b/worker/cmd_provider.go
index 292ed0e..6b05681 100644
--- a/worker/cmd_provider.go
+++ b/worker/cmd_provider.go
@@ -24,6 +24,9 @@ type cmdProvider struct {
 
 func newCmdProvider(c cmdConfig) (*cmdProvider, error) {
 	// TODO: check config options
+	if c.retry == 0 {
+		c.retry = defaultMaxRetry
+	}
 	provider := &cmdProvider{
 		baseProvider: baseProvider{
 			name:     c.name,
diff --git a/worker/rsync_provider.go b/worker/rsync_provider.go
index 1ee0709..c0ed57f 100644
--- a/worker/rsync_provider.go
+++ b/worker/rsync_provider.go
@@ -32,6 +32,9 @@ func newRsyncProvider(c rsyncConfig) (*rsyncProvider, error) {
 	if !strings.HasSuffix(c.upstreamURL, "/") {
 		return nil, errors.New("rsync upstream URL should ends with /")
 	}
+	if c.retry == 0 {
+		c.retry = defaultMaxRetry
+	}
 	provider := &rsyncProvider{
 		baseProvider: baseProvider{
 			name:     c.name,
diff --git a/worker/two_stage_rsync_provider.go b/worker/two_stage_rsync_provider.go
index 1346444..991e8c8 100644
--- a/worker/two_stage_rsync_provider.go
+++ b/worker/two_stage_rsync_provider.go
@@ -43,6 +43,9 @@ func newTwoStageRsyncProvider(c twoStageRsyncConfig) (*twoStageRsyncProvider, er
 	if !strings.HasSuffix(c.upstreamURL, "/") {
 		return nil, errors.New("rsync upstream URL should ends with /")
 	}
+	if c.retry == 0 {
+		c.retry = defaultMaxRetry
+	}
 
 	provider := &twoStageRsyncProvider{
 		baseProvider: baseProvider{