温馨提示×

怎么使用Go协程并发遍历数组或切片

go
小亿
126
2024-04-02 19:31:06
栏目: 编程语言

在Go语言中,可以使用协程(goroutine)来并发遍历数组或切片。下面是一个简单的示例代码:

package main

import (
	"fmt"
	"runtime"
)

func main() {
	runtime.GOMAXPROCS(2) // 设置使用的CPU核心数

	nums := []int{1, 2, 3, 4, 5}

	// 使用协程并发遍历数组
	for i := 0; i < len(nums); i++ {
		go func(idx int) {
			fmt.Println(nums[idx])
		}(i)
	}

	// 阻塞主协程,避免程序提前退出
	fmt.Scanln()
}

在上面的代码中,我们使用runtime.GOMAXPROCS(2)来设置程序使用的CPU核心数为2,然后定义了一个包含一些整数的切片nums。接着我们通过一个for循环遍历数组nums,在每一次遍历中使用go func()创建一个新的协程,每个协程打印数组中的某个元素。

需要注意的是,协程可能会以不同的顺序执行,所以打印的结果可能不是按照数组的顺序来的。此外,为了防止主协程提前退出,我们通过fmt.Scanln()来阻塞主协程。

0