owncast/webserver/handlers/playbackMetrics.go

package handlers

import (
	"encoding/json"
	"net/http"

	"github.com/owncast/owncast/services/metrics"
	"github.com/owncast/owncast/utils"
	"github.com/owncast/owncast/webserver/responses"
	log "github.com/sirupsen/logrus"
)

// ReportPlaybackMetrics will accept playback metrics from a client and save
// them for future video health reporting.
func (h *Handlers) ReportPlaybackMetrics(w http.ResponseWriter, r *http.Request) {
	if r.Method != http.MethodPost {
		responses.WriteSimpleResponse(w, false, r.Method+" not supported")
		return
	}

	type reportPlaybackMetricsRequest struct {
		Bandwidth             float64 `json:"bandwidth"`
		Latency               float64 `json:"latency"`
		Errors                float64 `json:"errors"`
		DownloadDuration      float64 `json:"downloadDuration"`
		QualityVariantChanges float64 `json:"qualityVariantChanges"`
	}

	decoder := json.NewDecoder(r.Body)
	var request reportPlaybackMetricsRequest
	if err := decoder.Decode(&request); err != nil {
		log.Errorln("error decoding playback metrics payload:", err)
		responses.WriteSimpleResponse(w, false, err.Error())
		return
	}

	clientID := utils.GenerateClientIDFromRequest(r)

	metrics.RegisterPlaybackErrorCount(clientID, request.Errors)
	if request.Bandwidth != 0.0 {
		metrics.RegisterPlayerBandwidth(clientID, request.Bandwidth)
	}

	if request.Latency != 0.0 {
		metrics.RegisterPlayerLatency(clientID, request.Latency)
	}

	if request.DownloadDuration != 0.0 {
		metrics.RegisterPlayerSegmentDownloadDuration(clientID, request.DownloadDuration)
	}

	metrics.RegisterQualityVariantChangesCount(clientID, request.QualityVariantChanges)
}