From 5baee4dda586c9c623ff746b17affacafa7f6951 Mon Sep 17 00:00:00 2001 From: Florent Poinsard Date: Thu, 20 Feb 2025 16:10:36 -0600 Subject: [PATCH] fix condition check to wrap error Signed-off-by: Florent Poinsard --- go/vt/vttablet/queryservice/wrapped.go | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/go/vt/vttablet/queryservice/wrapped.go b/go/vt/vttablet/queryservice/wrapped.go index aa19f911870..624ace3825b 100644 --- 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