Merge pull request #1095 from Entrivax/totalDurationMetrics

feat(metrics): add total and per channel vods duration

Author: Zibbp

internal/metrics/metrics.go

@@ -25,6 +25,8 @@ type Metrics struct {
 	totalUsers               prometheus.Gauge
 	totalLiveWatchedChannels prometheus.Gauge
 	channelVodCount          *prometheus.GaugeVec
+	totalVodsDuration        prometheus.Gauge
+	channelVodDuration       *prometheus.GaugeVec
 	totalVodsInQueue         prometheus.Gauge
 	riverTotalPendingJobs    prometheus.Gauge
 	riverTotalScheduledJobs  prometheus.Gauge
@@ -59,6 +61,14 @@ func NewService(store *database.Database, riverClient *tasks_client.RiverClient)
 			Name: "channel_vod_count",
 			Help: "Number of vods per channel",
 		}, []string{"channel"}),
+		totalVodsDuration: prometheus.NewGauge(prometheus.GaugeOpts{
+			Name: "total_vods_duration_seconds",
+			Help: "Total duration of all VODs in seconds",
+		}),
+		channelVodDuration: prometheus.NewGaugeVec(prometheus.GaugeOpts{
+			Name: "channel_vod_duration_seconds",
+			Help: "Total duration of VODs per channel in seconds",
+		}, []string{"channel"}),
 		totalVodsInQueue: prometheus.NewGauge(prometheus.GaugeOpts{
 			Name: "total_vods_in_queue",
 			Help: "Total number of vods in queue",
@@ -103,6 +113,8 @@ func NewService(store *database.Database, riverClient *tasks_client.RiverClient)
 		metrics.totalUsers,
 		metrics.totalLiveWatchedChannels,
 		metrics.channelVodCount,
+		metrics.totalVodsDuration,
+		metrics.channelVodDuration,
 		metrics.totalVodsInQueue,
 		metrics.riverTotalPendingJobs,
 		metrics.riverTotalScheduledJobs,
@@ -214,7 +226,8 @@ func (s *Service) GatherMetrics() (*prometheus.Registry, error) {
 		s.metrics.totalLiveWatchedChannels.Set(0)
 	}
 	s.metrics.totalLiveWatchedChannels.Set(float64(lwCount))
-	// Get all channels and the number of VODs they have
+	// Get all channels with the number of VODs they have and their total duration
+	var totalDurationSeconds int64 = 0
 	channels, err := s.Store.Client.Channel.Query().WithVods().All(context.Background())
 	if err != nil {
 		log.Error().Err(err).Msg("error getting all channels")
@@ -223,7 +236,14 @@ func (s *Service) GatherMetrics() (*prometheus.Registry, error) {
 	for _, channel := range channels {
 		cVCount := len(channel.Edges.Vods)
 		s.metrics.channelVodCount.With(prometheus.Labels{"channel": channel.Name}).Set(float64(cVCount))
+		var channelDuration int64 = 0
+		for _, vod := range channel.Edges.Vods {
+			channelDuration += int64(vod.Duration)
+		}
+		s.metrics.channelVodDuration.With(prometheus.Labels{"channel": channel.Name}).Set(float64(channelDuration))
+		totalDurationSeconds += channelDuration
 	}
+	s.metrics.totalVodsDuration.Set(float64(totalDurationSeconds))
 	// Total VODs in queue
 	qCount, err := s.Store.Client.Queue.Query().Where(queue.Processing(true)).Count(context.Background())
 	if err != nil {