Method InitiateStream

aibridge/intercept/eventstream/eventstream.go:75–95

InitiateStream initiates the SSE stream by sending headers and starting the ping ticker. This is safe to call multiple times as only the first call has any effect.

(w http.ResponseWriter)

// ping ticker. This is safe to call multiple times as only the first call has
// any effect.
func (s *EventStream) InitiateStream(w http.ResponseWriter) {
	s.initiateOnce.Do(func() {
		s.initiated.Store(true)
		s.logger.Debug(s.ctx, "stream initiated")

		// Send headers for Server-Sent Event stream.
		w.Header().Set("Content-Type", "text/event-stream")
		w.Header().Set("Cache-Control", "no-cache")
		w.Header().Set("Connection", "keep-alive")
		w.Header().Set("X-Accel-Buffering", "no")

		// Send initial flush to ensure connection is established.
		if err := flush(w); err != nil {
			http.Error(w, err.Error(), http.StatusInternalServerError)
			return
		}

		// Start ping ticker.
		s.tick.Reset(pingInterval)
	})
}

// Start handles sending Server-Sent Event to the client.
func (s *EventStream) Start(w http.ResponseWriter, r *http.Request) {