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>
96 lines
2.9 KiB
Go
96 lines
2.9 KiB
Go
// 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,
|
|
),
|
|
}
|
|
}
|