Go语言圣经 - 第8章 Goroutines 和 Channels - 8.9 并发的退出

第8章 Goroutines 和 Channels

Go语言中的并发程序可以用两种手段来实现：goroutine 和 channel，其支持顺序通信进程，或被简称为CSP，CSP是一种并发编程模型，在这种并发编程模型中，值会在不同运行实例中传递，第二个手段便是多线程共享内存

8.9 并发的退出

前面我们尝试了很多，主要的关注点在于如何让程序“并发”，然而我们也意识到搞大量不受控制的goroutine并不是什么好事。本节我们会介绍如果实现这种控制

这个问题主要分为两类，退出单个goroutine（前面我们使用了往通道中发送一个值来实现通信）和退出多个goroutine

既然往一个channel 中发送一个值可以实现通信，那么是否可以发送多个来实现这个操作呢？其实不太行，因为有时候我们无法确定channel的数量，由于channel本身特性，这会导致阻塞

现在我们使用广播机制，举个例子，当客户端中断了和服务端的联系，我们就需要告诉goroutine，让他们不要再继续执行了
我们看看代码怎么实现的

var done = make(chan struct{})

func cancelled() bool {
	select {
	case <-done:
		return true
	default:
		return false
	}
}

我们新建了一个done channel，用来它来广播，cancel是一个问询函数，另外我们还得获取外部事件输入，代码如下：

go func() {
		os.Stdin.Read(make([]byte, 1))
		close(done)
	}()

继续，我们在main函数中为select 增加一个case

loop:
	for {
		select {
		case <-done:
			for range fileSizes {
				//done nothing
			}
			return
		case size, ok := <-fileSizes:
			if !ok {
				break loop
			}
			nfiles++
			nbytes += size
		case <-tick:
			printDiskUsage(nfiles, nbytes)
		}
	}
	printDiskUsage(nfiles, nbytes)

我们还得让goroutine取消后不再执行其他操作

func walkDir(dir string, n *sync.WaitGroup, fileSizes chan<- int64) {
	defer n.Done()
	if cancelled() {
		return
	}
	for _, entry := range dirents(dir) {
		if entry.IsDir() {
			n.Add(1)
			subdir := filepath.Join(dir, entry.Name())
			walkDir(subdir, n, fileSizes)
		} else {
			fileSizes <- entry.Size()
		}
	}
}

但是取消本身是有代价的，我们检查一下发现瓶颈在于要在directs函数中获取一个信号量，所以我们用select把它取消，这种取消可以把程序效率从几百毫秒提升到几十毫秒

func dirents(dir string) []os.FileInfo {

	select {
	case sema <- struct{}{}:
	case <-done:
		return nil
	}
	defer func() { sema }()
	
	sema <- struct{}{}
	defer func() { <-sema }()

	entries, err := ioutil.ReadDir(dir)
	if err != nil {
		fmt.Fprintf(os.Stderr, "du1:%v\n", err)
		return nil
	}
	return entries
}

虽然上述程序可以在终端中断连接时正常取消退出，但是我们必须注意这种退出是所有goroutine都会停止并且主函数会返回，主函数一旦返回我们是无法确认程序是否释放了所有的资源。所以我们可以调用一个Panic，如果只剩main goroutine一个，它会清理掉一切资源，如果还有其他的goroutine没被取消或者取消操作很花时间，我们可以用Panic来获取足够的信息来验证我们的判断