如何在 Golang 中处理服务器连接或退出accept()循环

在Golang实战开发的过程中，我们经常会遇到一些这样那样的问题，然后要卡好半天，等问题解决了才发现原来一些细节知识点还是没有掌握好。今天golang学习网就整理分享《如何在 Golang 中处理服务器连接或退出accept()循环》，聊聊，希望可以帮助到正在努力赚钱的你。

我需要在 golang 中实现一个服务器。我正在使用 net 包来实现此目的，但我不明白如何优雅地摆脱 accept 循环。

查看 net 包中的示例：

ln, err := net.listen("tcp", ":8080")
if err != nil {
    // handle error
}
for {
    conn, err := ln.accept()
    if err != nil {
        // handle error
    }
    go handleconnection(conn)
}

我想做更多的事情：

for {
    select {
    case <-done:
        break
    case conn, err := <-ln.Accept():
    if err != nil {
        break
    }
    ...
}

换句话说，我希望能够以某种方式优雅地终止程序。

正确答案

对于任何服务，确保干净关闭的最佳实践是确保服务的每个部分都支持取消。添加 context.Context 支持是实现此目的的推荐方法。

首先，确保您的 net.listener 不会挂在恶意客户端连接上。因此要添加 context.context 支持：

var lc net.listenconfig

ln, err := lc.listen(ctx, "tcp", ":8080")

取消 ctx 将中断此处可能在握手期间阻塞的任何恶意客户端连接。

编辑： 为了确保我们可以在 listen() 调用处于阻塞等待状态时中断该调用，可以利用相同的 ctx 状态并（如之前已删除的答案中所述）在检测到取消事件时关闭连接：

go func() {
    <-ctx.Done()
    log.Println("shutting service down...")
    ln.Close()
}()

工作示例：https://play.golang.org/p/LO1XS4jBQ02

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《如何在 Golang 中处理服务器连接或退出accept()循环》文章吧，也可关注golang学习网公众号了解相关技术文章。