Merge branch 'master' of https://github.com/bingoohuang/seaweedfs

1 files changed, 43 insertions, 0 deletions

diff --git a/weed/util/retry.go b/weed/util/retry.go
new file mode 100644
index 000000000..7b0f2d3c3
--- /dev/null
+++ b/weed/util/retry.go
@@ -0,0 +1,43 @@
+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: err: %v", name, err)
+			time.Sleep(waitTime)
+			waitTime += waitTime / 2
+		} else {
+			break
+		}
+	}
+	return err
+}
+
+// return the first non empty string
+func Nvl(values ...string) string {
+	for _, s := range values {
+		if s != "" {
+			return s
+		}
+	}
+	return ""
+}