vane 0.9.2

A flow-based reverse proxy with multi-layer routing and programmable pipelines.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

/* integration/pkg/mock/handler.go */
package mock

import (
	"io"
	"net/http"
	"strconv"
)

// SmartEchoHandler is a dynamic handler for testing.
// It behaviors is controlled by request headers:
// - X-Test-Status: Sets the response status code (default 200).
// - Body is always echoed back.
// - Content-Type is preserved.
func SmartEchoHandler(w http.ResponseWriter, r *http.Request) {
	// 1. Determine Status Code
	status := http.StatusOK
	if s := r.Header.Get("X-Test-Status"); s != "" {
		if val, err := strconv.Atoi(s); err == nil {
			status = val
		}
	}

	// 2. Set Headers
	w.Header().Set("X-Upstream-Proto", r.Proto)
	if ct := r.Header.Get("Content-Type"); ct != "" {
		w.Header().Set("Content-Type", ct)
	}

	// 3. Write Status
	w.WriteHeader(status)

	// 4. Echo Body (if not 204/304 which imply no body)
	if status != http.StatusNoContent && status != http.StatusNotModified {
		io.Copy(w, r.Body)
	}
}