403 lines
9.9 KiB
Go
403 lines
9.9 KiB
Go
// Copyright (C) INFINI Labs & INFINI LIMITED.
|
|
//
|
|
// The INFINI Console is offered under the GNU Affero General Public License v3.0
|
|
// and as commercial software.
|
|
//
|
|
// For commercial licensing, contact us at:
|
|
// - Website: infinilabs.com
|
|
// - Email: hello@infini.ltd
|
|
//
|
|
// Open Source licensed under AGPL V3:
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Affero General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Affero General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
/* Copyright © INFINI Ltd. All rights reserved.
|
|
* web: https://infinilabs.com
|
|
* mail: hello#infini.ltd */
|
|
|
|
package alerting
|
|
|
|
import (
|
|
"fmt"
|
|
alerting2 "infini.sh/console/service/alerting"
|
|
"infini.sh/console/service/alerting/common"
|
|
"infini.sh/framework/core/elastic"
|
|
"infini.sh/framework/core/global"
|
|
"net/http"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
log "github.com/cihub/seelog"
|
|
"infini.sh/console/model/alerting"
|
|
httprouter "infini.sh/framework/core/api/router"
|
|
"infini.sh/framework/core/orm"
|
|
"infini.sh/framework/core/util"
|
|
)
|
|
|
|
func (h *AlertAPI) createChannel(w http.ResponseWriter, req *http.Request, ps httprouter.Params) {
|
|
var obj = &alerting.Channel{}
|
|
err := h.DecodeJSON(req, obj)
|
|
if err != nil {
|
|
h.WriteError(w, err.Error(), http.StatusInternalServerError)
|
|
log.Error(err)
|
|
return
|
|
}
|
|
|
|
err = orm.Create(nil, obj)
|
|
if err != nil {
|
|
h.WriteError(w, err.Error(), http.StatusInternalServerError)
|
|
log.Error(err)
|
|
return
|
|
}
|
|
|
|
h.WriteCreatedOKJSON(w, obj.ID)
|
|
|
|
}
|
|
|
|
func (h *AlertAPI) getChannel(w http.ResponseWriter, req *http.Request, ps httprouter.Params) {
|
|
id := ps.MustGetParameter("channel_id")
|
|
|
|
obj := alerting.Channel{}
|
|
obj.ID = id
|
|
|
|
exists, err := orm.Get(&obj)
|
|
if !exists || err != nil {
|
|
h.WriteJSON(w, util.MapStr{
|
|
"_id": id,
|
|
"found": false,
|
|
}, http.StatusNotFound)
|
|
return
|
|
}
|
|
if err != nil {
|
|
h.WriteError(w, err.Error(), http.StatusInternalServerError)
|
|
log.Error(err)
|
|
return
|
|
}
|
|
|
|
h.WriteGetOKJSON(w, id, obj)
|
|
}
|
|
|
|
func (h *AlertAPI) updateChannel(w http.ResponseWriter, req *http.Request, ps httprouter.Params) {
|
|
id := ps.MustGetParameter("channel_id")
|
|
obj := alerting.Channel{}
|
|
|
|
obj.ID = id
|
|
exists, err := orm.Get(&obj)
|
|
if !exists || err != nil {
|
|
h.WriteJSON(w, util.MapStr{
|
|
"_id": id,
|
|
"result": "not_found",
|
|
}, http.StatusNotFound)
|
|
return
|
|
}
|
|
|
|
id = obj.ID
|
|
create := obj.Created
|
|
obj = alerting.Channel{}
|
|
err = h.DecodeJSON(req, &obj)
|
|
if err != nil {
|
|
h.WriteError(w, err.Error(), http.StatusInternalServerError)
|
|
log.Error(err)
|
|
return
|
|
}
|
|
|
|
//protect
|
|
obj.ID = id
|
|
obj.Created = create
|
|
err = orm.Update(nil, &obj)
|
|
if err != nil {
|
|
h.WriteError(w, err.Error(), http.StatusInternalServerError)
|
|
log.Error(err)
|
|
return
|
|
}
|
|
|
|
h.WriteUpdatedOKJSON(w, id)
|
|
}
|
|
|
|
func (h *AlertAPI) deleteChannel(w http.ResponseWriter, req *http.Request, ps httprouter.Params) {
|
|
reqBody := struct {
|
|
ChannelIDs []string `json:"ids"`
|
|
}{}
|
|
err := h.DecodeJSON(req, &reqBody)
|
|
if err != nil {
|
|
h.WriteError(w, err.Error(), http.StatusInternalServerError)
|
|
log.Error(err)
|
|
return
|
|
}
|
|
if len(reqBody.ChannelIDs) == 0 {
|
|
if err != nil {
|
|
h.WriteError(w, "channel ids required", http.StatusInternalServerError)
|
|
log.Error(err)
|
|
return
|
|
}
|
|
}
|
|
queryDsl := util.MapStr{
|
|
"size": 1,
|
|
"query": util.MapStr{
|
|
"bool": util.MapStr{
|
|
"minimum_should_match": 1,
|
|
"should": []util.MapStr{
|
|
{
|
|
"terms": util.MapStr{
|
|
"notification_config.normal.id": reqBody.ChannelIDs,
|
|
},
|
|
},
|
|
{
|
|
"terms": util.MapStr{
|
|
"notification_config.escalation.id": reqBody.ChannelIDs,
|
|
},
|
|
},
|
|
{
|
|
"terms": util.MapStr{
|
|
"recovery_notification_config.normal.id": reqBody.ChannelIDs,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
q := orm.Query{
|
|
RawQuery: util.MustToJSONBytes(queryDsl),
|
|
}
|
|
err, result := orm.Search(alerting.Rule{}, &q)
|
|
if err != nil {
|
|
h.WriteError(w, err.Error(), http.StatusInternalServerError)
|
|
log.Error(err)
|
|
return
|
|
}
|
|
if len(result.Result) > 0 {
|
|
var ruleName interface{} = ""
|
|
if m, ok := result.Result[0].(map[string]interface{}); ok {
|
|
ruleName = m["name"]
|
|
}
|
|
h.WriteError(w, fmt.Sprintf("failed to delete channel since it is used by rule [%s]", ruleName), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
queryDsl = util.MapStr{
|
|
"query": util.MapStr{
|
|
"terms": util.MapStr{
|
|
"id": reqBody.ChannelIDs,
|
|
},
|
|
},
|
|
}
|
|
|
|
err = orm.DeleteBy(alerting.Channel{}, util.MustToJSONBytes(queryDsl))
|
|
if err != nil {
|
|
h.WriteError(w, err.Error(), http.StatusInternalServerError)
|
|
log.Error(err)
|
|
return
|
|
}
|
|
|
|
h.WriteJSON(w, util.MapStr{
|
|
"ids": reqBody.ChannelIDs,
|
|
"result": "deleted",
|
|
}, 200)
|
|
}
|
|
|
|
func (h *AlertAPI) searchChannel(w http.ResponseWriter, req *http.Request, ps httprouter.Params) {
|
|
|
|
var (
|
|
keyword = h.GetParameterOrDefault(req, "keyword", "")
|
|
strSize = h.GetParameterOrDefault(req, "size", "20")
|
|
strFrom = h.GetParameterOrDefault(req, "from", "0")
|
|
subType = h.GetParameterOrDefault(req, "sub_type", "")
|
|
typ = h.GetParameterOrDefault(req, "type", "")
|
|
sort = h.GetParameterOrDefault(req, "sort", "updated:desc")
|
|
)
|
|
mustQ := []interface{}{}
|
|
if keyword != "" {
|
|
mustQ = append(mustQ, util.MapStr{
|
|
"query_string": util.MapStr{"default_field": "*", "query": keyword},
|
|
})
|
|
}
|
|
if typ != "" {
|
|
mustQ = append(mustQ, util.MapStr{
|
|
"term": util.MapStr{
|
|
"type": util.MapStr{
|
|
"value": typ,
|
|
},
|
|
},
|
|
})
|
|
}
|
|
if subType != "" {
|
|
mustQ = append(mustQ, util.MapStr{
|
|
"term": util.MapStr{
|
|
"sub_type": util.MapStr{
|
|
"value": subType,
|
|
},
|
|
},
|
|
})
|
|
}
|
|
size, _ := strconv.Atoi(strSize)
|
|
if size <= 0 {
|
|
size = 20
|
|
}
|
|
from, _ := strconv.Atoi(strFrom)
|
|
if from < 0 {
|
|
from = 0
|
|
}
|
|
var (
|
|
sortField string
|
|
sortDirection string
|
|
)
|
|
sortParts := strings.Split(sort, ":")
|
|
sortField = sortParts[0]
|
|
if len(sortParts) >= 2 {
|
|
sortDirection = sortParts[1]
|
|
}
|
|
if sortDirection == "" {
|
|
sortDirection = "asc"
|
|
}
|
|
query := util.MapStr{
|
|
"size": size,
|
|
"from": from,
|
|
"query": util.MapStr{
|
|
"bool": util.MapStr{
|
|
"must": mustQ,
|
|
},
|
|
},
|
|
"sort": []util.MapStr{
|
|
{
|
|
sortField: util.MapStr{
|
|
"order": sortDirection,
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
q := orm.Query{
|
|
RawQuery: util.MustToJSONBytes(query),
|
|
}
|
|
|
|
err, res := orm.Search(&alerting.Channel{}, &q)
|
|
if err != nil {
|
|
log.Error(err)
|
|
h.WriteError(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
h.Write(w, res.Raw)
|
|
}
|
|
|
|
func (h *AlertAPI) testChannel(w http.ResponseWriter, req *http.Request, ps httprouter.Params) {
|
|
obj := alerting.Channel{}
|
|
err := h.DecodeJSON(req, &obj)
|
|
if err != nil {
|
|
h.WriteError(w, err.Error(), http.StatusInternalServerError)
|
|
log.Error(err)
|
|
return
|
|
}
|
|
envVariables, err := alerting2.GetEnvVariables()
|
|
if err != nil {
|
|
log.Error(err)
|
|
h.WriteError(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
firstGrpValue := global.MustLookupString(elastic.GlobalSystemElasticsearchID)
|
|
ctx := map[string]interface{}{
|
|
"title": "INFINI platform test alert message",
|
|
"message": "This is just a test message, do not reply!",
|
|
"objects": []string{".infini_metrics"},
|
|
"trigger_at": time.Now().UnixMilli(),
|
|
"duration": "5m10s",
|
|
"rule_id": util.GetUUID(),
|
|
"rule_name": "test rule",
|
|
"resource_id": util.GetUUID(),
|
|
"resource_name": "test resource",
|
|
"event_id": util.GetUUID(),
|
|
"timestamp": time.Now().UnixMilli(),
|
|
"first_group_value": firstGrpValue,
|
|
"first_threshold": "90",
|
|
"priority": "critical",
|
|
"results": []util.MapStr{
|
|
{"threshold": "90",
|
|
"priority": "critical",
|
|
"group_values": []string{firstGrpValue, "group_value2"},
|
|
"issue_timestamp": time.Now().UnixMilli() - 500,
|
|
"result_value": 90,
|
|
"relation_values": util.MapStr{"a": 100, "b": 90},
|
|
},
|
|
},
|
|
"env": envVariables,
|
|
}
|
|
_, err, _ = common.PerformChannel(&obj, ctx)
|
|
if err != nil {
|
|
h.WriteError(w, err.Error(), http.StatusInternalServerError)
|
|
log.Error(err)
|
|
return
|
|
}
|
|
h.WriteAckOKJSON(w)
|
|
}
|
|
|
|
func (alertAPI *AlertAPI) batchEnableChannel(w http.ResponseWriter, req *http.Request, ps httprouter.Params) {
|
|
var channelIDs = []string{}
|
|
err := alertAPI.DecodeJSON(req, &channelIDs)
|
|
if err != nil {
|
|
log.Error(err)
|
|
alertAPI.WriteError(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
if len(channelIDs) == 0 {
|
|
alertAPI.WriteJSON(w, util.MapStr{}, http.StatusOK)
|
|
return
|
|
}
|
|
if len(channelIDs) > 0 {
|
|
err = setChannelEnabled(true, channelIDs)
|
|
if err != nil {
|
|
log.Error(err)
|
|
alertAPI.WriteError(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
}
|
|
alertAPI.WriteAckOKJSON(w)
|
|
}
|
|
|
|
func (alertAPI *AlertAPI) batchDisableChannel(w http.ResponseWriter, req *http.Request, ps httprouter.Params) {
|
|
var channelIDs = []string{}
|
|
err := alertAPI.DecodeJSON(req, &channelIDs)
|
|
if err != nil {
|
|
log.Error(err)
|
|
alertAPI.WriteError(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
if len(channelIDs) == 0 {
|
|
alertAPI.WriteJSON(w, util.MapStr{}, http.StatusOK)
|
|
return
|
|
}
|
|
if len(channelIDs) > 0 {
|
|
err = setChannelEnabled(false, channelIDs)
|
|
if err != nil {
|
|
log.Error(err)
|
|
alertAPI.WriteError(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
}
|
|
alertAPI.WriteAckOKJSON(w)
|
|
}
|
|
|
|
func setChannelEnabled(enabled bool, channelIDs []string) error {
|
|
q := util.MapStr{
|
|
"query": util.MapStr{
|
|
"terms": util.MapStr{
|
|
"id": channelIDs,
|
|
},
|
|
},
|
|
"script": util.MapStr{
|
|
"source": fmt.Sprintf("ctx._source['enabled'] = %v", enabled),
|
|
},
|
|
}
|
|
err := orm.UpdateBy(alerting.Channel{}, util.MustToJSONBytes(q))
|
|
return err
|
|
}
|