[ENG-2342] unit and fixes

livepeer · Jan 8, 2025 · de6fb68 · de6fb68
1 parent c8cbe79
commit de6fb68
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/monitor/census.go b/monitor/census.go
@@ -376,7 +376,7 @@ func InitCensus(nodeType NodeType, version string) {
 	census.mAIResultUploadTime = stats.Float64("ai_result_upload_time_seconds", "Upload (to Orchestrator) time", "sec")
 	census.mAIResultSaveFailed = stats.Int64("ai_result_upload_failed_total", "AIResultUploadFailed", "tot")
 	census.mAICurrentLivePipelines = stats.Int64("ai_current_live_pipelines", "Number of live AI pipelines currently running", "tot")
-	census.mAIFirstSegmentDelay = stats.Int64("ai_first_segment_delay", "Delay of the first live AI segment being processed", "ms")
+	census.mAIFirstSegmentDelay = stats.Int64("ai_first_segment_delay_ms", "Delay of the first live AI segment being processed", "ms")
 
 	glog.Infof("Compiler: %s Arch %s OS %s Go version %s", runtime.Compiler, runtime.GOARCH, runtime.GOOS, runtime.Version())
 	glog.Infof("Livepeer version: %s", version)
@@ -985,7 +985,7 @@ func InitCensus(nodeType NodeType, version string) {
 			Aggregation: view.LastValue(),
 		},
 		{
-			Name:        "ai_first_segment_delay",
+			Name:        "ai_first_segment_delay_ms",
 			Measure:     census.mAIFirstSegmentDelay,
 			Description: "Delay of the first live AI segment being processed",
 			TagKeys:     baseTags,

diff --git a/server/ai_live_video.go b/server/ai_live_video.go
@@ -137,7 +137,7 @@ func startTrickleSubscribe(ctx context.Context, url *url.URL, params aiRequestPa
 	// read segments from trickle subscription
 	go func() {
 		var err error
-		firstSegment := false
+		firstSegment := true
 
 		defer w.Close()
 		retries := 0