floraachy
/
console
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add timeout to context

This commit is contained in:
medcl 2023-11-18 13:28:23 +08:00
parent 6139c6adbb
commit 3b91a57bc1
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
plugin/task_manager/cluster_comparison/cluster_comparison.go
View File

@ -97,7 +97,9 @@ func (p *processor) splitMajorTask(taskItem *task.Task) error {
var sourceTotalDocs int64 var sourceTotalDocs int64
var targetTotalDocs 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 { for i, index := range clusterComparisonTask.Indices {
sourceDump := migration_model.IndexComparisonDumpConfig{ sourceDump := migration_model.IndexComparisonDumpConfig{

4
plugin/task_manager/cluster_migration/cluster_migration.go
View File

@ -96,7 +96,9 @@ func (p *processor) splitMajorMigrationTask(taskItem *task.Task) error {
pids = append(pids, taskItem.ID) pids = append(pids, taskItem.ID)
var totalDocs int64 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 { for i, index := range clusterMigrationTask.Indices {
source := migration_model.IndexMigrationSourceConfig{ source := migration_model.IndexMigrationSourceConfig{