pass ctx properly in internal client.ExecuteStatement

Signed-off-by: Miguel Palau Zarza <miguel.palau@grafana.com>
databricks · Aug 5, 2024 · 7319619 · 7319619
1 parent 93222ab
commit 7319619
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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)