From 97731564c1a9350676a7cda161ec1a1e171bbae3 Mon Sep 17 00:00:00 2001
From: liugq <silenceqi@hotmail.com>
Date: Sun, 8 Oct 2023 11:00:04 +0800
Subject: [PATCH] add name and tags config

---
 plugin/task_manager/cluster_comparison/orm.go | 4 ++++
 plugin/task_manager/model/comparison.go       | 3 +++
 2 files changed, 7 insertions(+)

diff --git a/plugin/task_manager/cluster_comparison/orm.go b/plugin/task_manager/cluster_comparison/orm.go
index f3d37c57..5682a61d 100644
--- a/plugin/task_manager/cluster_comparison/orm.go
+++ b/plugin/task_manager/cluster_comparison/orm.go
@@ -105,6 +105,7 @@ func buildTask(config *migration_model.ClusterComparisonTaskConfig, creator *rba
 				"source_total_docs": sourceTotalDocs,
 				"target_total_docs": targetTotalDocs,
 				"permit_nodes": config.Settings.Execution.Nodes.Permit,
+				"name": config.Name,
 			},
 		},
 		Cancellable:  true,
@@ -112,6 +113,9 @@ func buildTask(config *migration_model.ClusterComparisonTaskConfig, creator *rba
 		Status:       task.StatusInit,
 		ConfigString: util.MustToJSON(config),
 	}
+	if len(config.Tags) > 0 {
+		t.Metadata.Labels["tags"] = config.Tags
+	}
 	t.ID = util.GetUUID()
 	return &t, nil
 }
diff --git a/plugin/task_manager/model/comparison.go b/plugin/task_manager/model/comparison.go
index ba9d046a..07a4085a 100644
--- a/plugin/task_manager/model/comparison.go
+++ b/plugin/task_manager/model/comparison.go
@@ -5,6 +5,8 @@ import (
 )
 
 type ClusterComparisonTaskConfig struct {
+	Name string `json:"name"`
+	Tags []string `json:"tags"`
 	Cluster struct {
 		Source ClusterInfo `json:"source"`
 		Target ClusterInfo `json:"target"`
@@ -31,6 +33,7 @@ type ClusterComparisonIndexConfig struct {
 	// only used in API
 	ScrollPercent   float64 `json:"scroll_percent,omitempty"`
 	ErrorPartitions int     `json:"error_partitions,omitempty"`
+	RunningChildren int `json:"running_children,omitempty"`
 }
 
 type IndexComparisonTaskConfig struct {