Change Evaluate signature

This commit is contained in:
nemunaire 2026-04-23 16:36:51 +07:00
commit 73f5782ee3
4 changed files with 65 additions and 53 deletions

View file

@ -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
} }

View file

@ -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
View file

@ -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
View file

@ -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=