fix: notify about number of listeners immediately

Author: cheatsnake

internal/http/handlers.go

@@ -44,6 +44,11 @@ func (s *Server) handleEvents(w http.ResponseWriter, r *http.Request) {
 		close(eventChan)
 	}()
 
+	// Send current number of listeners immediately
+	countEvent := s.countListeners()
+	fmt.Fprint(w, countEvent.Stringify())
+	w.(http.Flusher).Flush()
+
 	for {
 		event, isOpen := <-eventChan
 		if !isOpen {

internal/http/server.go

@@ -114,15 +114,20 @@ func (s *Server) registerMP2TMimeType() {
 	}
 }
 
+func (s *Server) countListeners() *sse.Event {
+	count := s.eventsEmitter.CountSubscribers()
+	return sse.NewEvent(eventCountListeners, strconv.Itoa(count))
+}
+
 func (s *Server) listenEvents() {
 	countConnectionTicker := time.Tick(5 * time.Second)
 
 	// TODO: add context for gracefull shutdown
 
 	go func() {
 		for range countConnectionTicker {
-			count := s.eventsEmitter.CountSubscribers()
-			s.eventsEmitter.RegisterEvent(eventCountListeners, strconv.Itoa(count))
+			event := s.countListeners()
+			s.eventsEmitter.RegisterEvent(event.Name, event.Data)
 		}
 	}()