go - 传递什么上下文来等待某事完成？

Question

由于种种原因，我觉得http.ListenAndServe不适合我的需要。
我需要能够确定绑定(bind)地址和端口(即使用 ":0" 时)，所以我引入了 net.Listener , 阅读 listener.Addr()然后传递给 http.Serve(listener, nil) .
然后我需要能够运行两个具有不同 URL 处理程序的 HTTP 服务器，所以我引入了一个 http.NewServeMux() ，添加了必要的 mux.HandleFunc("/path", fn)处理程序，并作为 http.Serve(listener, mux) 传递.
然后我需要能够干净地停止这些服务器，并关闭所有连接，独立于主程序本身，所以现在我介绍了&http.Server{Handler: mux}我可以go func() { server.Serve(listener) }() .
理论上，我可以通过调用 server.Shutdown(ctx) 来阻止它。 ，但现在 import "context" 中没有可用的上下文似乎也提供了我想要的东西。我希望能够等到干净关闭完成，然后继续我的代码。
我的理解是我应该能够<- ctx.Done()来实现这一点，但我都尝试过 context.Background()和 context.TODO()似乎都没有“触发”ctx.Done() ，我最终永远阻塞。其他context选项似乎是基于时间的。
如果我不等待，或者通过 nil , server.Shutdown(ctx)似乎完成得太快了，我看不到任何东西实际上是关闭的(runtime.Numgoroutine() != 1)
我可以time.Sleep(duration)一段任意的持续时间，但我不想要任意的持续时间。我想知道server.Shutdown已经干净地完成了。

package main

import (
    "fmt"
    "net"
    "net/http"
    "runtime"
    "time"
)

func main() {
    var err error

    listener, err := net.Listen("tcp", "localhost:0")
    fmt.Printf("Listening on http://%v\n", listener.Addr())

    mux := http.NewServeMux()
    mux.HandleFunc("/", handleIndex)

    stop, err := startHTTPServer(listener, mux)

    d, _ := time.ParseDuration("5s")

    time.Sleep(d)   // delay here just for example of "long-running" server
    close(stop)     // closing the channel returned by my helper should trigger shutdown
    time.Sleep(d)   // if this delay is here, I see the "Stopped" message

    if err != nil {
        panic(err)
    }

    fmt.Printf("End of program, active goroutines: %v", runtime.NumGoroutine())
}

// startHTTPServer is a helper function to start a server and return a channel that can trigger shutdown
func startHTTPServer(listener net.Listener, handler http.Handler) (stop chan struct{}, err error) {
    stop = make(chan struct{})
    server := &http.Server{Handler: handler}

    go func() {
        fmt.Println("Starting server...")
        err = server.Serve(listener)
    }()
    go func() {
        select {
        case <-stop:
            fmt.Println("Stop channel closed; stopping server...")
            err = server.Shutdown(nil)    // what is passed instead of nil here?
            fmt.Println("Stopped.")
            return
        }
    }()
    return
}

func handleIndex(w http.ResponseWriter, r *http.Request) {
    fmt.Fprintln(w, "Hello world")
}

我都试过了context.Background()和 context.TODO() .我试过new(context.Context)但这抛出了SIGSEGV .我试过nil这根本不需要等待。
我尝试添加 sync.WaitGroup并调用 wg.Wait()而不是第二个 time.Sleep(d) ，但我还是要等到server.Shutdown()在调用 wg.Done() 之前完成(而 defer wg.Done() 称它为时过早)。
我觉得，对于 Contexts、WaitGroups 等，我只是在代码中添加了一些杂乱无章的东西，而没有真正理解为什么它们是必要的。
等待 server.Shutdown 的正确、干净、惯用的方法是什么？去完成？

Answer 1

要在主 goroutine 中等待 Shutdown 完成，请从该 goroutine 调用 Shutdown。消除 channel 和额外的 goroutine。

listener, _ := net.Listen("tcp", "localhost:0")
fmt.Printf("Listening on http://%v\n", listener.Addr())

mux := http.NewServeMux()
mux.HandleFunc("/", handleIndex)

server := &http.Server{Handler: mux}
go func() {
    fmt.Println("Starting server...")
    err := server.Serve(listener)
    if err != nil && err != http.ErrServerClosed {
        log.Fatal(err)
    }
}()

time.Sleep(5 * time.Second) // delay here just for example of "long-running" server

// Shutdown and wait for server to complete.
server.Shutdown(context.Background())

如果要限制 Shutdown 等待服务器关闭的时间，请替换 context.Background()使用截止日期创建的上下文。