github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-07-27 21:47:07 +00:00
Code Issues Projects Releases Wiki Activity

add ability to close connection for serve-host

Browse Source
This commit is contained in:
Minhan Xia 2018-11-06 13:27:22 -08:00
parent e26f5d19d4
commit 0fb2da902b
2 changed files with 12 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test/images/serve-hostname/VERSION
View File

@ -1 +1 @@
1.1 1.2

7
test/images/serve-hostname/serve_hostname.go
View File

@ -33,6 +33,7 @@ var (
doTCP = flag.Bool("tcp", false, "Serve raw over TCP.") doTCP = flag.Bool("tcp", false, "Serve raw over TCP.")
doUDP = flag.Bool("udp", false, "Serve raw over UDP.") doUDP = flag.Bool("udp", false, "Serve raw over UDP.")
doHTTP = flag.Bool("http", true, "Serve HTTP.") doHTTP = flag.Bool("http", true, "Serve HTTP.")
doClose = flag.Bool("close", false, "Close connection per each HTTP request")
port = flag.Int("port", 9376, "Port number.") port = flag.Int("port", 9376, "Port number.")
) )
@ -88,6 +89,12 @@ func main() {
if *doHTTP { if *doHTTP {
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) { http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
log.Printf("HTTP request from %s", r.RemoteAddr) log.Printf("HTTP request from %s", r.RemoteAddr)
if *doClose {
// Add this header to force to close the connection after serving the request.
w.Header().Add("Connection", "close")
}
fmt.Fprintf(w, "%s", hostname) fmt.Fprintf(w, "%s", hostname)
}) })
go func() { go func() {