mirror of
https://github.com/hanzoai/o11y.git
synced 2026-10-15 12:58:33 +00:00
zeekay
b9f9cdf918
Re-sync the entire SigNoz-derived Go tree to a clean fork of latest SigNoz main (upstream 3e6339019), eliminating the version-skew that broke `go build ./...` (mixed-era packages, ~14 imported-but-absent packages, partial signoz->o11y / clickhouse->datastore rebrands, self-import cycle, duplicate package decls). - pkg/ + cmd/ wholesale replaced with latest SigNoz; imports rewritten github.com/SigNoz/signoz/ -> github.com/hanzoai/o11y/. - Adopt SigNoz's CURRENT structure (resourcefilter->telemetryresourcefilter, integrationtypes->cloudintegrationtypes, roletypes/grammar/times folded in upstream); old paths dropped. - Fork deps repathed to hanzoai + bumped: signoz-otel-collector v0.144.6, clickhouse-go-mock v0.14.1, govaluate v0.1.0; expr-lang/expr => hanzoai/expr v1.17.8 (replace). go.mod reconciled from latest SigNoz; go.sum re-tidied. - ee/ + cmd/enterprise stripped — o11y stays MIT-only; community edition imports no ee packages. - Preserved o11y additions: module path github.com/hanzoai/o11y, mount.go (cloud.Register/zip mount adapter, HIP-0106), NOTICE + LICENSE (SigNoz §4c attribution retained; upstream pkg files are headerless per SigNoz convention). - Reverted o11y cosmetic renames (o11y*/datastore*) and experimental originals (authz/iamauthz, billing, zapreceiver, zapmetricreceiver) to SigNoz canonical to kill the skew; snapshots retained in git history for re-layering. go build ./... = exit 0. go vet clean except benign "unreachable code" in generated ANTLR parser files (inherited from upstream).
102 lines
2.5 KiB
Go
102 lines
2.5 KiB
Go
package rules
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"log/slog"
|
|
|
|
"github.com/google/uuid"
|
|
|
|
"github.com/hanzoai/o11y/pkg/errors"
|
|
ruletypes "github.com/hanzoai/o11y/pkg/types/ruletypes"
|
|
)
|
|
|
|
// TestNotification prepares a dummy rule for given rule parameters and
|
|
// sends a test notification. returns alert count and error (if any)
|
|
func defaultTestNotification(opts PrepareTestRuleOptions) (int, error) {
|
|
ctx := context.Background()
|
|
|
|
if opts.Rule == nil {
|
|
return 0, errors.NewInvalidInputf(errors.CodeInvalidInput, "rule is required")
|
|
}
|
|
|
|
parsedRule := opts.Rule
|
|
var alertname = parsedRule.AlertName
|
|
if alertname == "" {
|
|
// alertname is not mandatory for testing, so picking
|
|
// a random string here
|
|
alertname = uuid.New().String()
|
|
}
|
|
|
|
// append name to indicate this is test alert
|
|
parsedRule.AlertName = fmt.Sprintf("%s%s", alertname, ruletypes.TestAlertPostFix)
|
|
|
|
var rule Rule
|
|
var err error
|
|
|
|
if parsedRule.RuleType == ruletypes.RuleTypeThreshold {
|
|
|
|
// add special labels for test alerts
|
|
parsedRule.Labels[ruletypes.RuleSourceLabel] = ""
|
|
parsedRule.Labels[ruletypes.AlertRuleIDLabel] = ""
|
|
|
|
// create a threshold rule
|
|
rule, err = NewThresholdRule(
|
|
alertname,
|
|
opts.OrgID,
|
|
parsedRule,
|
|
opts.Querier,
|
|
opts.Logger,
|
|
opts.ManagerOpts.Alertmanager.Config().ExternalURL,
|
|
WithSendAlways(),
|
|
WithSendUnmatched(),
|
|
WithSQLStore(opts.SQLStore),
|
|
WithQueryParser(opts.ManagerOpts.QueryParser),
|
|
WithMetadataStore(opts.ManagerOpts.MetadataStore),
|
|
)
|
|
|
|
if err != nil {
|
|
slog.Error("failed to prepare a new threshold rule for test", errors.Attr(err))
|
|
return 0, err
|
|
}
|
|
|
|
} else if parsedRule.RuleType == ruletypes.RuleTypeProm {
|
|
|
|
// create promql rule
|
|
rule, err = NewPromRule(
|
|
alertname,
|
|
opts.OrgID,
|
|
parsedRule,
|
|
opts.Logger,
|
|
opts.ManagerOpts.Prometheus,
|
|
opts.ManagerOpts.Alertmanager.Config().ExternalURL,
|
|
WithSendAlways(),
|
|
WithSendUnmatched(),
|
|
WithSQLStore(opts.SQLStore),
|
|
WithQueryParser(opts.ManagerOpts.QueryParser),
|
|
WithMetadataStore(opts.ManagerOpts.MetadataStore),
|
|
)
|
|
|
|
if err != nil {
|
|
slog.Error("failed to prepare a new promql rule for test", errors.Attr(err))
|
|
return 0, err
|
|
}
|
|
} else {
|
|
return 0, errors.NewInvalidInputf(errors.CodeInvalidInput, "invalid rule type")
|
|
}
|
|
|
|
// set timestamp to current utc time
|
|
ts := time.Now().UTC()
|
|
|
|
alertsFound, err := rule.Eval(ctx, ts)
|
|
if err != nil {
|
|
slog.Error("evaluating rule failed", slog.String("rule.id", rule.ID()), errors.Attr(err))
|
|
return 0, err
|
|
}
|
|
rule.SendAlerts(ctx, ts, 0, time.Duration(1*time.Minute), opts.NotifyFunc)
|
|
|
|
return alertsFound, nil
|
|
}
|