curben
/
cloudflared-mirror
mirror of https://github.com/cloudflare/cloudflared.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

TUN-2425: Enable cloudflared to serve multiple Hello World servers by having each of them create its own ServeMux

This commit is contained in:
Chung-Ting Huang 2019-10-22 10:09:59 -05:00
parent e0c2758b64
commit 7133eceb9b
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
hello/hello.go
View File

@ -103,15 +103,16 @@ func StartHelloWorldServer(logger *logrus.Logger, listener net.Listener, shutdow
WriteBufferSize: 1024, WriteBufferSize: 1024,
} }
httpServer := &http.Server{Addr: listener.Addr().String(), Handler: nil} muxer := http.NewServeMux()
muxer.HandleFunc("/uptime", uptimeHandler(time.Now()))
muxer.HandleFunc("/ws", websocketHandler(logger, upgrader))
muxer.HandleFunc("/", rootHandler(serverName))
httpServer := &http.Server{Addr: listener.Addr().String(), Handler: muxer}
go func() { go func() {
<-shutdownC <-shutdownC
httpServer.Close() httpServer.Close()
}() }()
http.HandleFunc("/uptime", uptimeHandler(time.Now()))
http.HandleFunc("/ws", websocketHandler(logger, upgrader))
http.HandleFunc("/", rootHandler(serverName))
err := httpServer.Serve(listener) err := httpServer.Serve(listener)
return err return err
} }