Function websocketHandler

hello/hello.go:161–188

This handler will echo message

(log *zerolog.Logger, upgrader websocket.Upgrader)

// This handler will echo message
func websocketHandler(log *zerolog.Logger, upgrader websocket.Upgrader) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		// This addresses the issue of r.Host includes port but origin header doesn't
		host, _, err := net.SplitHostPort(r.Host)
		if err == nil {
			r.Host = host
		}

		conn, err := upgrader.Upgrade(w, r, nil)
		if err != nil {
			log.Err(err).Msg("failed to upgrade to websocket connection")
			return
		}
		defer conn.Close()
		for {
			mt, message, err := conn.ReadMessage()
			if err != nil {
				log.Err(err).Msg("websocket read message error")
				break
			}

			if err := conn.WriteMessage(mt, message); err != nil {
				log.Err(err).Msg("websocket write message error")
				break
			}
		}
	}
}

func sseHandler(log *zerolog.Logger) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {