Commit f6fadc4c authored by Jeromy's avatar Jeromy

do http server properly so daemon can shut down

parent 5c94a68a
...@@ -5,6 +5,7 @@ high-level HTTP interfaces to IPFS. ...@@ -5,6 +5,7 @@ high-level HTTP interfaces to IPFS.
package corehttp package corehttp
import ( import (
"net"
"net/http" "net/http"
"time" "time"
...@@ -56,12 +57,15 @@ func ListenAndServe(n *core.IpfsNode, listeningMultiAddr string, options ...Serv ...@@ -56,12 +57,15 @@ func ListenAndServe(n *core.IpfsNode, listeningMultiAddr string, options ...Serv
} }
func listenAndServe(node *core.IpfsNode, addr ma.Multiaddr, handler http.Handler) error { func listenAndServe(node *core.IpfsNode, addr ma.Multiaddr, handler http.Handler) error {
_, host, err := manet.DialArgs(addr) netarg, host, err := manet.DialArgs(addr)
if err != nil { if err != nil {
return err return err
} }
server := &http.Server{Addr: host, Handler: handler} list, err := net.Listen(netarg, host)
if err != nil {
return err
}
// if the server exits beforehand // if the server exits beforehand
var serverError error var serverError error
...@@ -71,7 +75,7 @@ func listenAndServe(node *core.IpfsNode, addr ma.Multiaddr, handler http.Handler ...@@ -71,7 +75,7 @@ func listenAndServe(node *core.IpfsNode, addr ma.Multiaddr, handler http.Handler
defer node.Children().Done() defer node.Children().Done()
go func() { go func() {
serverError = server.ListenAndServe() serverError = http.Serve(list, handler)
close(serverExited) close(serverExited)
}() }()
...@@ -83,14 +87,15 @@ func listenAndServe(node *core.IpfsNode, addr ma.Multiaddr, handler http.Handler ...@@ -83,14 +87,15 @@ func listenAndServe(node *core.IpfsNode, addr ma.Multiaddr, handler http.Handler
case <-node.Closing(): case <-node.Closing():
log.Infof("server at %s terminating...", addr) log.Infof("server at %s terminating...", addr)
// make sure keep-alive connections do not keep the server running list.Close()
server.SetKeepAlivesEnabled(false)
outer: outer:
for { for {
// wait until server exits // wait until server exits
select { select {
case <-serverExited: case <-serverExited:
// if the server exited as we are closing, we really dont care about errors
serverError = nil
break outer break outer
case <-time.After(5 * time.Second): case <-time.After(5 * time.Second):
log.Infof("waiting for server at %s to terminate...", addr) log.Infof("waiting for server at %s to terminate...", addr)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment