Fix jira_user table to retry rate limit exceeded errors closes #68

jira/errors.go

@@ -1,6 +1,12 @@
 package jira
 
-import "strings"
+import (
+	"context"
+	"log"
+	"strings"
+
+	"github.com/turbot/steampipe-plugin-sdk/v4/plugin"
+)
 
 func isNotFoundError(err error) bool {
 	return strings.Contains(err.Error(), "404")
@@ -9,3 +15,14 @@ func isNotFoundError(err error) bool {
 func isBadRequestError(err error) bool {
 	return strings.Contains(err.Error(), "400")
 }
+
+func shouldRetryError(retryErrors []string) plugin.ErrorPredicateWithContext {
+	return func(ctx context.Context, d *plugin.QueryData, h *plugin.HydrateData, err error) bool {
+
+		if strings.Contains(err.Error(), "429") {
+			log.Printf("[WARN] Received Rate Limit Error")
+			return true
+		}
+		return false
+	}
+}

jira/table_jira_user.go

@@ -26,6 +26,7 @@ func tableUser(_ context.Context) *plugin.Table {
 				// Limit concurrency to avoid a 429 too many requests error
 				Func:           getUserGroups,
 				MaxConcurrency: 50,
+				RetryConfig: 		&plugin.RetryConfig{ ShouldRetryErrorFunc: shouldRetryError([]string{"429"})},
 			},
 		},
 		Columns: []*plugin.Column{
@@ -155,12 +156,13 @@ func getUserGroups(ctx context.Context, d *plugin.QueryData, h *plugin.HydrateDa
 	}
 
 	groups, _, err := client.User.GetGroups(user.AccountID)
+
 	if err != nil {
 		plugin.Logger(ctx).Error("jira_user.getUserGroups", "api_error", err)
 		return nil, err
 	}
 
-	return groups, nil
+	return  groups, nil
 }
 
 //// TRANSFORM FUNCTION