当前位置：首页 > news >正文

Go语言实现优雅关机和重启的示例

news 来源：原创 2025/5/13 8:36:13

在Go语言中，实现优雅关机和重启通常涉及到处理系统信号（如SIGINT和SIGTERM），并确保在关闭前完成所有必要的清理工作。以下是一个示例，展示了如何使用http.Server和os/signal包来实现优雅关机和重启。

优雅关机的实现

package mainimport ("context""fmt""log""net/http""os""os/signal""syscall""time"
)func main() {// 创建一个HTTP服务器server := &http.Server{Addr: ":8080", Handler: http.DefaultServeMux}// 注册一个简单的路由http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {fmt.Fprintf(w, "Hello, World!")})// 启动服务器go func() {if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {log.Fatalf("ListenAndServe error: %v", err)}}()// 创建一个通道来接收系统信号quit := make(chan os.Signal, 1)signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)// 等待信号<-quitlog.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 forced to shutdown: %v", err)}log.Println("Server exiting")
}

优雅重启的实现

优雅重启通常涉及到在接收到信号后，启动一个新的服务器实例，并关闭旧的实例。以下是一个简单的示例：

package mainimport ("context""fmt""log""net/http""os""os/exec""os/signal""syscall""time"
)func main() {// 创建一个HTTP服务器server := &http.Server{Addr: ":8080", Handler: http.DefaultServeMux}// 注册一个简单的路由http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {fmt.Fprintf(w, "Hello, World!")})// 启动服务器go func() {if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {log.Fatalf("ListenAndServe error: %v", err)}}()// 创建一个通道来接收系统信号quit := make(chan os.Signal, 1)signal.Notify(quit, syscall.SIGUSR2)// 等待信号<-quitlog.Println("Restarting server...")// 创建一个带有超时的上下文ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)defer cancel()// 优雅地关闭服务器if err := server.Shutdown(ctx); err != nil {log.Fatalf("Server forced to shutdown: %v", err)}// 启动新的服务器实例cmd := exec.Command(os.Args[0], os.Args[1:]...)cmd.Stdout = os.Stdoutcmd.Stderr = os.Stderrif err := cmd.Start(); err != nil {log.Fatalf("Failed to restart server: %v", err)}log.Println("Server restarted")
}