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

package main
 
import (
	"context"
	"log"
	"net/http"
	"os"
	"os/signal"
	"syscall"
	"time"
 
	"github.com/gin-gonic/gin"
)
 
func main() {
	// 创建路由器
	router := gin.Default()
	router.GET("/cook-hotpot", func(c *gin.Context) {
		time.Sleep(5 * time.Second) // 模拟煮火锅需要时间
		c.String(http.StatusOK, "火锅煮好了，可以吃啦！")
	})
 
	// 创建服务器
	srv := &http.Server{
		Addr:    ":8080",
		Handler: router,
	}
 
	// 在单独的goroutine中启动服务器
	go func() {
		if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
			log.Fatalf("服务器监听错误: %s", err)
		}
	}()
 
	// 创建一个通道来接收系统信号
	quit := make(chan os.Signal, 1)
	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM) // 监听Ctrl+C和系统终止信号
 
	// 阻塞等待信号
	<-quit
	log.Println("开始优雅关机...")
 
	// 设置一个5秒超时的上下文
	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()
 
	// 调用Shutdown方法优雅关闭服务器
	if err := srv.Shutdown(ctx); err != nil {
		log.Fatalf("服务器优雅关机失败: %s", err)
	}
 
	log.Println("服务器已安全退出")
}

package main
 
import (
	"log"
	"net/http"
	"time"
 
	"github.com/fvbock/endless"
	"github.com/gin-gonic/gin"
)
 
func main() {
	// 创建路由器
	router := gin.Default()
	router.GET("/greet", func(c *gin.Context) {
		time.Sleep(5 * time.Second) // 模拟处理请求需要时间
		c.String(http.StatusOK, "你好呀，欢迎来到小区！")
	})
 
	// 使用endless启动服务器，支持优雅重启
	if err := endless.ListenAndServe(":8080", router); err != nil {
		log.Fatalf("服务器监听错误: %s", err)
	}
 
	log.Println("服务器已安全退出")
}