[ENG-2343] Stream attempts metric (AI Gateway side)

livepeer · Jan 13, 2025 · 06eba68 · 06eba68
1 parent 4d09ba7
commit 06eba68
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/server/ai_http.go b/server/ai_http.go
@@ -95,10 +95,6 @@ func aiHttpHandle[I any](h *lphttp, decoderFunc func(*I, *http.Request) error) h
 
 func (h *lphttp) StartLiveVideoToVideo() http.Handler {
 	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
-		if monitor.Enabled {
-			monitor.AILiveVideoAttempt()
-		}
-
 		remoteAddr := getRemoteAddr(r)
 		ctx := clog.AddVal(r.Context(), clog.ClientIP, remoteAddr)
 		requestID := string(core.RandomManifestID())

diff --git a/server/ai_mediaserver.go b/server/ai_mediaserver.go
@@ -439,6 +439,7 @@ func (ls *LivepeerServer) StartLiveVideo() http.Handler {
 				return
 			}
 		}
+
 		mediaMTXClient := media.NewMediaMTXClient(remoteHost, ls.mediaMTXApiPassword, sourceID, sourceType)
 
 		if LiveAIAuthWebhookURL != nil {
@@ -488,6 +489,12 @@ func (ls *LivepeerServer) StartLiveVideo() http.Handler {
 		ctx = clog.AddVal(ctx, "stream_id", streamID)
 		clog.Infof(ctx, "Received live video AI request for %s. pipelineParams=%v", streamName, pipelineParams)
 
+		// Count `ai_live_attempts` after successful parameters validation
+		clog.V(common.VERBOSE).Infof(ctx, "AI Live video attempt")
+		if monitor.Enabled {
+			monitor.AILiveVideoAttempt()
+		}
+
 		// Kick off the RTMP pull and segmentation as soon as possible
 		ssr := media.NewSwitchableSegmentReader()
 		go func() {