Go语言中利用Channel与Select Default模式高效中断循环

Loop:
   for {
      // 在循环中执行一些快速操作
      // do something repeatedly very fast in the for loop

      // 检查exitMessage通道，判断是否需要退出
      select {
          case <- exitMessage:
               break Loop
          case <- time.After(1 * time.Millisecond): // 引入一个短时间延迟
               // 继续循环
       }
   }

exitFlag := 0

// 启动一个goroutine来监听exitMessage并设置退出标志
go func(in chan int){
    exitFlag = <-in // 收到消息后将exitFlag设为非零值
}(exitMessage)

for exitFlag == 0 {
       // 在循环中执行一些快速操作
       // do something repeatedly very fast in the for loop
 }

Loop:
   for {
      // 检查exitMessage通道，判断是否需要退出
      select {
          case <- exitMessage:
               break Loop // 收到退出信号，跳出循环
          default:
               // 如果没有退出信号，立即执行这里的代码
               // do something repeatedly very fast in the for loop
       }
       // 循环的其他部分，如果"do something"在default之外
   }

package main

import (
    "fmt"
    "sync"
    "time"
)

func worker(exitSignal chan struct{}, wg *sync.WaitGroup) {
    defer wg.Done() // 确保goroutine结束时通知WaitGroup

    fmt.Println("Worker: 开始执行任务...")
    iterations := 0
Loop:
    for {
        iterations++
        // 模拟快速的计算任务
        // fmt.Printf("Worker: 正在执行第 %d 次迭代...\n", iterations) // 如果打印会慢很多

        select {
        case <-exitSignal:
            fmt.Printf("Worker: 收到退出信号，在第 %d 次迭代后停止。\n", iterations)
            break Loop // 收到信号，跳出for循环
        default:
            // 没有退出信号，继续执行循环体内的任务
            // 实际应用中，这里会是核心的业务逻辑
            // 为了演示，这里不做任何耗时操作，以体现其“快速”
        }

        // 假设每隔一定迭代次数需要做一些稍微耗时的操作，或者只是为了观察
        if iterations%1_000_000 == 0 {
            fmt.Printf("Worker: 已完成 %d 次迭代，持续运行...\n", iterations)
        }
    }
    fmt.Println("Worker: 任务已结束。")
}

func main() {
    exitSignal := make(chan struct{}) // 用于发送退出信号的通道
    var wg sync.WaitGroup

    wg.Add(1)
    go worker(exitSignal, &wg) // 启动工作goroutine

    // 主goroutine等待一段时间，然后发送退出信号
    fmt.Println("Main: 等待5秒后发送退出信号...")
    time.Sleep(5 * time.Second)

    fmt.Println("Main: 发送退出信号...")
    close(exitSignal) // 关闭通道以发送退出信号，所有监听者都会收到零值

    wg.Wait() // 等待worker goroutine完成
    fmt.Println("Main: 所有任务已完成，程序退出。")
}

Worker: 开始执行任务...
Main: 等待5秒后发送退出信号...
Worker: 已完成 1000000 次迭代，持续运行...
Worker: 已完成 2000000 次迭代，持续运行...
... (可能有很多行，取决于CPU速度和5秒内能完成的迭代次数)
Worker: 已完成 XXXXXXXX 次迭代，持续运行...
Main: 发送退出信号...
Worker: 收到退出信号，在第 XXXXXXXX 次迭代后停止。
Worker: 任务已结束。
Main: 所有任务已完成，程序退出。