From 0e8231fd59d89e6bd98a472e2b2fa1bbdc581c9d Mon Sep 17 00:00:00 2001 From: Miguel Palau Zarza Date: Mon, 5 Aug 2024 10:43:54 -0600 Subject: [PATCH] pass ctx properly in internal client.ExecuteStatement Signed-off-by: Miguel Palau --- internal/client/client.go | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/internal/client/client.go b/internal/client/client.go index febab52..47d3a5e 100644 --- a/internal/client/client.go +++ b/internal/client/client.go @@ -167,8 +167,7 @@ func (tsc *ThriftServiceClient) ExecuteStatement(ctx context.Context, req *cli_s log, ctx = LoggerAndContext(ctx, req) msg, start := log.Track("ExecuteStatement") - // We use context.Background to fix a problem where on context done the query would not be cancelled. - resp, err := tsc.TCLIServiceClient.ExecuteStatement(context.Background(), req) + resp, err := tsc.TCLIServiceClient.ExecuteStatement(ctx, req) log, ctx = LoggerAndContext(ctx, resp) logDisplayMessage(resp, log) logExecStatementState(resp, log)