feature: add context to state API

Signed-off-by: 1046102779 <seachen@tencent.com>
2022-10-24 11:04:48 +08:00 · 2022-10-24 11:04:48 +08:00 · 8bc4286cdf
parent aa68429bbe
commit 8bc4286cdf
1 changed files with 2 additions and 1 deletions
--- a/state/azure/cosmosdb/cosmosdb_query.go
+++ b/state/azure/cosmosdb/cosmosdb_query.go
@ -160,7 +160,8 @@ func (q *Query) execute(ctx context.Context, client *azcosmos.ContainerClient) (

 	token := ""
 	for queryPager.More() {
-		ctx, cancel := context.WithTimeout(ctx, defaultTimeout)
+		var cancel context.CancelFunc
+		ctx, cancel = context.WithTimeout(ctx, defaultTimeout)
 		queryResponse, innerErr := queryPager.NextPage(ctx)
 		cancel()
 		if innerErr != nil {