From cfe30782ecc732a8b186a9b76a5aab939559765e Mon Sep 17 00:00:00 2001
From: Lisa Kim <lisa@goteleport.com>
Date: Thu, 9 Jun 2022 17:38:57 -0700
Subject: [PATCH] Move predicate err check earlier, inside RetryWithRelogin

---
 lib/client/api.go | 4 ++++
 tool/tsh/db.go    | 3 ---
 tool/tsh/kube.go  | 3 ---
 tool/tsh/tsh.go   | 6 ------
 4 files changed, 4 insertions(+), 12 deletions(-)

diff --git a/lib/client/api.go b/lib/client/api.go
index 742b7d6f954b1..83160d6b94d7e 100644
--- 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
index 7550404e5bf4f..7d92d49ba6543 100644
--- 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
index 9bd13982afe8d..11c787c3e7cc0 100644
--- 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
index 8cbc64cdd9621..5ee13994ab635 100644
--- 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)
 	}