如何利用 Go 语言进行并发编程？

PHPz 原创: 2023-06-10 10:33:07 612浏览

随着计算机硬件的不断发展，处理器中的 CPU 核心不再单独增加时钟频率，而是增加核心数量。这引发了一个显而易见的问题：如何发挥这些核心的性能？

一种解决方法是通过并行编程，即同时执行多个任务，以充分利用 CPU 核心。这就是 Go 语言的一个独特之处，它是一门专为并发编程而设计的语言。

在本文中，我们将探讨如何利用 Go 语言进行并发编程。

协程

首先，我们需要了解的是 Go 语言中的一种特殊机制：协程。协程（Coroutine）是一种轻量级线程，可以在一个线程中多次切换执行，实现并行执行。

与操作系统线程相比，协程的切换成本非常低。它们由 Go 运行时（runtime）管理，该运行时使用 m:n 的映射方式将 m 个协程映射到 n 个操作系统线程上。这使得 Go 语言具有非常高效和稳定的并发执行能力。

在 Go 语言中，可以使用 go 关键字来启动一个协程。例如：

func main() {
    go hello()
}

func hello() {
    fmt.Println("Hello, world!")
}

在上面的代码中，hello() 函数将在一个新的协程中执行。当程序退出 main() 函数时，hello() 函数可能还在执行，因此程序不会立即退出。

通道

协程之间的通信非常重要，因为它们需要共享数据。Go 语言中有一种特殊类型的变量，称为通道（Channel），用于在协程之间传递数据。

可以通过 make() 函数创建一个通道，例如：

ch := make(chan int)

上面的代码将创建一个整数类型的通道。

数据可以通过通道的发送和接收操作进行传递。可以使用 <- 运算符对通道进行发送和接收操作。例如：

ch <- 42 // 发送数据
x := <-ch // 接收数据

<- 运算符可以在左侧或右侧使用，以用于发送或接收数据。如果通道是无缓冲的，则发送操作将阻塞，直到另一个协程接收数据。类似地，如果没有可用的数据，则接收操作将阻塞。

WaitGroup

在处理多个协程时，可能需要等待它们全部执行完毕。可以使用 sync.WaitGroup 来实现这个目的。例如：

func main() {
    var wg sync.WaitGroup
    wg.Add(2) // 增加计数器

    go func() {
        defer wg.Done() // 完成时减少计数器
        fmt.Println("Hello,")
    }()

    go func() {
        defer wg.Done() // 完成时减少计数器
        fmt.Println("world!")
    }()

    wg.Wait() // 等待协程全部完成
}

在上面的代码中，wg 是一个 sync.WaitGroup 对象，包含一个计数器。Add() 方法将计数器增加，表示需要等待的协程数。Done() 方法将计数器减少，表示一个协程已经完成。Wait() 方法将一直等待，直到计数器为零。

例子

下面是一个示例程序，演示了如何利用协程和通道进行并发编程：

func main() {
    ch := make(chan int)

    go func() {
        for i := 0; i < 10; i++ {
            ch <- i // 发送数据
        }
        close(ch) // 关闭通道
    }()

    for i := range ch { // 循环接收数据，直到通道关闭
        fmt.Println(i)
    }
}

在上面的代码中，我们创建了一个整数类型的通道 ch。然后，我们在一个新的协程中向通道发送 0 到 9 的整数。最后，我们使用 range 关键字循环接收通道中的数据，并打印出来。

注意，我们在发送完所有数据后，通过 close() 方法关闭了通道。这使得循环读取通道的协程可以退出。