怎么在Golang中优雅地关闭HTTP服务器

寻技术 Go编程 2023年07月12日 97

这篇“怎么在Golang中优雅地关闭HTTP服务器”文章的知识点大部分人都不太理解,所以小编给大家总结了以下内容,内容详细,步骤清晰,具有一定的借鉴价值,希望大家阅读完这篇文章能有所收获,下面我们一起来看看这篇“怎么在Golang中优雅地关闭HTTP服务器”文章吧。

  1. HTTP 服务器的关闭方式

在 Golang 中,我们通常使用 http.ListenAndServe() 函数来启动 HTTP 服务器。该函数会一直运行直到程序被杀死或服务器关闭。如果我们直接关闭程序,HTTP 服务器就会立即停止运行。但是,这种方式并不是一个优雅的关闭方式,因为正在处理的客户端会被强制断开连接,这可能会导致数据丢失或异常。

另外一种关闭方式是使用服务端上下文(context)的 Done() 方法。当我们调用 Done() 方法时,HTTP 服务器会正确关闭并停止接受新的客户端连接,但仍会处理一些已经接受了的连接。这样虽然不会强制断开正在处理的客户端的连接,但仍可能会导致某些请求无响应的问题。

最后,我们还有一种优雅的关闭方式,即使用 http.Server 结构体中的 Shutdown() 方法。该方法会优雅地关闭 HTTP 服务器,即等待正在处理的请求处理完成后再关闭服务器。

  1. 实现 HTTP 服务器的优雅关闭

假设我们已经编写了以下代码用于启动 HTTP 服务器:

package main

import (
    "log"
    "net/http"
)

func main() {
    http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
        w.Write([]byte("Hello World!"))
    })

    if err := http.ListenAndServe(":8080", nil); err != nil {
        log.Fatal(err)
    }
}

现在我们需要在程序退出时正确关闭 HTTP 服务器。我们可以采用以下步骤:

1) 创建一个 http.Server 实例,并将该实例的 Handler 属性设置为我们现有的 Handler 函数。

server := http.Server{
    Addr:    ":8080",
    Handler: http.DefaultServeMux,
}

2) 使用一个 goroutine 来启动 http.Server 的 ListenAndServe() 方法,以处理客户端连接请求。

go func() {
    if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
        log.Fatal(err)
    }
}()

3) 在我们的程序退出时,使用 http.Server 的 Shutdown() 方法来安全地关闭服务器。

// 用于停止 HTTP 服务器的通道
stopChan := make(chan os.Signal)

// 监听 Interrupt 和 Terminate 信号
signal.Notify(stopChan, os.Interrupt, syscall.SIGTERM)

// 等待接收到信号,执行 Shutdown()
<-stopChan
log.Println("Shutting down server...")

// 等待处理完成的请求处理完成
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()

if err := server.Shutdown(ctx); err != nil {
    log.Fatal(err)
}

log.Println("Server gracefully stopped.")

这段代码中我们首先创建了一个通道,用于监听操作系统发送的中断或终止信号(比如 Ctrl + C)。然后我们等待接收到信号,执行 Shutdown() 方法。在调用 Shutdown() 方法之前,我们使用 context.WithTimeout() 创建了一个上下文(Context),用于在规定的超时时间内等待正在处理的请求处理完成。待所有请求处理完成后,Shutdown() 方法能够安全地关闭 HTTP 服务器。

关闭

用微信“扫一扫”