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

[migration] fix task doc stats

This commit is contained in:
Kassian Sun 2023-06-05 17:09:12 +08:00
parent 0f0794aca7
commit 5af5b2142e
2 changed files with 11 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
plugin/migration/comparison_api.go
View File

@ -142,11 +142,6 @@ func (h *APIHandler) getComparisonMajorTaskInfo(taskID string) (taskStats Cluste
},
},
},
{
"terms": util.MapStr{
"status": []string{task.StatusComplete, task.StatusError},
},
},
},
},
},

19
plugin/migration/migration_api.go
View File

@ -89,23 +89,26 @@ func (h *APIHandler) getMigrationMajorTaskInfo(id string) (taskStats migration_m
continue
}
if subTask.Status == task.StatusRunning {
indexMigrationTaskIDs = append(indexMigrationTaskIDs, subTask.ID)
continue
}
cfg := migration_model.IndexMigrationTaskConfig{}
err = migration_util.GetTaskConfig(&subTask, &cfg)
if err != nil {
log.Errorf("failed to get task config, err: %v", err)
continue
}
indexDocs := migration_util.GetMapIntValue(taskLabels, "index_docs")
taskStats.IndexDocs += indexDocs
taskStats.SourceDocs += cfg.Source.DocCount
st := indexState[indexName]
st.IndexDocs += indexDocs
st.SourceDocs += cfg.Source.DocCount
indexState[indexName] = st
if subTask.Status == task.StatusRunning {
indexMigrationTaskIDs = append(indexMigrationTaskIDs, subTask.ID)
continue
}
indexDocs := migration_util.GetMapIntValue(taskLabels, "index_docs")
taskStats.IndexDocs += indexDocs
st.IndexDocs += indexDocs
if subTask.Status == task.StatusError {
st.ErrorPartitions += 1
taskStats.ErrorPartitions += 1