(编辑:jimmy 日期: 2024/12/22 浏览:2)
Golang 运行时(runtime)管理了一种轻量级线程,被叫做 goroutine。创建数十万级的 goroutine 是没有问题的。范例:
复制代码 代码如下:
package main
import (
"fmt"
"time"
)
func say(s string) {
for i := 0; i < 5; i++ {
time.Sleep(100 * time.Millisecond)
fmt.Println(s)
}
}
func main() {
// 开启一个 goroutine 执行 say 函数
go say("world")
say("hello")
}
我们使用 channel 和 goroutine 通讯。channel 中是一种带有类型的通道,被用于接收和发送特定类型的值。操作符 <- 被叫做 channel 操作符(这个操作符中箭头表明了值的流向):
复制代码 代码如下:
// 发送 v 到 channel ch
ch <- v
// 接收 channel ch 中的值并赋值给 v
v := <-ch
使用 channel 和 goroutine 通讯能够避免显式使用锁机制,通过 channel 发送和接收值时默认是阻塞的。
通过 make 函数创建 channel:
复制代码 代码如下:
// int 指定 channel 收发值的类型为 int
ch := make(chan int)
一个完整的例子:
复制代码 代码如下:
package main
import "fmt"
// 计算数组 a 中所有元素值之和
func sum(a []int, c chan int) {
sum := 0
for _, v := range a {
sum += v
}
// 计算结果发送到 channel c
c <- sum
}
func main() {
a := []int{7, 2, 8, -9, 4, 0}
// 创建 channel c
c := make(chan int)
go sum(a[:len(a)/2], c)
go sum(a[len(a)/2:], c)
// 接收两个 goroutine 发送的计算结果
x, y := <-c, <-c
fmt.Println(x, y, x+y)
}package main
import "fmt"
// 计算数组 a 中所有元素值之和
func sum(a []int, c chan int) {
sum := 0
for _, v := range a {
sum += v
}
// 计算结果发送到 channel c
c <- sum
}
func main() {
a := []int{7, 2, 8, -9, 4, 0}
// 创建 channel c
c := make(chan int)
go sum(a[:len(a)/2], c)
go sum(a[len(a)/2:], c)
// 接收两个 goroutine 发送的计算结果
x, y := <-c, <-c
fmt.Println(x, y, x+y)
}
channel 可以带有一个缓冲区(buffer)来缓存被传递的值,向 channel 中发送时只有缓冲区满的情况下会阻塞,接收 channel 中的值时只有在缓冲区空的情况下阻塞:
复制代码 代码如下:
package main
import "fmt"
func main() {
// 创建 channel,缓冲区长度为 2
c := make(chan int, 2)
// 由于 channel 的缓冲区长度为 2
// 因此发送不会阻塞
c <- 1
c <- 2
fmt.Println(<-c)
fmt.Println(<-c)
}
发送者可以调用 close 来关闭 channel,接收者可以检测到 channel 是否被关闭:
复制代码 代码如下:
// 这里的 ok 为 false 表示已经没有值可以接收了,并且 channel 被关闭了
v, ok := <-ch
不要向已经关闭的 channel 发送值了(will cause a panic)。
我们可以使用 for range 来接收 channel 中的值:
复制代码 代码如下:
package main
import "fmt"
func fibonacci(n int, c chan int) {
x, y := 0, 1
for i := 0; i < n; i++ {
c <- x
x, y = y, x+y
}
// 必须要关闭 c
close(c)
}
func main() {
c := make(chan int, 10)
go fibonacci(cap(c), c)
// 这里 for 和 range 组合使用
// 不断的接收 c 中的值一直到它被关闭
for i := range c {
fmt.Println(i)
}
}
通常来说,我们不需要主动的关闭 channel。但有时候接收者必须被告知已经没有值可以接收了,这时候主动关闭是必要的,例如终止 for range 循环。
使用 select 语句可以让一个 goroutine 等待多个通讯操作。select 会阻塞直到某个 case 能够运行,如果同时存在多个可执行的,那么将随机选择一个:
复制代码 代码如下:
package main
import "fmt"
func fibonacci(c, quit chan int) {
x, y := 0, 1
for {
select {
case c <- x:
x, y = y, x+y
// 控制此线程退出
case <-quit:
fmt.Println("quit")
return
}
}
}
func main() {
c := make(chan int)
quit := make(chan int)
go func() {
for i := 0; i < 10; i++ {
fmt.Println(<-c)
}
quit <- 0
}()
fibonacci(c, quit)
}
select 中的 default 会在没有任何 case 可执行时执行(类似于 switch):
复制代码 代码如下:
package main
import (
"fmt"
"time"
)
func main() {
// 创建一个 tick channel
// 在 100 毫秒后会向 tick channel 中发送当前时间
tick := time.Tick(100 * time.Millisecond)
// 创建一个 boom channel
// 在 500 毫秒后会向 boom channel 中发送当前时间
boom := time.After(500 * time.Millisecond)
for {
select {
case <-tick:
fmt.Println("tick.")
case <-boom:
fmt.Println("BOOM!")
return
default:
fmt.Println(" .")
time.Sleep(50 * time.Millisecond)
}
}
}