golang如何停止goroutine（四种方法）

PHPz 原创: 2023-04-11 11:39:16 708浏览

Golang是一门设计用于构建高度并发、高效编程的语言，它提供了方便且高效的goroutine机制，能够极大地提高程序的运行效率和性能。但是，当我们使用goroutine时，有一些需要注意的事项，比如如何停止goroutine运行。下面将着重介绍如何停止goroutine。

Goroutine的基础

在了解如何停止goroutine之前，我们需要先来了解goroutine的基本知识。Goroutine是Golang语言中轻量级线程的实现，可以在一个或多个线程上运行。它是由Go运行时系统管理的，是一个独立、轻量级的执行单元。与传统线程不同，goroutine运行时，它是由Go运行时系统自动进行调度，而且它的调度模型也是非常高效的，它不仅可以很好地支持高并发，而且还能够充分利用多核CPU的优势。

如何创建goroutine

在Go语言中，我们可以通过如下两种常见的方式来创建goroutine：

使用go关键字

go func() {
    //goroutine运行的代码
}()

使用go关键字和函数名

func myFunc() {
    //goroutine运行的代码
}

go myFunc()

如何停止goroutine

停止goroutine是一个比较复杂的问题，因为goroutine在运行时没有提供显式的终止和暂停的方法。这是由于Go语言的设计目标之一就是让goroutine在运行时尽量不被阻塞，因此我们需要一些特殊的技巧来停止它们。

在Go语言中，我们停止goroutine的方法有以下四种:

1. 使用channel

使用channel是最简单、最安全的停止goroutine的方法之一。可以创建一个bool类型的channel，当我们需要停止goroutine时，给这个channel发送一个信号，goroutine收到信号后就可以正常退出了。

func worker(stopChan chan bool) {
    for {
        select {
        case <-stopChan:
            fmt.Println("worker stopped")
            return
        default:
            fmt.Println("worker is running")
            time.Sleep(time.Second)
        }
    }
}

func main() {
    stopChan := make(chan bool)
    go worker(stopChan)
    time.Sleep(3 * time.Second)
    stopChan <- true
    time.Sleep(time.Second)
}

2. 使用context

在Go1.7中，标准库中加入了context包，提供了一种新的挂起、取消goroutine的方法。我们可以在每个goroutine中传入一个Context参数，然后在需要停止goroutine时，对这个Context变量执行cancel操作。

func worker(ctx context.Context) {
    for {
        select {
        case <-ctx.Done():
            fmt.Println("worker stopped")
            return
        default:
            fmt.Println("worker is running")
            time.Sleep(time.Second)
        }
    }
}

func main() {
    ctx, cancel := context.WithCancel(context.Background())
    go worker(ctx)
    time.Sleep(3 * time.Second)
    cancel()
    time.Sleep(time.Second)
}

3. 使用Mutex和WaitGroup

通过共享变量的方式来控制goroutine的停止，这种方法需要使用sync包中的Mutex和WaitGroup。Mutex用于保护共享变量的读写，WaitGroup用于等待所有goroutine完成。

var wg sync.WaitGroup
var stop bool
var mutex sync.Mutex

func worker() {
    defer wg.Done()
    for {
        mutex.Lock()
        if stop {
            mutex.Unlock()
            fmt.Println("worker stopped")
            return
        }
        fmt.Println("worker is running")
        mutex.Unlock()
        time.Sleep(time.Second)
    }
}

func main() {
    for i := 0; i < 3; i++ {
        wg.Add(1)
        go worker()
    }
    time.Sleep(3 * time.Second)
    mutex.Lock()
    stop = true
    mutex.Unlock()
    wg.Wait()
    time.Sleep(time.Second)
}

4. 使用runtime包

使用runtime包的方法则要稍微麻烦一些，在goroutine运行的代码中，需要定期地检查全局变量，然后在需要停止goroutine时，使用runtime包中的函数强制将其终止。

var stop bool

func worker() {
    for {
        if stop {
            fmt.Println("worker stopped")
            return
        }
        fmt.Println("worker is running")
        time.Sleep(time.Second)
    }
}

func main() {
    go worker()
    time.Sleep(3 * time.Second)
    stop = true
    time.Sleep(time.Second)
    runtime.Goexit() //退出当前运行的goroutine
}