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

[migration] fix progress when index task is running

This commit is contained in:
Kassian Sun 2023-04-19 14:22:34 +08:00 committed by Gitea
parent 7d675ba3d1
commit 1bca4335f1
2 changed files with 19 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
plugin/migration/api.go
View File

@ -686,10 +686,8 @@ func (h *APIHandler) getDataMigrationTaskOfIndex(w http.ResponseWriter, req *htt
} }
} }
taskInfo["data_partition"] = len(subTasks) taskInfo["data_partition"] = len(subTasks)
var taskStats = map[string]struct { var scrollStats = map[string]int64{}
ScrolledDocs float64 var bulkStats = map[string]int64{}
IndexDocs float64
}{}
for instID, taskIDs := range pipelineTaskIDs { for instID, taskIDs := range pipelineTaskIDs {
inst := &model.Instance{} inst := &model.Instance{}
inst.ID = instID inst.ID = instID
@ -706,19 +704,11 @@ func (h *APIHandler) getDataMigrationTaskOfIndex(w http.ResponseWriter, req *htt
for pipelineID, status := range pipelines { for pipelineID, status := range pipelines {
if pid, ok := pipelineSubParentIDs[pipelineID]; ok { if pid, ok := pipelineSubParentIDs[pipelineID]; ok {
if v, err := status.Context.GetValue("es_scroll.scrolled_docs"); err == nil { if vv := migration_util.GetMapIntValue(status.Context, "es_scroll.scrolled_docs"); vv > 0 {
if vv, ok := v.(float64); ok { scrollStats[pid] = vv
stat := taskStats[pid]
stat.ScrolledDocs = vv
taskStats[pid] = stat
}
} }
if v, err := status.Context.GetValue("bulk_indexing.success.count"); err == nil { if vv := migration_util.GetMapIntValue(status.Context, "bulk_indexing.success.count"); vv > 0 {
if vv, ok := v.(float64); ok { bulkStats[pid] = vv
stat := taskStats[pid]
stat.IndexDocs = vv
taskStats[pid] = stat
}
} }
} }
} }
@ -756,23 +746,19 @@ func (h *APIHandler) getDataMigrationTaskOfIndex(w http.ResponseWriter, req *htt
} }
} }
var ( var (
scrollDocs float64 scrollDocs int64
indexDocs float64 indexDocs int64
) )
if stat, ok := taskStats[ptask.ID]; ok { ptaskLabels := util.MapStr(ptask.Metadata.Labels)
scrollDocs = stat.ScrolledDocs if vv, ok := scrollStats[ptask.ID]; ok {
indexDocs = stat.IndexDocs scrollDocs = vv
} else { } else {
if ptask.Status == task2.StatusComplete || ptask.Status == task2.StatusError { scrollDocs = migration_util.GetMapIntValue(ptaskLabels, "scrolled_docs")
if ptask.Metadata.Labels != nil { }
if v, ok := ptask.Metadata.Labels["scrolled_docs"].(float64); ok { if vv, ok := bulkStats[ptask.ID]; ok {
scrollDocs = v indexDocs = vv
} } else {
if v, ok := ptask.Metadata.Labels["index_docs"].(float64); ok { indexDocs = migration_util.GetMapIntValue(ptaskLabels, "index_docs")
indexDocs = v
}
}
}
} }
var subCompletedTime int64 var subCompletedTime int64

4
plugin/migration/pipeline.go
View File

@ -361,7 +361,7 @@ func (p *DispatcherProcessor) handleRunningSubTask(taskItem *task2.Task) error {
func (p *DispatcherProcessor) checkScrollPipelineTaskStatus(scrollTask *task2.Task, totalDocs int64) (scrolled bool, scrolledDocs int64, err error) { func (p *DispatcherProcessor) checkScrollPipelineTaskStatus(scrollTask *task2.Task, totalDocs int64) (scrolled bool, scrolledDocs int64, err error) {
// NOTE: old-version pipeline tasks has empty status // NOTE: old-version pipeline tasks has empty status
if scrollTask.Status == task2.StatusError || scrollTask.Status == "" { if scrollTask.Status == task2.StatusError || scrollTask.Status == "" {
return false, 0, errors.New("scroll pipeline failed") return true, 0, errors.New("scroll pipeline failed")
} }
// scroll not finished yet // scroll not finished yet
@ -383,7 +383,7 @@ func (p *DispatcherProcessor) checkScrollPipelineTaskStatus(scrollTask *task2.Ta
func (p *DispatcherProcessor) checkBulkPipelineTaskStatus(bulkTask *task2.Task, totalDocs int64) (bulked bool, successDocs int64, err error) { func (p *DispatcherProcessor) checkBulkPipelineTaskStatus(bulkTask *task2.Task, totalDocs int64) (bulked bool, successDocs int64, err error) {
if bulkTask.Status == task2.StatusError || bulkTask.Status == "" { if bulkTask.Status == task2.StatusError || bulkTask.Status == "" {
return false, 0, errors.New("bulk pipeline failed") return true, 0, errors.New("bulk pipeline failed")
} }
// start bulk as needed // start bulk as needed