Go | 守护线程

守护线程

“Go 守护线程”通常指在 Go 语言中持续运行在后台执行特定任务，并且在主程序结束时不会自动结束的线程或协程。

在 Go 语言中，可以通过 go 关键字启动一个协程来实现类似守护线程的功能。

例如：

package main

import (
    "fmt"
    "time"
)

func daemonTask() {
    for {
        // 执行守护任务的逻辑
        fmt.Println("Running daemon task...")
        time.Sleep(5 * time.Second)
    }
}

func main() {
    go daemonTask()
    // 主程序的其他逻辑
    fmt.Println("Main program is doing something...")
    time.Sleep(10 * time.Second)
}

在上述示例中，daemonTask 函数中的代码会在后台持续运行，每隔 5 秒打印一条消息。

关闭守护线程

在 Go 语言中，如果要终止一个类似守护线程的协程，常见的方式是通过通道来进行控制。

以下是一个示例代码，展示如何通过通道来终止一个协程：

package main

import (
    "fmt"
    "time"
)

func daemonTask(stopCh chan struct{}) {
    for {
        select {
        case <-stopCh:
            fmt.Println("Daemon task stopped.")
            return
        default:
            // 执行守护任务的逻辑
            fmt.Println("Running daemon task...")
            time.Sleep(5 * time.Second)
        }
    }
}

func main() {
    stopCh := make(chan struct{})
    go daemonTask(stopCh)
    // 主程序的其他逻辑
    fmt.Println("Main program is doing something...")
    time.Sleep(10 * time.Second)
    // 发送停止信号
    close(stopCh)
}

在上述示例中，daemonTask 函数通过 select 语句监听 stopCh 通道。当 stopCh 被关闭时，协程会收到信号并终止执行。在 main 函数中，经过一定时间后通过 close(stopCh) 来关闭通道，通知协程终止。