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

Golang如何优雅的退出程序

article 2024/9/24 19:20:21

Golang如何优雅的退出程序

在 Go 中优雅地退出程序，通常需要处理一些清理工作，如关闭文件、网络连接、释放资源等。以下是一些常见的方法：

一、使用 os.Signal 和 signal.Notify

捕获系统信号：可以使用 os/signal 包来捕获中断信号（如 SIGINT 或 SIGTERM）并执行清理工作。
实现清理逻辑：在接收到信号时执行必要的清理操作。

以下是一个简单的示例：

package main

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

func main() {
	// 创建一个通道来接收信号
	sigs := make(chan os.Signal, 1)

	// 注册要接收的信号
	signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)

	// 创建一个通道来指示退出
	done := make(chan bool, 1)

	// 启动一个 goroutine 来处理信号
	go func() {
		sig := <-sigs
		fmt.Println()
		fmt.Println("收到信号:", sig)
		// 执行清理工作
		cleanup()
		// 通知主 goroutine 可以退出了
		done <- true
	}()

	fmt.Println("等待信号...")
	// 主 goroutine 等待退出信号
	<-done
	fmt.Println("程序退出")
}

func cleanup() {
	fmt.Println("执行清理工作...")
	// 在这里添加清理逻辑
	time.Sleep(2 * time.Second) // 模拟清理工作
	fmt.Println("清理完成")
}

二、使用 context.Context

在需要控制多个 goroutine 的情况下，使用 context.Context 可以更方便地管理他们的生命周期。

package main

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

func main() {
	ctx, cancel := context.WithCancel(context.Background())

	// 捕获信号
	sigs := make(chan os.Signal, 1)
	signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)

	go func() {
		<-sigs
		fmt.Println("\n收到退出信号")
		cancel() // 取消 context，通知所有 goroutine 停止
	}()

	// 启动一些需要管理生命周期的 goroutine
	go worker(ctx, "worker1")
	go worker(ctx, "worker2")

	// 阻塞，直到 context 被取消
	<-ctx.Done()
	fmt.Println("程序退出")
}

func worker(ctx context.Context, name string) {
	for {
		select {
		case <-ctx.Done():
			fmt.Printf("%s 停止\n", name)
			return
		default:
			fmt.Printf("%s 工作中\n", name)
			time.Sleep(1 * time.Second)
		}
	}
}