golang http 停止后怎么清理内部服务

2024年4月10日

在Go语言中，实现HTTP服务的优雅重启或停止并清理内部服务，通常涉及以下几个步骤：

监听信号：首先，你的服务需要监听操作系统发出的信号，如SIGINT（通常是由Ctrl+C触发）或SIGTERM（通常由系统或服务管理器发送）。
准备关闭：当收到这些信号时，你的服务应该开始准备关闭过程。这可能包括拒绝新的连接、完成正在处理的请求等。
关闭监听器：关闭HTTP服务器监听的端口。这一步通常是通过调用http.Server的Close方法或net.Listener的Close方法完成的。
等待请求完成：在关闭监听器之后，你的服务应该等待所有正在处理的请求完成。这可以通过同步机制（如sync.WaitGroup）来实现。
清理资源：在所有请求处理完毕后，你的服务应该清理所有打开的资源，如数据库连接、文件句柄等。
退出：最后，服务可以安全地退出。

下面是一个简单的示例代码，展示了如何实现这个过程：

package main

import (
    "context"
    "fmt"
    "net/http"
    "os"
    "os/signal"
    "syscall"
    "time"
)

func main() {
    mux := http.NewServeMux()
    mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
        fmt.Fprintf(w, "Hello, World!")
    })

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

    // 设置一个通道来接收信号
    sigChan := make(chan os.Signal, 1)
    signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM)

    // 启动服务器
    go func() {
        if err := server.ListenAndServe(); err != nil {
            fmt.Printf("Server shutdown: %v\n", err)
        }
    }()

    // 阻塞，直到收到信号
    sig := <-sigChan
    fmt.Printf("Got %s signal. Shutting down...\n", sig)

    // 创建一个上下文，并设置超时时间
    ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
    defer cancel()

    // 关闭服务器
    if err := server.Shutdown(ctx); err != nil {
        fmt.Printf("Server forced shutdown: %v\n", err)
    } else {
        fmt.Println("Server gracefully shut down.")
    }

    // 清理资源（如果有的话）
    // ...

    os.Exit(0)
}

package main

import (

"context"

"fmt"

"net/http"

"os"

"os/signal"

"syscall"

"time"

)

func main() {

mux := http.NewServeMux()

mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {

fmt.Fprintf(w, "Hello, World!")

})

server := &http.Server{

Addr: ":8080",

Handler: mux,

}

// 设置一个通道来接收信号

sigChan := make(chan os.Signal, 1)

signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM)

// 启动服务器

go func() {

if err := server.ListenAndServe(); err != nil {

fmt.Printf("Server shutdown: %v\n", err)

}

}()

// 阻塞，直到收到信号

sig := <-sigChan

fmt.Printf("Got %s signal. Shutting down...\n", sig)

// 创建一个上下文，并设置超时时间

ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)

defer cancel()

// 关闭服务器

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

fmt.Printf("Server forced shutdown: %v\n", err)

} else {

fmt.Println("Server gracefully shut down.")

}

// 清理资源（如果有的话）

// ...

os.Exit(0)

}

在这个示例中，我们使用了signal.Notify来监听SIGINT和SIGTERM信号。当收到这些信号时，我们启动服务器的关闭过程。通过调用server.Shutdown(ctx)，我们告诉服务器优雅地关闭，这意味着它会等待当前正在处理的请求完成。在服务器关闭之后，我们可以执行任何必要的资源清理操作。

请注意，优雅地关闭HTTP服务器并不总是可能的，特别是当服务器遇到无法恢复的错误时。在这种情况下，你可能需要实现一种强制关闭的机制，以确保资源得到释放。