github
/
k3s
mirror of https://github.com/k3s-io/k3s
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Merge pull request #70714 from freehan/serve-hostname-close 

add ability to close connection for serve-host
pull/564/head
k8s-ci-robot 2018-11-29 02:01:23 -08:00 committed by GitHub
parent d0e8e6b78f 0fb2da902b
commit 9f75e03481
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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

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

@ -30,10 +30,11 @@ import (
)
var (
doTCP = flag.Bool("tcp", false, "Serve raw over TCP.")
doUDP = flag.Bool("udp", false, "Serve raw over UDP.")
doHTTP = flag.Bool("http", true, "Serve HTTP.")
port = flag.Int("port", 9376, "Port number.")
doTCP = flag.Bool("tcp", false, "Serve raw over TCP.")
doUDP = flag.Bool("udp", false, "Serve raw over UDP.")
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.")
)
func main() {
@ -88,6 +89,12 @@ func main() {
if *doHTTP {
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
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)
})
go func() {