Move predicate err check earlier, inside RetryWithRelogin

gravitational · Jun 10, 2022 · cfe3078 · cfe3078
1 parent fd2b4a1
commit cfe3078
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 12 deletions.
diff --git a/lib/client/api.go b/lib/client/api.go
@@ -714,6 +714,10 @@ func RetryWithRelogin(ctx context.Context, tc *TeleportClient, fn func() error)
 		return nil
 	}
 
+	if utils.IsPredicateError(err) {
+		return trace.Wrap(utils.PredicateError{Err: err})
+	}
+
 	if !IsErrorResolvableWithRelogin(err) {
 		return trace.Wrap(err)
 	}

diff --git a/tool/tsh/db.go b/tool/tsh/db.go
@@ -54,9 +54,6 @@ func onListDatabases(cf *CLIConf) error {
 		return trace.Wrap(err)
 	})
 	if err != nil {
-		if utils.IsPredicateError(err) {
-			return trace.Wrap(utils.PredicateError{Err: err})
-		}
 		return trace.Wrap(err)
 	}
 

diff --git a/tool/tsh/kube.go b/tool/tsh/kube.go
@@ -958,9 +958,6 @@ func fetchKubeClusters(ctx context.Context, tc *client.TeleportClient) (teleport
 				}
 				return nil
 			}
-			if utils.IsPredicateError(err) {
-				return trace.Wrap(utils.PredicateError{Err: err})
-			}
 			return trace.Wrap(err)
 		}
 

diff --git a/tool/tsh/tsh.go b/tool/tsh/tsh.go
@@ -1644,9 +1644,6 @@ func onListNodes(cf *CLIConf) error {
 		return err
 	})
 	if err != nil {
-		if utils.IsPredicateError(err) {
-			return trace.Wrap(utils.PredicateError{Err: err})
-		}
 		return trace.Wrap(err)
 	}
 	sort.Slice(nodes, func(i, j int) bool {
@@ -3505,9 +3502,6 @@ func onApps(cf *CLIConf) error {
 		return err
 	})
 	if err != nil {
-		if utils.IsPredicateError(err) {
-			return trace.Wrap(utils.PredicateError{Err: err})
-		}
 		return trace.Wrap(err)
 	}