idanoo/autobrr
1
0
Fork 0
mirror of https://github.com/idanoo/autobrr synced 2025-07-23 08:49:13 +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

89
internal/metrics/collector/feed.go Normal file
View file

@ -0,0 +1,89 @@
// 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/feed"
"github.com/prometheus/client_golang/prometheus"
)
type feedCollector struct {
feedService feed.Service
totalCount *prometheus.Desc
enabledCount *prometheus.Desc
LastRunTimestamp *prometheus.Desc
NextRunTimestamp *prometheus.Desc
errorMetric *prometheus.Desc
}
func (collector *feedCollector) Describe(ch chan<- *prometheus.Desc) {
ch <- collector.totalCount
ch <- collector.enabledCount
ch <- collector.LastRunTimestamp
ch <- collector.NextRunTimestamp
ch <- collector.errorMetric
}
func (collector *feedCollector) Collect(ch chan<- prometheus.Metric) {
feeds, err := collector.feedService.Find(context.TODO())
if err != nil {
ch <- prometheus.NewInvalidMetric(collector.errorMetric, err)
return
}
enabled := 0
for _, f := range feeds {
if f.Enabled {
enabled++
}
if !f.LastRun.IsZero() {
ch <- prometheus.MustNewConstMetric(collector.LastRunTimestamp, prometheus.GaugeValue, float64(int(f.LastRun.Unix())), f.Name)
}
if !f.NextRun.IsZero() {
ch <- prometheus.MustNewConstMetric(collector.NextRunTimestamp, prometheus.GaugeValue, float64(int(f.NextRun.Unix())), f.Name)
}
}
ch <- prometheus.MustNewConstMetric(collector.totalCount, prometheus.GaugeValue, float64(len(feeds)))
ch <- prometheus.MustNewConstMetric(collector.enabledCount, prometheus.GaugeValue, float64(enabled))
}
func NewFeedCollector(feedService feed.Service) *feedCollector {
return &feedCollector{
feedService: feedService,
totalCount: prometheus.NewDesc(
"autobrr_feed_total",
"Number of feeds",
nil,
nil,
),
enabledCount: prometheus.NewDesc(
"autobrr_feed_enabled_total",
"Number of enabled feeds",
nil,
nil,
),
LastRunTimestamp: prometheus.NewDesc(
"autobrr_feed_last_run_timestamp_seconds",
"The timestamp of the last feed run",
[]string{"feed"},
nil,
),
NextRunTimestamp: prometheus.NewDesc(
"autobrr_feed_next_run_timestamp_seconds",
"The timestamp of the next feed run",
[]string{"feed"},
nil,
),
errorMetric: prometheus.NewDesc(
"autobrr_feed_collector_error",
"Error while collecting feed metrics",
nil,
nil,
),
}
}