要传递什么上下文来等待某件事完成？

学习知识要善于思考，思考，再思考！今天golang学习网小编就给大家带来《要传递什么上下文来等待某件事完成？》，以下内容主要包含等知识点，如果你正在学习或准备学习Golang，就都不要错过本文啦~让我们一起来看看吧，能帮助到你就更好了！

由于种种原因，我觉得http.listenandserve不适合我的需求。

我需要能够确定绑定的地址和端口（即使用":0"时），所以我引入了一个net.listener，读取listener.addr()然后传递给http.serve(听众，无）。

然后我需要能够使用不同的 url 处理程序运行两个 http 服务器，因此我引入了 http.newservemux()，添加了必要的 mux.handlefunc("/path", fn) 处理程序，并传递为http.serve(监听器，多路复用器)。

然后我需要能够干净地停止这些服务器，并关闭任何连接，独立于主程序本身，所以现在我引入了 &http.server{handler: mux} ，我可以 go func() { server.serve(监听器) }()。

理论上，我可以通过调用 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 完成的正确、干净、惯用的方法是什么？

解决方案

要等待主 goroutine 中的 shutdown 完成，请从该 goroutine 中调用 shutdown。消除通道和额外的 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() 替换为按截止日期创建的上下文。

理论要掌握，实操不能落！以上关于《要传递什么上下文来等待某件事完成？》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！