mirror of
https://github.com/idanoo/autobrr
synced 2025-07-22 16:29:12 +00:00
3f8bc0140c
* feat(metrics): add metrics server * chore: update license headers * feat(metrics): add optional basic auth * feat(metrics): add go and process collectors --------- Co-authored-by: ze0s <43699394+zze0s@users.noreply.github.com> Co-authored-by: ze0s <ze0s@riseup.net>
52 lines
1.6 KiB
Go
52 lines
1.6 KiB
Go
// Copyright (c) 2021 - 2025, Ludvig Lundgren and the autobrr contributors.
|
|
// SPDX-License-Identifier: GPL-2.0-or-later
|
|
|
|
package metrics
|
|
|
|
import (
|
|
"github.com/autobrr/autobrr/internal/feed"
|
|
"github.com/autobrr/autobrr/internal/filter"
|
|
"github.com/autobrr/autobrr/internal/irc"
|
|
"github.com/autobrr/autobrr/internal/list"
|
|
"github.com/autobrr/autobrr/internal/metrics/collector"
|
|
"github.com/autobrr/autobrr/internal/release"
|
|
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"github.com/prometheus/client_golang/prometheus/collectors"
|
|
)
|
|
|
|
type MetricsManager struct {
|
|
registry *prometheus.Registry
|
|
}
|
|
|
|
func NewMetricsManager(version string, commit string, date string, releaseService release.Service, ircService irc.Service, feedService feed.Service, listService list.Service, filterService filter.Service) *MetricsManager {
|
|
registry := prometheus.NewRegistry()
|
|
registry.MustRegister(
|
|
collectors.NewGoCollector(),
|
|
collectors.NewProcessCollector(collectors.ProcessCollectorOpts{}),
|
|
prometheus.NewGaugeFunc(
|
|
prometheus.GaugeOpts{
|
|
Name: "autobrr_info",
|
|
Help: "Autobrr version information",
|
|
ConstLabels: prometheus.Labels{
|
|
"version": version,
|
|
"build_time": date,
|
|
"revision": commit,
|
|
},
|
|
},
|
|
func() float64 { return 1 },
|
|
),
|
|
collector.NewReleaseCollector(releaseService),
|
|
collector.NewIRCCollector(ircService),
|
|
collector.NewFeedCollector(feedService),
|
|
collector.NewListCollector(listService),
|
|
collector.NewFilterCollector(filterService),
|
|
)
|
|
return &MetricsManager{
|
|
registry: registry,
|
|
}
|
|
}
|
|
|
|
func (s *MetricsManager) GetRegistry() *prometheus.Registry {
|
|
return s.registry
|
|
}
|