Make API client retry on rate-limit errors

janeczku · May 24, 2016 · ddc9440 · ddc9440
1 parent ca39723
commit ddc9440
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/vultr.go b/vultr.go
@@ -37,10 +37,11 @@ type Driver struct {
 }
 
 const (
-	defaultOS      = 159
-	defaultRegion  = 1
-	defaultPlan    = 29
-	defaultSSHuser = "root"
+	defaultOS        = 159
+	defaultRegion    = 1
+	defaultPlan      = 29
+	defaultSSHuser   = "root"
+	clientMaxRetries = 5
 )
 
 // GetCreateFlags registers the flags this driver adds to
@@ -410,7 +411,7 @@ func (d *Driver) Kill() error {
 func (d *Driver) getClient() *vultr.Client {
 	log.Infof("getting client")
 	if d.client == nil {
-		d.client = vultr.NewClient(d.APIKey, nil)
+		d.client = vultr.NewClient(d.APIKey, &vultr.Options{MaxRetries: clientMaxRetries})
 	}
 	return d.client
 }