Go to Learn Go之并发

概述

在上一节的内容中，我们介绍了Go的类型转换，包括：断言类型转换、显式类型转换、隐式类型转换、强制类型转换、strconv包等。在本节中，我们将介绍Go的并发。Go语言以其强大的并发模型而闻名，其并发特性主要通过以下几个元素来实现：Goroutines、Channels、WaitGroups、Mutex和Select。通过结合使用以上元素，Go语言提供了强大的并发支持，使得编写高效、高性能、高吞吐量的并发程序变得相对容易。

Goroutines

Goroutines是Go语言中轻量级的并发单元，可以与其他goroutine并发执行。它们在相同的地址空间内运行，但每个goroutine都有自己的栈和局部变量。Goroutine的启动和销毁开销很小，使得在程序中可以创建大量的Goroutine。相比于线程，Goroutine的创建和管理成本更低，因为它们不需要像线程一样分配固定的内存空间。此外，Goroutine之间可以通过Channels进行通信，避免使用共享内存和信号量等机制，从而避免了竞态条件和数据竞争等问题。

Goroutine是Go语言的主要并发原语，通常用于实现高并发的应用程序。Go运行时将Goroutine有效地调度到真实的线程上，以避免浪费资源。因此，可以轻松地创建大量的Goroutine（比如：每个请求一个Goroutine），并且可以编写简单的、命令式的阻塞代码。

Goroutine的语法格式为：

 go <func_name>(<arguments>)

其中，go关键字表示启动一个新的Goroutine，func_name表示要启动的函数名，arguments表示传递给函数的参数列表。通过在函数调用前加上go关键字，可以启动一个新的Goroutine来执行该函数。这个Goroutine将与其他Goroutine并发执行，并且不需要显式地创建和管理线程。

在下面的示例代码中，我们使用go关键字启动了两个Goroutine来执行printMsg函数。每个Goroutine都会打印出相应的消息，并且通过time.Sleep函数来模拟一些耗时的操作。主Goroutine在启动了其他两个Goroutine之后会等待一段时间，以确保所有Goroutine都有足够的时间来执行完毕。

package main

import (
    "fmt"
    "time"
)

func main() {
    // 启动第1个Goroutine
    go printMsg("Hello")
    // 启动第2个Goroutine
    go printMsg("World")

    // 等待一段时间，以确保所有Goroutine执行完毕
    time.Sleep(time.Second)
}

func printMsg(msg string) {
    for i := 0; i < 5; i++ {
        fmt.Println(msg)
        // 模拟耗时的操作
        time.Sleep(200 * time.Millisecond)
    }
}

注意：Goroutine之间的执行顺序是不确定的，因此每次运行程序都会得到不同的输出结果，这取决于Go运行时调度器的实现细节和系统负载等因素。

Channels

Channels是一种通信机制，用于在goroutines之间进行数据传输和同步操作。Channels支持发送和接收操作，并且可以在发送和接收操作之间进行阻塞，以实现同步。Channels的使用非常灵活，可以根据需要进行单向或双向数据传输。它们可以用于在不同的goroutines之间传递数据，以及实现数据共享。

在创建Channels时，可以指定其缓冲区大小，缓冲区的大小决定了可以存储在Channels中的数据量。如果空闲缓冲区为空，发送操作会被阻塞，直到有接收操作。如果空闲缓冲区已满，接收操作会被阻塞，直到有发送操作。这种机制可以实现数据在goroutines之间的有效传输和同步。

注意：不同类型的Channel有不同的性能和用途。无缓冲的Channel（即缓冲区大小为0）可以在发送和接收操作之间进行同步，而有缓冲的Channel可以提高并发性能，但需要小心处理缓冲区溢出的问题。

Channel的语法格式为：

 chan <type>

其中，type表示Channel中传输的数据类型。比如：chan int表示一个用于传输整数类型的Channel。除了指定数据类型之外，还可以使用chan来创建具有不同缓冲区大小的Channel。比如：chan int buffer(10)表示创建一个缓冲区大小为10的整数类型Channel。

除了使用chan来创建Channel之外，还可以使用内置的make函数来创建具有指定类型的Channel。比如：make(chan int)表示创建一个整数类型的无缓冲Channel。

在使用Channel时，可以使用以下操作进行数据传输和同步。

x := <-ch：从Channel中接收数据，并将接收到的数据赋值给变量x。

ch <- x：向Channel中发送数据，并将变量x的值发送到Channel中。

如果Channel被阻塞，则接收操作将阻塞直到有数据可用。如果发送操作导致缓冲区已满，则发送操作将阻塞直到有空间可用。

在下面的示例代码中，我们将数组分为两个切片，并通过两个goroutine来计算切片之和。在goroutine完成计算并将切片之和发送到通道后，main函数会从通道中接收数据，并计算最终的总和。

package main

import "fmt"

func sum(s []int, c chan int) {
    total := 0
    for _, v := range s {
        total += v
    }

    // 把total发送到通道
    c <- total
}

func main() {
    data := []int{1, 2, 3, 4, 5, 6}
    c := make(chan int)

    offset := len(data) / 2
    go sum(data[:offset], c)
    go sum(data[offset:], c)

    // 从通道中接收结果
    x, y := <-c, <-c
    // 输出：15 6 21 或 6 15 21
    fmt.Println(x, y, x + y)
}

除了逐个接收数据之外，还可以通过range关键字来遍历读取到的数据。注意：使用range遍历时，需要确保发送完数据后，及时调用close()函数来关闭通道。否则，range遍历不会结束，会一直阻塞等待接收新的数据。

在下面的示例代码中，我们首先使用make函数创建了一个整数类型的Channel。然后，我们启动一个匿名的Goroutine来循环发送数字10至50到Channel中，并在发送完毕后关闭Channel。最后，我们在主Goroutine中使用range关键字来迭代接收Channel中的数据，并将其打印输出。

通过调用close函数可关闭一个Channel，关闭Channel表示再也不会向该Channel发送任何数据。对于已经发送到Channel中的数据，仍然可以被接收。由于Channel已经被关闭，迭代接收数据将自动停止。

package main

import "fmt"

func main() {
    // 创建一个整数类型的Channel
    ch := make(chan int)
  
    // 启动一个Goroutine
    go func() {
        for i := 10; i <= 50; i += 10 {
            // 发送数据到Channel
            ch <- i
            fmt.Println("sub routine:", i)
        }

        // 关闭Channel
        close(ch)
    }()  

    // 从Channel接收数据，依次输出：10 20 30 40 50
    for num := range ch {
        fmt.Println(num)
    }  
}

WaitGroups

在Go语言中，WaitGroups是sync包中的一个类型，用于等待一组Goroutine执行完成。它提供了一种方便的方式，以确保所有的Goroutine都执行完毕后，再继续执行后续的逻辑。

WaitGroups的使用比较简单：首先，需要创建一个WaitGroups实例；然后，通过调用Add()函数增加等待的Goroutine数量，每个Goroutine执行完毕后要调用Done()函数进行计数减一；最后，在主Goroutine中调用Wait()函数来等待所有的Goroutine都执行完毕。

在下面的示例代码中，我们创建了一个WaitGroups实例wg，然后通过调用Add()函数增加了两个Goroutine。每个Goroutine中使用defer语句调用Done()函数来标记该Goroutine的执行完成。最后，在主Goroutine中调用Wait()函数来等待所有的Goroutine都执行完毕，然后继续执行后续的逻辑。

package main

import "fmt"
import "sync"
import "time"

func main() {
    var wg sync.WaitGroup

    // 启动第一个Goroutine
    wg.Add(1)
    go func() {
        defer wg.Done()
        fmt.Println("Goroutine 1 started")
        time.Sleep(1 * time.Second)
        fmt.Println("Goroutine 1 finished")
    }()

    // 启动第二个Goroutine
    wg.Add(1)
    go func() {
        defer wg.Done()
        fmt.Println("Goroutine 2 started")
        time.Sleep(2 * time.Second)
        fmt.Println("Goroutine 2 finished")
    }()

    // 等待所有Goroutine执行完毕
    wg.Wait()

    // 所有Goroutine执行完毕后，继续执行后续逻辑
    fmt.Println("All Goroutines finished")
}

Mutex

在Go语言中，mutex是一种用于实现并发安全的锁机制。它提供了一种简单的方式来保护共享资源，以避免多个Goroutine同时访问和修改数据，从而导致竞争条件或数据不一致的问题。mutex通常是通过sync.Mutex类型来实现的，这个类型提供了两个函数：Lock和Unlock。

在下面的示例代码中，我们定义了一个全局变量counter和一个sync.Mutex类型的变量mutex。在increment函数中，我们使用mutex.Lock()来锁定mutex，以确保在同一时间只有一个Goroutine可以访问和修改counter。在完成对counter的修改后，使用defer mutex.Unlock()来解锁mutex，以确保在函数返回之前释放锁，从而允许其他Goroutine获取锁并访问共享资源。最后，在主函数中，我们启动了5个并发的Goroutine来增加计数器的值，并等待一段时间后打印最终的计数结果。

package main

import "fmt"
import "sync"
import "time"

var (
    counter int
    mutex sync.Mutex
)  

func increment() {
    // 锁定mutex，确保同一时间只有一个Goroutine可以访问和修改counter
    mutex.Lock()
    defer mutex.Unlock()

    // 增加计数器的值
    fmt.Println("Current counter:", counter)
    counter++
}

func main() {
    // 启动5个并发的Goroutine来增加计数器的值
    for i := 0; i < 5; i++ {
        go increment()
    }

    // 等待所有Goroutine执行完毕
    time.Sleep(time.Second)
    fmt.Println("Final counter:", counter)
}

Select

select语句用于在多个通道操作之间进行选择，它允许你等待多个通道操作中的任意一个完成，然后执行对应的代码块。其语法如下：

select {
case <-channel1:
    // 执行channel1操作完成的代码块
case <-channel2:
    // 执行channel2操作完成的代码块
case <-channel3:
    // 执行channel3操作完成的代码块
default:
    // 如果没有任何通道操作完成，执行default代码块
}

在select语句中，每个case子句必须是一个通道操作。当其中一个通道操作完成时，对应的代码块将被执行。如果没有任何通道操作完成，且存在default子句，则执行default代码块。

在下面的示例代码中，我们创建了三个通道，并使用三个Goroutine分别向这三个通道发送消息。然后，在select语句中等待哪个通道先完成操作，并打印收到的消息。由于发送消息的Goroutine使用了不同的延迟时间，因此最终打印的消息取决于哪个通道最先完成操作。

package main

import "fmt"
import "time"

func func1(channel1 chan string) {
    time.Sleep(1 * time.Second)
    channel1 <- "Channel 1"
}

func func2(channel2 chan string) {
    time.Sleep(2 * time.Second)
    channel2 <- "Channel 2"
}

func func3(channel3 chan string) {
    time.Sleep(3 * time.Second)
    channel3 <- "Channel 3"
}

func main() {  
    channel1 := make(chan string)
    channel2 := make(chan string)
    channel3 := make(chan string)

    go func1(channel1)
    go func2(channel2)
    go func3(channel3)

    select {
    case msg1 := <-channel1:
        fmt.Println("Received from Channel 1:", msg1)
    case msg2 := <-channel2:
        fmt.Println("Received from Channel 2:", msg2)
    case msg3 := <-channel3:
        fmt.Println("Received from Channel 3:", msg3)
    }
}