mirror of
https://github.com/stefanprodan/podinfo.git
synced 2026-03-04 11:00:23 +00:00
198211e20b
Adds an endpoint that does chunk based encoding. The endpoint just stalls and eventually returns the stall time. Similar to the delay endpoint but in a chunked maner. Fixed up the metrics interceptor to wrap ResponseWriter correctly too.
37 lines
752 B
Go
37 lines
752 B
Go
package api
|
|
|
|
import (
|
|
"math/rand"
|
|
"net/http"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/gorilla/mux"
|
|
)
|
|
|
|
func (s *Server) chunkedHandler(w http.ResponseWriter, r *http.Request) {
|
|
vars := mux.Vars(r)
|
|
|
|
delay, err := strconv.Atoi(vars["wait"])
|
|
if err != nil {
|
|
delay = rand.Intn(int(s.config.HttpServerTimeout*time.Second)-10) + 10
|
|
}
|
|
|
|
flusher, ok := w.(http.Flusher)
|
|
if !ok {
|
|
s.ErrorResponse(w, r, "Streaming unsupported!", http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
w.Header().Set("Connection", "Keep-Alive")
|
|
w.Header().Set("Transfer-Encoding", "chunked")
|
|
w.Header().Set("X-Content-Type-Options", "nosniff")
|
|
|
|
flusher.Flush()
|
|
|
|
time.Sleep(time.Duration(delay) * time.Second)
|
|
s.JSONResponse(w, r, map[string]int{"delay": delay})
|
|
|
|
flusher.Flush()
|
|
}
|