NotAShelf/watchdog
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

watchdog: migrate to Cobra and Viper for config management; search /etc for configs

Browse source 
Signed-off-by: NotAShelf <raf@notashelf.dev>
Change-Id: I65dbf466cb030dccc7025585d6282bd26a6a6964
This commit is contained in:
raf 2026-03-01 18:09:34 +03:00
commit f988174145
Signed by: NotAShelf
GPG key ID: 29D95B64378DB4BF
9 changed files with 187 additions and 61 deletions
Download patch file Download diff file Expand all files Collapse all files

4
internal/aggregate/metrics.go
View file

@ -15,7 +15,7 @@ var prometheusLabelPattern = regexp.MustCompile(`^[a-zA-Z0-9_/:.-]*$`)
type MetricsAggregator struct {
pathRegistry *PathRegistry
eventRegistry *CustomEventRegistry
cfg config.Config
cfg *config.Config
pageviews *prometheus.CounterVec
customEvents *prometheus.CounterVec
pathOverflow prometheus.Counter
@ -30,7 +30,7 @@ type MetricsAggregator struct {
func NewMetricsAggregator(
pathRegistry *PathRegistry,
eventRegistry *CustomEventRegistry,
cfg config.Config,
cfg *config.Config,
) *MetricsAggregator {
// Build label names based on what's enabled in config
labels := []string{"path"} // path is always included

12
internal/aggregate/metrics_test.go
View file

@ -22,7 +22,7 @@ func TestMetricsAggregator_RecordPageview(t *testing.T) {
},
}
agg := NewMetricsAggregator(registry, NewCustomEventRegistry(100), cfg)
agg := NewMetricsAggregator(registry, NewCustomEventRegistry(100), &cfg)
// Record pageview with all dimensions
agg.RecordPageview("/home", "US", "desktop", "google.com", "")
@ -51,7 +51,7 @@ func TestMetricsAggregator_RecordPageview_MinimalDimensions(t *testing.T) {
},
}
agg := NewMetricsAggregator(registry, NewCustomEventRegistry(100), cfg)
agg := NewMetricsAggregator(registry, NewCustomEventRegistry(100), &cfg)
// Record pageview with only path
agg.RecordPageview("/home", "", "", "", "")
@ -78,7 +78,7 @@ func TestMetricsAggregator_PathOverflow(t *testing.T) {
},
}
agg := NewMetricsAggregator(registry, NewCustomEventRegistry(100), cfg)
agg := NewMetricsAggregator(registry, NewCustomEventRegistry(100), &cfg)
// Add two paths successfully
registry.Add("/path1")
@ -114,7 +114,7 @@ func TestMetricsAggregator_RecordCustomEvent(t *testing.T) {
},
}
agg := NewMetricsAggregator(registry, NewCustomEventRegistry(100), cfg)
agg := NewMetricsAggregator(registry, NewCustomEventRegistry(100), &cfg)
// Record custom event
agg.RecordCustomEvent("signup")
@ -141,7 +141,7 @@ func TestMetricsAggregator_RecordCustomEvent_MultipleEvents(t *testing.T) {
},
}
agg := NewMetricsAggregator(registry, NewCustomEventRegistry(100), cfg)
agg := NewMetricsAggregator(registry, NewCustomEventRegistry(100), &cfg)
// Record multiple events
agg.RecordCustomEvent("signup")
@ -170,7 +170,7 @@ func TestMetricsAggregator_MustRegister(t *testing.T) {
}
promRegistry := prometheus.NewRegistry()
agg := NewMetricsAggregator(registry, NewCustomEventRegistry(100), cfg)
agg := NewMetricsAggregator(registry, NewCustomEventRegistry(100), &cfg)
// Register metrics
agg.MustRegister(promRegistry)