golang中怎么优雅地关闭http服务
更新时间：2023-08-01
前言
在使用Golang开发HTTP服务时，我们经常需要手动地关闭HTTP服务器。关闭HTTP服务器的目的是为了优雅地停止正在运行的服务器，同时确保已经建立的连接能够正常关闭，避免因为突然关闭导致网络异常或者资源的泄露。本文将介绍如何优雅地关闭Golang中的HTTP服务器。
关闭HTTP服务器的常见方式
在Golang中，关闭HTTP服务器的常见方式有两种：使用http.Server的Shutdown方法和使用os.Signal信号。下面将分别介绍这两种方式的具体实现。
1. 使用http.Server的Shutdown方法
使用http.Server的Shutdown方法可以平滑地关闭HTTP服务器，让服务器先停止接收新的请求，等待所有已经建立的连接结束后再关闭。
```go
// 创建HTTP服务器
server := &http.Server{
Addr: ":8080",
// ...
}
// 启动HTTP服务器
go func() {
if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
log.Fatalf("server started failed: %s", err)
}
}()
// 等待接收到系统中断信号
sigCh := make(chan os.Signal, 1)
signal.Notify(sigCh, syscall.SIGINT, syscall.SIGTERM)
<-sigCh
// 关闭HTTP服务器
log.Println("Shutting down server...")
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()
if err := server.Shutdown(ctx); err != nil {
log.Fatalf("server shutdown failed: %s", err)
}
log.Println("Server gracefully stopped")
```
2. 使用os.Signal信号
另一种常见的关闭HTTP服务器的方式是使用os.Signal来捕获系统中断信号并处理关闭逻辑。
```go
// 创建HTTP服务器
server := &http.Server{
Addr: ":8080",
// ...
}
// 启动HTTP服务器
go func() {
if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
log.Fatalf("server started failed: %s", err)
}
}()
// 等待接收到系统中断信号
sigCh := make(chan os.Signal, 1)
signal.Notify(sigCh, syscall.SIGINT, syscall.SIGTERM)
<-sigCh
// 关闭HTTP服务器
log.Println("Shutting down server...")
if err := server.Close(); err != nil {
log.Fatalf("server shutdown failed: %s", err)
}
log.Println("Server gracefully stopped")
```
总结
关闭HTTP服务器是Golang中常见的任务之一，为了保证服务器能够平滑地停止，需要使用合适的方法来关闭服务器。本文介绍了两种常见的关闭HTTP服务器的方式：使用http.Server的Shutdown方法和使用os.Signal信号。前者通过等待已经建立的连接结束后再关闭服务器，后者通过捕获系统中断信号来关闭服务器。根据实际需求选择合适的方式来关闭HTTP服务器，以确保服务器的安全稳定运行。
