Method handleEvents

http/event.go:29–89

handleEvents handles the "GET /events" route. This route provides real-time event notification over Websockets.

(w http.ResponseWriter, r *http.Request)

// handleEvents handles the "GET /events" route. This route provides real-time
// event notification over Websockets.
func (s *Server) handleEvents(w http.ResponseWriter, r *http.Request) {
	websocketConnections.Inc()
	defer websocketConnections.Dec()

	// Upgrade HTTP connection to use websockets.
	conn, err := upgrader.Upgrade(w, r, nil)
	if err != nil {
		LogError(r, err)
		return
	}

	ctx, cancel := context.WithCancel(r.Context())
	r = r.WithContext(ctx)
	conn.SetCloseHandler(func(code int, text string) error {
		cancel()
		return nil
	})

	// We defer the connection close to ensure it is disconnected when we
	// exit this function. This can occur if the HTTP request disconnects or
	// if the subscription from the event service closes.
	defer conn.Close()

	// Ignore all incoming messages.
	go ignoreWebSocketReaders(conn)

	// Subscribe to all events for the current user.
	sub, err := s.EventService.Subscribe(r.Context())
	if err != nil {
		LogError(r, err)
		return
	}
	defer sub.Close()

	// Stream all events to outgoing websocket writer.
	for {
		select {
		case <-r.Context().Done():
			return // disconnect when HTTP connection disconnects

		case event, ok := <-sub.C():
			// If subscription is closed then exit.
			if !ok {
				return
			}

			// Marshal event data to JSON.
			buf, err := json.Marshal(event)
			if err != nil {
				LogError(r, err)
				return
			}

			// Write JSON data out to the websocket connection.
			if err := conn.WriteMessage(websocket.TextMessage, buf); err != nil {
				LogError(r, err)
				return
			}