92 lines
2.1 KiB
Go
92 lines
2.1 KiB
Go
package datasource
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/alexanderzobnin/grafana-zabbix/pkg/zabbix"
|
|
"github.com/grafana/grafana-plugin-sdk-go/data"
|
|
)
|
|
|
|
func (ds *ZabbixDatasourceInstance) queryProblemsAlerting(ctx context.Context, query *QueryModel) ([]*data.Frame, error) {
|
|
triggers, err := ds.zabbix.GetTriggers(ctx, query.Group.Filter, query.Host.Filter)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if len(triggers) == 0 {
|
|
return []*data.Frame{}, nil
|
|
}
|
|
|
|
triggerIDs := make([]string, len(triggers))
|
|
for i, t := range triggers {
|
|
triggerIDs[i] = t.ID
|
|
}
|
|
|
|
problems, err := ds.zabbix.GetProblems(ctx, triggerIDs)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
problemMap := buildProblemMap(problems)
|
|
currentTime := time.Now()
|
|
|
|
frames := make([]*data.Frame, 0)
|
|
for _, trigger := range triggers {
|
|
for _, host := range trigger.Hosts {
|
|
value := 0.0
|
|
if problemMap[trigger.ID] {
|
|
value = 1.0
|
|
}
|
|
frame := createAlertingFrame(trigger, host, value, currentTime, query.RefID)
|
|
frames = append(frames, frame)
|
|
}
|
|
}
|
|
|
|
return frames, nil
|
|
}
|
|
|
|
func buildProblemMap(problems []zabbix.Problem) map[string]bool {
|
|
problemMap := make(map[string]bool)
|
|
for _, p := range problems {
|
|
problemMap[p.ObjectID] = true
|
|
}
|
|
return problemMap
|
|
}
|
|
|
|
func createAlertingFrame(trigger zabbix.Trigger, host zabbix.ItemHost, value float64, ts time.Time, refID string) *data.Frame {
|
|
labels := data.Labels{
|
|
"host": host.Name,
|
|
"trigger": trigger.Description,
|
|
"trigger_id": trigger.ID,
|
|
"severity": trigger.Priority,
|
|
"tags": formatTags(trigger.Tags),
|
|
}
|
|
|
|
timeField := data.NewField("time", nil, []time.Time{ts})
|
|
valueField := data.NewField("value", labels, []float64{value})
|
|
|
|
frameName := fmt.Sprintf("%s: %s", host.Name, trigger.Description)
|
|
frame := data.NewFrame(frameName, timeField, valueField)
|
|
frame.RefID = refID
|
|
|
|
return frame
|
|
}
|
|
|
|
func formatTags(tags []zabbix.Tag) string {
|
|
if len(tags) == 0 {
|
|
return ""
|
|
}
|
|
parts := make([]string, len(tags))
|
|
for i, t := range tags {
|
|
if t.Value != "" {
|
|
parts[i] = fmt.Sprintf("%s:%s", t.Tag, t.Value)
|
|
} else {
|
|
parts[i] = t.Tag
|
|
}
|
|
}
|
|
return strings.Join(parts, ",")
|
|
}
|