idanoo/autobrr
1
0
Fork 0
mirror of https://github.com/idanoo/autobrr synced 2025-07-23 16:59:12 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(metrics): add metrics server (#1930)

Browse source 
* 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>
This commit is contained in:
Antoine 2025-01-25 17:58:18 +01:00 • committed by GitHub
parent 0d5902c8f6
commit 3f8bc0140c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
16 changed files with 1191 additions and 83 deletions
Download patch file Download diff file Expand all files Collapse all files

96
internal/metrics/collector/release.go Normal file
View file

@ -0,0 +1,96 @@
// Copyright (c) 2021 - 2025, Ludvig Lundgren and the autobrr contributors.
// SPDX-License-Identifier: GPL-2.0-or-later
package collector
import (
"context"
"github.com/autobrr/autobrr/internal/release"
"github.com/prometheus/client_golang/prometheus"
)
type releaseCollector struct {
releaseService release.Service
totalCount *prometheus.Desc
filteredCount *prometheus.Desc
filterRejectedCount *prometheus.Desc
pushApprovedCount *prometheus.Desc
pushRejectedCount *prometheus.Desc
pushErrorCount *prometheus.Desc
errorMetric *prometheus.Desc
}
func (collector *releaseCollector) Describe(ch chan<- *prometheus.Desc) {
ch <- collector.totalCount
ch <- collector.filteredCount
ch <- collector.filterRejectedCount
ch <- collector.pushApprovedCount
ch <- collector.pushRejectedCount
ch <- collector.pushErrorCount
ch <- collector.errorMetric
}
func (collector *releaseCollector) Collect(ch chan<- prometheus.Metric) {
stats, err := collector.releaseService.Stats(context.TODO())
if err != nil {
ch <- prometheus.NewInvalidMetric(collector.errorMetric, err)
return
}
ch <- prometheus.MustNewConstMetric(collector.totalCount, prometheus.GaugeValue, float64(stats.TotalCount))
ch <- prometheus.MustNewConstMetric(collector.filteredCount, prometheus.GaugeValue, float64(stats.FilteredCount))
ch <- prometheus.MustNewConstMetric(collector.filterRejectedCount, prometheus.GaugeValue, float64(stats.FilterRejectedCount))
ch <- prometheus.MustNewConstMetric(collector.pushApprovedCount, prometheus.GaugeValue, float64(stats.PushApprovedCount))
ch <- prometheus.MustNewConstMetric(collector.pushRejectedCount, prometheus.GaugeValue, float64(stats.PushRejectedCount))
ch <- prometheus.MustNewConstMetric(collector.pushErrorCount, prometheus.GaugeValue, float64(stats.PushErrorCount))
}
func NewReleaseCollector(releaseService release.Service) *releaseCollector {
return &releaseCollector{
releaseService: releaseService,
totalCount: prometheus.NewDesc(
"autobrr_release_total",
"Number of releases",
nil,
nil,
),
filteredCount: prometheus.NewDesc(
"autobrr_release_filtered_total",
"Number of releases filtered",
nil,
nil,
),
filterRejectedCount: prometheus.NewDesc(
"autobrr_release_filter_rejected_total",
"Number of releases that got rejected because of a filter",
nil,
nil,
),
pushApprovedCount: prometheus.NewDesc(
"autobrr_release_push_approved_total",
"Number of releases push approved",
nil,
nil,
),
pushRejectedCount: prometheus.NewDesc(
"autobrr_release_push_rejected_total",
"Number of releases push rejected",
nil,
nil,
),
pushErrorCount: prometheus.NewDesc(
"autobrr_release_push_error_total",
"Number of releases push errored",
nil,
nil,
),
errorMetric: prometheus.NewDesc(
"autobrr_release_collector_error",
"Error while collecting release metrics",
nil,
nil,
),
}
}