Add a simple Retry to preload

[aaf/sms.git] / sms-service / src / preload / preload.go

diff --git a/sms-service/src/preload/preload.go b/sms-service/src/preload/preload.go
index 7e54fb8..9e9b987 100644 (file)
--- a/sms-service/src/preload/preload.go
+++ b/sms-service/src/preload/preload.go
@@ -76,6 +76,7 @@ type smsClient struct {
        //In seconds
        Timeout    int
        CaCertPath string
+       RetryCount int
 
        httpClient *http.Client
 }
@@ -260,17 +261,36 @@ func (c *smsClient) uploadToSMS(data DataJSON) error {
        fmt.Println("Uploading data...")
 
        for _, d := range ldata {
-               err := c.createDomain(d.Name)
+               var err error
+               for i := 0; i < c.RetryCount; i++ {
+                       err = c.createDomain(d.Name)
+                       if err != nil {
+                               fmt.Println(pkgerrors.Cause(err))
+                               fmt.Println("Retrying...")
+                               time.Sleep(5 * time.Second)
+                       } else {
+                               break
+                       }
+               }
                if err != nil {
                        return pkgerrors.Cause(err)
                }
 
                for _, s := range d.Secrets {
-                       err = c.createSecret(d.Name, s.Name, s.Values)
-                       if err != nil {
-                               return pkgerrors.Cause(err)
+                       for i := 0; i < c.RetryCount; i++ {
+                               err = c.createSecret(d.Name, s.Name, s.Values)
+                               if err != nil {
+                                       fmt.Println(pkgerrors.Cause(err))
+                                       fmt.Println("Retrying...")
+                                       time.Sleep(5 * time.Second)
+                               } else {
+                                       break
+                               }
                        }
                }
+               if err != nil {
+                       return pkgerrors.Cause(err)
+               }
        }
 
        return nil
@@ -310,6 +330,7 @@ func main() {
                Timeout:    30,
                BaseURL:    serviceURL,
                CaCertPath: *cacert,
+               RetryCount: 5,
        }
        client.init()