掌握Go并发模式，利用管道和取消实现协程通信

2023-12-12 10:23:14

在Go语言中，并发编程是利用多个协程同时执行任务的一种编程范式。协程是轻量级的线程，它允许您在同一时间内执行多个任务，而无需创建多个操作系统线程。这使得Go语言非常适合构建高性能、可扩展的应用程序。

在Go语言中，管道（channel）是一种特殊的类型，它允许您在协程之间传递数据。管道可以是双向的，也可以是单向的。双向管道允许协程在管道两端同时读写数据，而单向管道只允许协程在管道的一端读数据或写数据。

取消（cancel）是一种机制，它允许您在协程执行过程中终止协程。当您调用cancel函数时，协程将收到一个信号，告知它应该终止执行。这对于在协程执行过程中遇到错误时非常有用。

在本文中，我们将通过一个示例来演示如何使用管道和取消来实现协程之间的通信与控制。在这个示例中，我们将创建一个管道，并将它传递给两个协程。第一个协程将向管道中写入数据，而第二个协程将从管道中读取数据。当第一个协程遇到错误时，我们将调用cancel函数来终止第二个协程。

package main

import (
	"context"
	"fmt"
	"sync"
)

func main() {
	// 创建一个管道
	ch := make(chan int)

	// 创建一个上下文对象
	ctx, cancel := context.WithCancel(context.Background())

	// 创建两个协程
	go func() {
		for {
			select {
			case <-ctx.Done():
				fmt.Println("协程1已终止")
				return
			case ch <- 1:
				fmt.Println("协程1已写入数据")
			}
		}
	}()

	go func() {
		for {
			select {
			case <-ctx.Done():
				fmt.Println("协程2已终止")
				return
			case v := <-ch:
				fmt.Println("协程2已读取数据:", v)
			}
		}
	}()

	// 等待协程执行完毕
	var wg sync.WaitGroup
	wg.Add(2)
	go func() {
		wg.Wait()
		cancel()
	}()
	wg.Wait()
}