From 2442335aa2e0d25666aca68ee647c5bc42792959 Mon Sep 17 00:00:00 2001 From: "cai.zhang" Date: Tue, 27 Sep 2022 14:56:52 +0800 Subject: [PATCH] Fix indexnode ut bug (#19468) Signed-off-by: cai.zhang Signed-off-by: cai.zhang --- internal/indexnode/indexnode_service_test.go | 3 ++- internal/indexnode/task_scheduler.go | 1 + 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/internal/indexnode/indexnode_service_test.go b/internal/indexnode/indexnode_service_test.go index b1370cfa0d..8fbf666892 100644 --- a/internal/indexnode/indexnode_service_test.go +++ b/internal/indexnode/indexnode_service_test.go @@ -217,7 +217,7 @@ func TestIndexNodeComplex(t *testing.T) { var i int64 t.Logf("preparing mock data...") wg := sync.WaitGroup{} - for i = 0; i < 256; i++ { + for i = 0; i < 10; i++ { task := &testTask{ buildID: i + buildID0, collID: i + collID0, @@ -289,6 +289,7 @@ Loop: if jobNumRet.TotalJobNum == 0 { break Loop } + time.Sleep(time.Second) } } buildIDs := make([]int64, 0, len(tasks)) diff --git a/internal/indexnode/task_scheduler.go b/internal/indexnode/task_scheduler.go index 429adfb50a..558e5841d0 100644 --- a/internal/indexnode/task_scheduler.go +++ b/internal/indexnode/task_scheduler.go @@ -212,6 +212,7 @@ func (sched *TaskScheduler) processTask(t task, q TaskQueue) { }() sched.IndexBuildQueue.AddActiveTask(t) defer sched.IndexBuildQueue.PopActiveTask(t.Name()) + log.Ctx(t.Ctx()).Debug("process task", zap.String("task", t.Name())) pipelines := []func(context.Context) error{t.Prepare, t.LoadData, t.BuildIndex, t.SaveIndexFiles} for _, fn := range pipelines { if err := wrap(fn); err != nil {