首发于taowen

goroutine 所有权和退出

Go 的标准实现里 goroutine 是无主的。concurrent.UnboundedExecutor 的目标就是把 ownership 的概念加到 goroutine 上。通过把启动的goroutine都挂在一个executor上,从而可以跟踪这一组goroutine的存活状况。当需要退出的时候,可以把这些个goroutine通过context cancel的方式退出。

package concurrent

import (
	"context"
	"fmt"
	"time"
)

func ExampleUnboundedExecutor_Go() {
	executor := NewUnboundedExecutor()
	executor.Go(func(ctx context.Context) {
		fmt.Println("abc")
	})
	time.Sleep(time.Second)
	// output: abc
}

func ExampleUnboundedExecutor_StopAndWaitForever() {
	executor := NewUnboundedExecutor()
	executor.Go(func(ctx context.Context) {
		everyMillisecond := time.NewTicker(time.Millisecond)
		for {
			select {
			case <-ctx.Done():
				fmt.Println("goroutine exited")
				return
			case <-everyMillisecond.C:
				// do something
			}
		}
	})
	time.Sleep(time.Second)
	executor.StopAndWaitForever()
	fmt.Println("exectuor stopped")
	// output:
	// goroutine exited
	// exectuor stopped
}

在stop的等待过程中,还可以打印日志输出哪里 goroutine 还没有 quit。这个需要实现 LogInfo 这个 callback。

除了提供goroutine所有权之外。还默认对panic做了recover。对于新启动的 goroutine 非常容易忘记实现 defer 来 recover 它的 panic。绝大部分时候 goroutine panic 行为应该是 log 下来,而不是直接挂掉整个进程。目前的实现是会调用全局的 HandlePanic 的 callback。

实现代码:

modern-go/concurrentgithub.com图标

发布于 2018-03-01

文章被以下专栏收录