如何在K8s容器中优雅中断Go语言的无限for循环

要实现仅在f()执行完成后响应终止信号，你可以利用Go的信号监听机制，在每次f()执行完毕后检查是否收到终止信号，具体示例如下：

package main

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

// 你的业务函数，执行时间几秒到1分钟不等
func f() {
    // 这里替换为实际业务逻辑
}

func main() {
    // 创建带缓冲的信号通道，避免信号丢失
    sigChan := make(chan os.Signal, 1)
    // 监听K8s常用的SIGTERM信号和本地测试用的SIGINT信号
    signal.Notify(sigChan, syscall.SIGTERM, syscall.SIGINT)

    for {
        // 先执行完整的f()逻辑
        f()

        // f()执行完成后，检查是否收到终止信号
        select {
        case <-sigChan:
            // 收到信号，退出程序
            return
        default:
            // 无信号，继续下一次循环
        }
    }
}

关键逻辑说明 #

• 信号通道设置缓冲为1，确保在f()执行过程中收到的信号不会被丢弃，会被暂存到通道中。
• 每次循环先执行完f()，再通过select检查信号：如果有终止信号就直接退出，没有则进入下一轮循环。
• 监听SIGTERM是因为K8s容器终止时默认发送该信号，SIGINT用于本地通过Ctrl+C测试终止逻辑。

内容的提问来源于stack exchange，提问作者BAE