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

fix getting alerting rule 404

This commit is contained in:
liugq 2023-04-03 18:05:27 +08:00
parent abe7acf177
commit 272f29403d
2 changed files with 26 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
plugin/api/alerting/rule.go
View File

@ -5,6 +5,7 @@
package alerting
import (
"errors"
"fmt"
log "github.com/cihub/seelog"
"github.com/r3labs/diff/v2"
@ -21,6 +22,7 @@ import (
"infini.sh/framework/core/queue"
"infini.sh/framework/core/task"
"infini.sh/framework/core/util"
elastic2 "infini.sh/framework/modules/elastic"
"infini.sh/framework/modules/elastic/api"
"infini.sh/framework/modules/elastic/common"
"net/http"
@ -109,13 +111,17 @@ func (alertAPI *AlertAPI) getRule(w http.ResponseWriter, req *http.Request, ps h
obj := alerting.Rule{}
obj.ID = id
exists, err := orm.Get(&obj)
if !exists || err != nil {
_, err := orm.Get(&obj)
if err != nil {
if errors.Is(err, elastic2.ErrNotFound){
alertAPI.WriteJSON(w, util.MapStr{
"_id": id,
"found": false,
}, http.StatusNotFound)
return
}
log.Error(err)
alertAPI.WriteJSON(w, util.MapStr{
"_id": id,
"found": false,
}, http.StatusNotFound)
alertAPI.WriteError(w, err.Error(), http.StatusInternalServerError)
return
}
@ -134,11 +140,15 @@ func (alertAPI *AlertAPI) getRuleDetail(w http.ResponseWriter, req *http.Request
exists, err := orm.Get(&obj)
if !exists || err != nil {
if errors.Is(err, elastic2.ErrNotFound){
alertAPI.WriteJSON(w, util.MapStr{
"_id": id,
"found": false,
}, http.StatusNotFound)
return
}
log.Error(err)
alertAPI.WriteJSON(w, util.MapStr{
"_id": id,
"found": false,
}, http.StatusNotFound)
alertAPI.WriteError(w, err.Error(), http.StatusInternalServerError)
return
}
metricExpression, _ := obj.Metrics.GenerateExpression()

12
plugin/migration/api.go
View File

@ -289,12 +289,12 @@ func (h *APIHandler) stopDataMigrationTask(w http.ResponseWriter, req *http.Requ
}, http.StatusNotFound)
return
}
if task2.IsEnded(obj.Status) {
h.WriteJSON(w, util.MapStr{
"success": true,
}, 200)
return
}
//if task2.IsEnded(obj.Status) {
// h.WriteJSON(w, util.MapStr{
// "success": true,
// }, 200)
// return
//}
//query all pipeline task(scroll/bulk_indexing) and then stop it
err = stopPipelineTasks(id)
if err != nil {