Merge pull request #34 from chrislusf/master

sync

1 files changed, 31 insertions, 0 deletions

diff --git a/weed/util/retry.go b/weed/util/retry.go
new file mode 100644
index 000000000..faaab0351
--- /dev/null
+++ b/weed/util/retry.go
@@ -0,0 +1,31 @@
+package util
+
+import (
+	"strings"
+	"time"
+
+	"github.com/chrislusf/seaweedfs/weed/glog"
+)
+
+var RetryWaitTime = 6 * time.Second
+
+func Retry(name string, job func() error) (err error) {
+	waitTime := time.Second
+	hasErr := false
+	for waitTime < RetryWaitTime {
+		err = job()
+		if err == nil {
+			if hasErr {
+				glog.V(0).Infof("retry %s successfully", name)
+			}
+			break
+		}
+		if strings.Contains(err.Error(), "transport") {
+			hasErr = true
+			glog.V(0).Infof("retry %s", name)
+			time.Sleep(waitTime)
+			waitTime += waitTime / 2
+		}
+	}
+	return err
+}