Go语言使用select切换协程

原创

mob64ca12eaf194 2023-08-10 14:31:18 ©著作权

文章标签 Go 代码示例基本语法 文章分类 Go语言后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12eaf194的原创作品，请联系作者获取转载授权，否则将追究法律责任

Go语言使用select切换协程

引言

在并发编程中，协程是一种轻量级的线程，可以在一个线程内同时运行多个协程，使得程序可以更高效地利用系统资源。Go语言内置了协程的支持，并提供了select语句来实现协程的切换。

本文将介绍使用Go语言的select语句切换协程的原理和使用方法，并通过代码示例来演示。

select语句的原理

select语句是Go语言中用于处理多个通道操作的语句。它的基本语法如下：

select {
case <- chan1:
    // 处理chan1的数据
case <- chan2:
    // 处理chan2的数据
...
default:
    // 没有任何通道准备就绪时执行的操作
}

select语句会监听多个通道的读取操作，并在其中一个通道准备就绪时执行对应的操作。如果有多个通道都准备就绪，select语句会随机选择一个通道来执行操作。

使用select切换协程

在Go语言中，可以使用select语句实现协程的轮转执行。具体步骤如下：

创建多个协程，并将它们的通信通道传入select语句；
在每个协程中，通过通道向主协程发送消息，表示自己已经执行完毕；
使用select语句监听多个协程的通道操作，当某个协程发送完消息后，select语句会选择该协程的通道进行处理，然后切换到下一个协程。

下面是一个示例代码，演示了使用select语句切换协程的过程：

package main

import (
	"fmt"
	"time"
)

func worker(id int, ch chan bool) {
	fmt.Printf("Worker %d started\n", id)
	time.Sleep(time.Second * time.Duration(id))
	ch <- true
	fmt.Printf("Worker %d finished\n", id)
}

func main() {
	numWorkers := 5
	ch := make(chan bool, numWorkers)

	for i := 0; i < numWorkers; i++ {
		go worker(i, ch)
	}

	for i := 0; i < numWorkers; i++ {
		select {
		case <-ch:
			fmt.Printf("Received message from worker %d\n", i)
		}
	}
}

在这段代码中，首先创建了一个带有缓冲区的通道ch，用于多个协程之间的通信。然后循环创建了5个协程，并将它们的通道ch传入select语句。

每个协程会通过通道ch向主协程发送消息，表示自己已经执行完毕。主协程使用select语句监听通道ch的操作，当某个协程发送完消息后，select语句会选择该协程的通道进行处理，并输出相应的信息。

通过运行这段代码，我们可以看到不同协程的执行顺序是随机的，但每个协程都会按照自己的延时时间发送消息，主协程则按照接收消息的顺序进行处理。