Function createWebsocketHandler

e2e/websocket_test.go:21–58

(opts wsHandlerOpts)

}

func createWebsocketHandler(opts wsHandlerOpts) http.Handler {
	mux := http.NewServeMux()

	mux.HandleFunc("/empty", func(w http.ResponseWriter, r *http.Request) {
	})

	mux.HandleFunc("/test", func(w http.ResponseWriter, r *http.Request) {
		upgrader := &websocket.Upgrader{}

		hdr := make(http.Header)
		hdr["X-Test"] = []string{"test_header"}

		c, err := upgrader.Upgrade(w, r, hdr)
		if err != nil {
			panic(err)
		}
		defer c.Close()

		for {
			if opts.preRead != nil {
				opts.preRead()
			}
			mt, message, err := c.ReadMessage()
			if err != nil {
				break
			}
			if opts.preWrite != nil {
				opts.preWrite()
			}
			err = c.WriteMessage(mt, message)
			if err != nil {
				break
			}
		}
	})

	return mux
}

func websocketFastHandler(ctx *fasthttp.RequestCtx) {
	var upgrader fastwebsocket.FastHTTPUpgrader