Function wsEchoEndpoint

connection/connection_test.go:133–158

A mock origin that echos data by streaming like a tcpOverWSConnection https://github.com/cloudflare/cloudflared/blob/master/ingress/origin_connection.go

(w ResponseWriter, r *http.Request)

// A mock origin that echos data by streaming like a tcpOverWSConnection
// https://github.com/cloudflare/cloudflared/blob/master/ingress/origin_connection.go
func wsEchoEndpoint(w ResponseWriter, r *http.Request) error {
	resp := &http.Response{
		StatusCode: http.StatusSwitchingProtocols,
	}
	if err := w.WriteRespHeaders(resp.StatusCode, resp.Header); err != nil {
		return err
	}
	wsCtx, cancel := context.WithCancel(r.Context())
	readPipe, writePipe := io.Pipe()

	wsConn := websocket.NewConn(wsCtx, NewHTTPResponseReadWriterAcker(w, w.(http.Flusher), r), &log)
	go func() {
		select {
		case <-wsCtx.Done():
		case <-r.Context().Done():
		}
		_ = readPipe.Close()
		_ = writePipe.Close()
	}()

	originConn := &echoPipe{reader: readPipe, writer: writePipe}
	stream.Pipe(wsConn, originConn, &log)
	cancel()
	wsConn.Close()
	return nil
}

type flakyConn struct {
	closeAt time.Time