快捷導航

Go語言實現(xiàn)優(yōu)雅關(guān)機和重啟的示例詳解

更新時間：2025年05月12日 08:42:45 作者：Ai?編碼

在Go語言中,實現(xiàn)優(yōu)雅關(guān)機和重啟通常涉及到處理系統(tǒng)信號,并確保在關(guān)閉前完成所有必要的清理工作,下面我們就來看看如何使用http.Server和os/signal包來實現(xiàn)優(yōu)雅關(guān)機和重啟吧

在Go語言中，實現(xiàn)優(yōu)雅關(guān)機和重啟通常涉及到處理系統(tǒng)信號（如SIGINT和SIGTERM），并確保在關(guān)閉前完成所有必要的清理工作。以下是一個示例，展示了如何使用http.Server和os/signal包來實現(xiàn)優(yōu)雅關(guān)機和重啟。

優(yōu)雅關(guān)機的實現(xiàn)

package main

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

func main() {
	// 創(chuàng)建一個HTTP服務(wù)器
	server := &http.Server{Addr: ":8080", Handler: http.DefaultServeMux}

	// 注冊一個簡單的路由
	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
		fmt.Fprintf(w, "Hello, World!")
	})

	// 啟動服務(wù)器
	go func() {
		if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
			log.Fatalf("ListenAndServe error: %v", err)
		}
	}()

	// 創(chuàng)建一個通道來接收系統(tǒng)信號
	quit := make(chan os.Signal, 1)
	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)

	// 等待信號
	<-quit
	log.Println("Shutting down server...")

	// 創(chuàng)建一個帶有超時的上下文
	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()

	// 優(yōu)雅地關(guān)閉服務(wù)器
	if err := server.Shutdown(ctx); err != nil {
		log.Fatalf("Server forced to shutdown: %v", err)
	}

	log.Println("Server exiting")
}

優(yōu)雅重啟的實現(xiàn)

優(yōu)雅重啟通常涉及到在接收到信號后，啟動一個新的服務(wù)器實例，并關(guān)閉舊的實例。以下是一個簡單的示例：

package main

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

func main() {
	// 創(chuàng)建一個HTTP服務(wù)器
	server := &http.Server{Addr: ":8080", Handler: http.DefaultServeMux}

	// 注冊一個簡單的路由
	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
		fmt.Fprintf(w, "Hello, World!")
	})

	// 啟動服務(wù)器
	go func() {
		if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
			log.Fatalf("ListenAndServe error: %v", err)
		}
	}()

	// 創(chuàng)建一個通道來接收系統(tǒng)信號
	quit := make(chan os.Signal, 1)
	signal.Notify(quit, syscall.SIGUSR2)

	// 等待信號
	<-quit
	log.Println("Restarting server...")

	// 創(chuàng)建一個帶有超時的上下文
	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()

	// 優(yōu)雅地關(guān)閉服務(wù)器
	if err := server.Shutdown(ctx); err != nil {
		log.Fatalf("Server forced to shutdown: %v", err)
	}

	// 啟動新的服務(wù)器實例
	cmd := exec.Command(os.Args[0], os.Args[1:]...)
	cmd.Stdout = os.Stdout
	cmd.Stderr = os.Stderr
	if err := cmd.Start(); err != nil {
		log.Fatalf("Failed to restart server: %v", err)
	}

	log.Println("Server restarted")
}