Fix messages sent metric missing initial messages

Spferical · copybara-github · commit a3f9826d0fb5 · 2025-01-27T05:58:12.000-08:00
PiperOrigin-RevId: 720138477
diff --git a/fleetspeak/src/server/https/streaming_message_server.go b/fleetspeak/src/server/https/streaming_message_server.go
@@ -262,6 +262,9 @@ func (s *streamingMessageServer) initialPoll(ctx context.Context, addr net.Addr,
 		return nil, false, makeError(fmt.Sprintf("error writing body: %v", err), http.StatusInternalServerError)
 	}
 	res.Flush()
+	for _, msg := range toSend.Messages {
+		s.fs.StatsCollector().MessageSent(msg)
+	}
 
 	pi.WriteTime = time.Since(st)
 	pi.End = time.Now()

Original file line number	Diff line number	Diff line change
`@@ -262,6 +262,9 @@ func (s *streamingMessageServer) initialPoll(ctx context.Context, addr net.Addr,`
`262`	`262`	`return nil, false, makeError(fmt.Sprintf("error writing body: %v", err), http.StatusInternalServerError)`
`263`	`263`	`}`
`264`	`264`	`res.Flush()`
	`265`	`+ for _, msg := range toSend.Messages {`
	`266`	`+ s.fs.StatsCollector().MessageSent(msg)`
	`267`	`+ }`
`265`	`268`
`266`	`269`	`pi.WriteTime = time.Since(st)`
`267`	`270`	`pi.End = time.Now()`