93 lines
3.0 KiB
Go
93 lines
3.0 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"
|
|
"infini.sh/console/model/insight"
|
|
"infini.sh/framework/modules/elastic/common"
|
|
"regexp"
|
|
)
|
|
|
|
type Metric struct {
|
|
insight.Metric
|
|
Title string `json:"title,omitempty"` //text template
|
|
Message string `json:"message,omitempty"` // text template
|
|
Expression string `json:"expression,omitempty" elastic_mapping:"expression:{type:keyword,copy_to:search_text}"` //告警表达式，自动生成 eg: avg(cpu) > 80
|
|
}
|
|
|
|
func (m *Metric) GenerateExpression() (string, error) {
|
|
if len(m.Items) == 1 {
|
|
return fmt.Sprintf("%s(%s)", m.Items[0].Statistic, m.Items[0].Field), nil
|
|
}
|
|
if m.Formula == "" {
|
|
return "", fmt.Errorf("formula should not be empty since there are %d metrics", len(m.Items))
|
|
}
|
|
var (
|
|
expressionBytes = []byte(m.Formula)
|
|
metricExpression string
|
|
)
|
|
for _, item := range m.Items {
|
|
metricExpression = fmt.Sprintf("%s(%s)", item.Statistic, item.Field)
|
|
reg, err := regexp.Compile(item.Name + `([^\w]|$)`)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
expressionBytes = reg.ReplaceAll(expressionBytes, []byte(metricExpression+"$1"))
|
|
}
|
|
|
|
return string(expressionBytes), nil
|
|
}
|
|
|
|
type MetricItem struct {
|
|
Name string `json:"name"`
|
|
Field string `json:"field"`
|
|
Statistic string `json:"statistic"`
|
|
}
|
|
|
|
type QueryResult struct {
|
|
Query string `json:"query"`
|
|
Raw string `json:"raw"`
|
|
MetricData []MetricData `json:"metric_data"`
|
|
Nodata bool `json:"nodata"`
|
|
Min interface{} `json:"-"`
|
|
Max interface{} `json:"-"`
|
|
}
|
|
|
|
type MetricData struct {
|
|
GroupValues []string `json:"group_values"`
|
|
Data map[string][]TimeMetricData `json:"data"`
|
|
}
|
|
|
|
type TimeMetricData []interface{}
|
|
|
|
type AlertMetricItem struct {
|
|
common.MetricItem
|
|
BucketGroups [][]string `json:"bucket_groups,omitempty"`
|
|
}
|