fix condition check to wrap error

Signed-off-by: Florent Poinsard <florent.poinsard@outlook.fr>
vitessio · Feb 20, 2025 · 5baee4d · 5baee4d
1 parent 83dc5aa
commit 5baee4d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/go/vt/vttablet/queryservice/wrapped.go b/go/vt/vttablet/queryservice/wrapped.go
@@ -84,10 +84,10 @@ func wrapInVT15001(err error, inTx bool) error {
 	}
 	c := vterrors.Code(err)
 	m := err.Error()
-	if c != vtrpcpb.Code_UNAVAILABLE && !strings.Contains(m, "connection refused") {
-		return err
+	if c == vtrpcpb.Code_UNAVAILABLE && strings.Contains(m, "connection refused") {
+		return vterrors.VT15001(c, m)
 	}
-	return vterrors.VT15001(c, m)
+	return err
 }
 
 // wrappedService wraps an existing QueryService with