From 8d27425fc09c51e1f3ffeb1d248e037af90b2ca1 Mon Sep 17 00:00:00 2001
From: Kassian Sun <kassiansun@outlook.com>
Date: Wed, 17 May 2023 16:57:27 +0800
Subject: [PATCH] [migration] fix GetChildTasks

---
 plugin/migration/util/orm.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/plugin/migration/util/orm.go b/plugin/migration/util/orm.go
index 2613b31f..bf569b5f 100644
--- a/plugin/migration/util/orm.go
+++ b/plugin/migration/util/orm.go
@@ -32,7 +32,7 @@ func GetChildTasks(elasticsearch, indexName string, taskID string, taskType stri
 	if len(status) > 0 {
 		musts = append(musts, util.MapStr{
 			"terms": util.MapStr{
-				"status": []string{task.StatusRunning, task.StatusPendingStop, task.StatusReady},
+				"status": status,
 			},
 		})
 	}
@@ -47,7 +47,7 @@ func GetChildTasks(elasticsearch, indexName string, taskID string, taskType stri
 	return GetTasks(elasticsearch, indexName, queryDsl)
 }
 
-func GetTasks(elasticsearch, indexName string, query interface{}) ([]task.Task, error) {
+func GetTasks(elasticsearch, indexName string, query util.MapStr) ([]task.Task, error) {
 	esClient := elastic.GetClient(elasticsearch)
 	res, err := esClient.SearchWithRawQueryDSL(indexName, util.MustToJSONBytes(query))
 	if err != nil {