From 3b91a57bc1fa5af164d542ecba43d6de7cb1fb78 Mon Sep 17 00:00:00 2001 From: medcl Date: Sat, 18 Nov 2023 13:28:23 +0800 Subject: [PATCH] add timeout to context --- plugin/task_manager/cluster_comparison/cluster_comparison.go | 4 +++- plugin/task_manager/cluster_migration/cluster_migration.go | 4 +++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/plugin/task_manager/cluster_comparison/cluster_comparison.go b/plugin/task_manager/cluster_comparison/cluster_comparison.go index 982419f2..72a07722 100644 --- a/plugin/task_manager/cluster_comparison/cluster_comparison.go +++ b/plugin/task_manager/cluster_comparison/cluster_comparison.go @@ -97,7 +97,9 @@ func (p *processor) splitMajorTask(taskItem *task.Task) error { var sourceTotalDocs int64 var targetTotalDocs int64 - ctx := context.Background() + + ctx, cancel := context.WithTimeout(context.Background(), time.Duration(time.Second*60)) + defer cancel() for i, index := range clusterComparisonTask.Indices { sourceDump := migration_model.IndexComparisonDumpConfig{ diff --git a/plugin/task_manager/cluster_migration/cluster_migration.go b/plugin/task_manager/cluster_migration/cluster_migration.go index 4424c0cd..27d9c757 100644 --- a/plugin/task_manager/cluster_migration/cluster_migration.go +++ b/plugin/task_manager/cluster_migration/cluster_migration.go @@ -96,7 +96,9 @@ func (p *processor) splitMajorMigrationTask(taskItem *task.Task) error { pids = append(pids, taskItem.ID) var totalDocs int64 - ctx := context.Background() + + ctx, cancel := context.WithTimeout(context.Background(), time.Duration(time.Second*60)) + defer cancel() for i, index := range clusterMigrationTask.Indices { source := migration_model.IndexMigrationSourceConfig{