Golang信号处理实战

package main

import (
	"fmt"
	"os"
	"os/signal"
	"syscall"
)

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

	// 订阅两个信号
	signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM)

	fmt.Println("程序启动，等待信号...")

	sig := <-sigChan // 阻塞等待
	fmt.Println("收到信号：", sig)

	fmt.Println("执行清理逻辑...")
	// 这里做关闭文件、断开连接等操作

	fmt.Println("程序退出")
}

go run main.go
# Ctrl+C 或 kill PID 会触发信号

package main

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

func main() {
	ctx, stop := signal.NotifyContext(context.Background(),
		syscall.SIGINT, syscall.SIGTERM)
	defer stop()

	fmt.Println("程序启动，等待信号...")

	// 模拟业务协程
	go func() {
		for {
			select {
			case <-ctx.Done():
				fmt.Println("业务收到退出信号，清理中...")
				time.Sleep(1 * time.Second)
				fmt.Println("业务清理完成")
				return
			default:
				fmt.Println("业务运行中...")
				time.Sleep(2 * time.Second)
			}
		}
	}()

	<-ctx.Done() // 阻塞，直到信号触发
	fmt.Println("主程序退出")
}

signal.Ignore(syscall.SIGPIPE) // 忽略管道断开

signal.Stop(sigChan) // 取消 channel 的订阅

signal.Notify(sigChan) // 不指定信号时，监听所有信号

package main

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

func main() {
	srv := &http.Server{Addr: ":8080"}

	// 启动 HTTP 服务
	go func() {
		fmt.Println("HTTP 服务启动在 :8080")
		if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
			fmt.Println("HTTP 服务器出错:", err)
		}
	}()

	// 信号监听
	ctx, stop := signal.NotifyContext(context.Background(),
		syscall.SIGINT, syscall.SIGTERM)
	defer stop()

	<-ctx.Done() // 等待信号
	fmt.Println("收到退出信号，正在关闭服务器...")

	// 设置超时的优雅关闭
	shutdownCtx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()

	if err := srv.Shutdown(shutdownCtx); err != nil {
		fmt.Println("服务器关闭错误:", err)
	}

	fmt.Println("服务器已优雅退出")
}