Change Evaluate signature
This commit is contained in:
parent
48785e2896
commit
73f5782ee3
4 changed files with 65 additions and 53 deletions
|
|
@ -23,7 +23,6 @@ package checker
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"strings"
|
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
@ -53,48 +52,43 @@ type Metric struct {
|
||||||
Timestamp time.Time `json:"timestamp"`
|
Timestamp time.Time `json:"timestamp"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// EvaluateResult holds the evaluation outcome.
|
// EvaluateResult holds the evaluation outcome for a single target.
|
||||||
type EvaluateResult struct {
|
type EvaluateResult struct {
|
||||||
|
Address string `json:"address"`
|
||||||
Status int `json:"status"`
|
Status int `json:"status"`
|
||||||
Message string `json:"message"`
|
Message string `json:"message"`
|
||||||
Code string `json:"code"`
|
Code string `json:"code"`
|
||||||
}
|
}
|
||||||
|
|
||||||
const (
|
const (
|
||||||
StatusOK = 1
|
StatusUnknown = 0
|
||||||
StatusWarn = 3
|
StatusOK = 1
|
||||||
StatusCrit = 4
|
StatusWarn = 3
|
||||||
|
StatusCrit = 4
|
||||||
)
|
)
|
||||||
|
|
||||||
// Evaluate checks the ping data against the given thresholds.
|
// Evaluate checks the ping data against the given thresholds and returns one
|
||||||
// StatusUnknown indicates the check could not be performed.
|
// result per target.
|
||||||
const StatusUnknown = 0
|
func Evaluate(data *PingData, warningRTT, criticalRTT, warningPacketLoss, criticalPacketLoss float64) []EvaluateResult {
|
||||||
|
|
||||||
func Evaluate(data *PingData, warningRTT, criticalRTT, warningPacketLoss, criticalPacketLoss float64) EvaluateResult {
|
|
||||||
if len(data.Targets) == 0 {
|
if len(data.Targets) == 0 {
|
||||||
return EvaluateResult{
|
return nil
|
||||||
Status: StatusUnknown,
|
|
||||||
Message: "No targets to ping",
|
|
||||||
Code: "ping_no_targets",
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
overallStatus := StatusOK
|
results := make([]EvaluateResult, 0, len(data.Targets))
|
||||||
var summaryParts []string
|
|
||||||
|
|
||||||
for _, target := range data.Targets {
|
for _, target := range data.Targets {
|
||||||
|
status := StatusOK
|
||||||
if target.PacketLoss >= criticalPacketLoss || target.RTTAvg >= criticalRTT {
|
if target.PacketLoss >= criticalPacketLoss || target.RTTAvg >= criticalRTT {
|
||||||
overallStatus = StatusCrit
|
status = StatusCrit
|
||||||
} else if (target.PacketLoss >= warningPacketLoss || target.RTTAvg >= warningRTT) && overallStatus < StatusWarn {
|
} else if target.PacketLoss >= warningPacketLoss || target.RTTAvg >= warningRTT {
|
||||||
overallStatus = StatusWarn
|
status = StatusWarn
|
||||||
}
|
}
|
||||||
|
|
||||||
summaryParts = append(summaryParts, fmt.Sprintf("%s: %.1fms avg, %.0f%% loss", target.Address, target.RTTAvg, target.PacketLoss))
|
results = append(results, EvaluateResult{
|
||||||
}
|
Address: target.Address,
|
||||||
|
Status: status,
|
||||||
return EvaluateResult{
|
Message: fmt.Sprintf("%.1fms avg, %.0f%% loss", target.RTTAvg, target.PacketLoss),
|
||||||
Status: overallStatus,
|
Code: "ping_result",
|
||||||
Message: strings.Join(summaryParts, " | "),
|
})
|
||||||
Code: "ping_result",
|
|
||||||
}
|
}
|
||||||
|
return results
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -106,14 +106,14 @@ func (r *pingRule) ValidateOptions(opts sdk.CheckerOptions) error {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (r *pingRule) Evaluate(ctx context.Context, obs sdk.ObservationGetter, opts sdk.CheckerOptions) sdk.CheckState {
|
func (r *pingRule) Evaluate(ctx context.Context, obs sdk.ObservationGetter, opts sdk.CheckerOptions) []sdk.CheckState {
|
||||||
var data PingData
|
var data PingData
|
||||||
if err := obs.Get(ctx, ObservationKeyPing, &data); err != nil {
|
if err := obs.Get(ctx, ObservationKeyPing, &data); err != nil {
|
||||||
return sdk.CheckState{
|
return []sdk.CheckState{{
|
||||||
Status: sdk.StatusError,
|
Status: sdk.StatusError,
|
||||||
Message: fmt.Sprintf("Failed to get ping data: %v", err),
|
Message: fmt.Sprintf("Failed to get ping data: %v", err),
|
||||||
Code: "ping_error",
|
Code: "ping_error",
|
||||||
}
|
}}
|
||||||
}
|
}
|
||||||
|
|
||||||
warningRTT := sdk.GetFloatOption(opts, "warningRTT", 100)
|
warningRTT := sdk.GetFloatOption(opts, "warningRTT", 100)
|
||||||
|
|
@ -121,26 +121,44 @@ func (r *pingRule) Evaluate(ctx context.Context, obs sdk.ObservationGetter, opts
|
||||||
warningPacketLoss := sdk.GetFloatOption(opts, "warningPacketLoss", 10)
|
warningPacketLoss := sdk.GetFloatOption(opts, "warningPacketLoss", 10)
|
||||||
criticalPacketLoss := sdk.GetFloatOption(opts, "criticalPacketLoss", 50)
|
criticalPacketLoss := sdk.GetFloatOption(opts, "criticalPacketLoss", 50)
|
||||||
|
|
||||||
result := Evaluate(&data, warningRTT, criticalRTT, warningPacketLoss, criticalPacketLoss)
|
results := Evaluate(&data, warningRTT, criticalRTT, warningPacketLoss, criticalPacketLoss)
|
||||||
|
if len(results) == 0 {
|
||||||
var status sdk.Status
|
return []sdk.CheckState{{
|
||||||
switch result.Status {
|
Status: sdk.StatusInfo,
|
||||||
case StatusOK:
|
Message: "No targets to ping",
|
||||||
status = sdk.StatusOK
|
Code: "ping_no_targets",
|
||||||
case StatusWarn:
|
}}
|
||||||
status = sdk.StatusWarn
|
|
||||||
case StatusCrit:
|
|
||||||
status = sdk.StatusCrit
|
|
||||||
default:
|
|
||||||
status = sdk.StatusUnknown
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return sdk.CheckState{
|
targetByAddr := make(map[string]PingTargetResult, len(data.Targets))
|
||||||
Status: status,
|
for _, t := range data.Targets {
|
||||||
Message: result.Message,
|
targetByAddr[t.Address] = t
|
||||||
Code: result.Code,
|
|
||||||
Meta: map[string]any{
|
|
||||||
"targets": data.Targets,
|
|
||||||
},
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
out := make([]sdk.CheckState, 0, len(results))
|
||||||
|
for _, r := range results {
|
||||||
|
var status sdk.Status
|
||||||
|
switch r.Status {
|
||||||
|
case StatusOK:
|
||||||
|
status = sdk.StatusOK
|
||||||
|
case StatusWarn:
|
||||||
|
status = sdk.StatusWarn
|
||||||
|
case StatusCrit:
|
||||||
|
status = sdk.StatusCrit
|
||||||
|
default:
|
||||||
|
status = sdk.StatusUnknown
|
||||||
|
}
|
||||||
|
|
||||||
|
state := sdk.CheckState{
|
||||||
|
Status: status,
|
||||||
|
Subject: r.Address,
|
||||||
|
Message: r.Message,
|
||||||
|
Code: r.Code,
|
||||||
|
}
|
||||||
|
if t, ok := targetByAddr[r.Address]; ok {
|
||||||
|
state.Meta = map[string]any{"target": t}
|
||||||
|
}
|
||||||
|
out = append(out, state)
|
||||||
|
}
|
||||||
|
return out
|
||||||
}
|
}
|
||||||
|
|
|
||||||
2
go.mod
2
go.mod
|
|
@ -3,12 +3,12 @@ module git.happydns.org/checker-ping
|
||||||
go 1.25.0
|
go 1.25.0
|
||||||
|
|
||||||
require (
|
require (
|
||||||
|
git.happydns.org/checker-sdk-go v1.2.0
|
||||||
git.happydns.org/happyDomain v0.7.0
|
git.happydns.org/happyDomain v0.7.0
|
||||||
github.com/prometheus-community/pro-bing v0.8.0
|
github.com/prometheus-community/pro-bing v0.8.0
|
||||||
)
|
)
|
||||||
|
|
||||||
require (
|
require (
|
||||||
git.happydns.org/checker-sdk-go v0.0.1
|
|
||||||
github.com/bytedance/gopkg v0.1.3 // indirect
|
github.com/bytedance/gopkg v0.1.3 // indirect
|
||||||
github.com/bytedance/sonic v1.15.0 // indirect
|
github.com/bytedance/sonic v1.15.0 // indirect
|
||||||
github.com/bytedance/sonic/loader v0.5.0 // indirect
|
github.com/bytedance/sonic/loader v0.5.0 // indirect
|
||||||
|
|
|
||||||
4
go.sum
4
go.sum
|
|
@ -1,5 +1,5 @@
|
||||||
git.happydns.org/checker-sdk-go v0.0.1 h1:4RxCJr73HWKxjOyU/6NJMO8lXJmH0gMLA68EzTqLbQI=
|
git.happydns.org/checker-sdk-go v1.2.0 h1:v4MpKAz0W3PwP+bxx3pya8w893sVH5xTD1of1cc0TV8=
|
||||||
git.happydns.org/checker-sdk-go v0.0.1/go.mod h1:aNAcfYFfbhvH9kJhE0Njp5GX0dQbxdRB0rJ0KvSC5nI=
|
git.happydns.org/checker-sdk-go v1.2.0/go.mod h1:aNAcfYFfbhvH9kJhE0Njp5GX0dQbxdRB0rJ0KvSC5nI=
|
||||||
git.happydns.org/happyDomain v0.7.0 h1:NV82/NbcSeRm0+IUZqaK3Vu9Ovl5+vv4AigUJZMdwws=
|
git.happydns.org/happyDomain v0.7.0 h1:NV82/NbcSeRm0+IUZqaK3Vu9Ovl5+vv4AigUJZMdwws=
|
||||||
git.happydns.org/happyDomain v0.7.0/go.mod h1:5tgkmqFE65kK359rY49V++49wgZ0gco+Gh9X6tbL+bY=
|
git.happydns.org/happyDomain v0.7.0/go.mod h1:5tgkmqFE65kK359rY49V++49wgZ0gco+Gh9X6tbL+bY=
|
||||||
github.com/bytedance/gopkg v0.1.3 h1:TPBSwH8RsouGCBcMBktLt1AymVo2TVsBVCY4b6TnZ/M=
|
github.com/bytedance/gopkg v0.1.3 h1:TPBSwH8RsouGCBcMBktLt1AymVo2TVsBVCY4b6TnZ/M=
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue