Go基础--goroutine和channel

2023-07-29,,

goroutine

在go语言中,每一个并发的执行单元叫做一个goroutine

这里说到并发,所以先解释一下并发和并行的概念:

并发:逻辑上具备同时处理多个任务的能力

并行:物理上在同一时刻执行多个并发任务

当一个程序启动时,其主函数即在一个单独的goroutine中运行,一般这个goroutine是主goroutine

如果想要创建新的goroutine,只需要再执行普通函数或者方法的的前面加上关键字go

通过下面一个例子演示并发的效果,主goroutine会计算第45个斐波那契函数,在计算的同时会循环打印:-\|/

这里需要知道:当主goroutine结束之后,所有的goroutine都会被打断,程序就会退出

package main

import (
"time"
"fmt"
) func spinner(delay time.Duration){
for {
for _,r := range `-\|/`{
fmt.Printf("\r%c",r)
time.Sleep(delay)
}
}
} func fib(x int) int{
// 斐波那契函数
if x < 2{
return x
}
return fib(x-1) + fib(x-2)
} func main() {
go spinner(100*time.Millisecond) //开启一个goroutine
const n = 45
fibN:= fib(n)
fmt.Printf("\rFib(%d) = %d\n",n,fibN)
}

当第一次看到go的并发,感觉真是太好用了!!!!

所以在网络编程里,服务端都是需要同时可以处理很多个连接,我们看一下下面的服务端和客户端例子

服务端:

package main

import (
"net"
"io"
"time"
"log"
) func handleConn(c net.Conn){
defer c.Close()
for{
_,err := io.WriteString(c,time.Now().Format("15:04:05\r\n"))
if err != nil{
return
}
time.Sleep(1*time.Second)
}
} func main() {
// 监听本地tcp的8000端口
listener,err := net.Listen("tcp","localhost:8000")
if err != nil{
log.Fatal(err)
}
for {
conn,err := listener.Accept()
if err!= nil{
log.Print(err)
continue
}
go handleConn(conn)
}
}

客户端:

package main

import (
"io"
"log"
"net"
"os"
) func mustCopy(dst io.Writer,src io.Reader){
// 从连接中读取内容,并写到标准输出
if _,err := io.Copy(dst,src);err !=nil{
log.Fatal(err)
} } func main(){
conn,err := net.Dial("tcp","localhost:8000")
if err != nil{
log.Fatal(err)
}
defer conn.Close()
mustCopy(os.Stdout, conn)
}

Channel

channel是不同的goroutine之间的通信机制。

一个goroutine通过channel给另外一个goroutine发送信息。

每个channel 都有一个特殊的类型,也就是channel可以发送的数据的类型

我们可以通过make创建一个channel如:

ch := make(chan int)  这就是创建了一个类型为int的channel

默认我们这样创建的是无缓存的channel,当然我们可以通过第二个参数来设置容量

ch := make(chan int,10)

注意:channel是引用类型,channel的零值也是nil

两个相同类型的channel可以使用==运算符比较。如果两个channel引用的是相通的对象,那么比较的结
果为真。一个channel也可以和nil进行比较。

因为channel是在不同的goroutine之间进行通信的,所以channel这里有两种操作:存数据和取数据,而这里两种操作的

方法都是通过运算符:<-

ch <- x  这里是发送一个值x到channel中

x = <- ch 这里是从channel获取一个值存到变量x

<-ch 这里是从channel中取出数据,但是不使用结果

close(ch) 用于关闭channel

当我们关闭channel后,再次发送就会导致panic异常,但是如果之前发送过数据,我们在关闭channel之后依然可以执行接收操作

如果没有数据的话,会产生一个零值

基于channel发送消息有两个重要方面,首先每个消息都有一个值,但是有时候通讯的事件和发送的时刻也同样重要。

我们更希望强调通讯发送的时刻时,我们将它称为消息事件。有些消息并不携带额外的信息,它仅仅是用做两个goroutine之间的同步,这个时候我们可以用struct{}空结构体作为channel元素的类型

 无缓存的channel

基于无缓存的channel的发送和接受操作将导致两个goroutine做一次同步操作,所以无缓存channel有时候也被称为同步channel

串联的channel (Pipeline)

channel也可以用于多个goroutine连接在一起,一个channel的输出作为下一个channel的输入,这种串联的channel就是所谓的pipeline

通过下面例子理解,第一个goroutine是一个计算器,用于生成0,1,2...形式的整数序列,然后通过channel将该整数序列

发送给第二个goroutine;第二个goroutine是一个求平方的程序,对收到的每个整数求平方,然后将平方后的结果通过第二个channel发送给第三个goroutine

第三个goroutine是一个打印程序,打印收到的每个整数

package main

import (
"fmt"
) func main(){
naturals := make(chan int)
squares := make(chan int) go func(){
for x:=0;;x++{
naturals <- x
}
}() go func(){
for {
x := <- naturals
squares <- x * x
}
}() for{
fmt.Println(<-squares)
}
}

但是如果我把第一个生成数的写成一个有范围的循环,这个时候程序其实会报错的

所以就需要想办法让发送知道没有可以发给channel的数据了,也让接受者知道没有可以接受的数据了

这个时候就需要用到close(chan)

当一个channel被关闭后,再向该channel发送数据就会导致panic异常

当从一个已经关闭的channel中接受数据,在接收完之前发送的数据后,并不会阻塞,而会立刻返回零值,所以在从channel里接受数据的时候可以多获取一个值如:

go func(){
for {
x ,ok := <-naturals
if !ok{
break
}
squares <- x*x
}
close(squares)
}() 第二位ok是一个布尔值,true表示成功从channel接受到值,false表示channel已经被关闭并且里面没有值可以接收

单方向的channel

当一个channel座位一个函数的参数时,它一般总是被专门用于只发送或者只接收

chan <- int :表示一个只发送int的channel,只能发送不能接收

< chan int : 表示一个只接受int的channel,只能接收不能发送

当然在有时候我们需要获取channel内部缓存的容量,可以通过内置的cap函数获取

而len函数则返回的是channel内实际有效的元素个数

基于select的多路复用

这里先说一个拥有的知识点:time.Tick函数

这个函数返回一个channel,通过下面代码进行理解:

package main

import (
"time"
"fmt"
) func main() {
tick := time.Tick(1*time.Second)
for countdown :=10;countdown>0;countdown--{
j :=<- tick
fmt.Println(j)
}
}

程序会循环打印一个时间戳

select 语句:

select {
case <-ch1:
// ...
case x := <-ch2:
// ...use x...
case ch3 <- y:
// ...
default:
// ...
}

select语句的形式其实和switch语句有点类似,这里每个case代表一个通信操作

在某个channel上发送或者接收,并且会包含一些语句组成的一个语句块 。

select中的default来设置当 其它的操作都不能够马上被处理时程序需要执行哪些逻辑

channel 的零值是nil,  并且对nil的channel 发送或者接收操作都会永远阻塞,在select语句中操作nil的channel永远都不会被select到。

这可以让我们用nil来激活或者禁用case,来达成处理其他输出或者输出时间超时和取消的逻辑

补充

不同的goroutine之间如何通信

首先我们能够想到的有:全局变量的方式,我们先通过这种本方法来演示:

package main

import (
"time"
"fmt"
) var exits [3]bool func calc(index int){
for i:=0;i<1000;i++{
time.Sleep(time.Millisecond)
}
exits[index] = true
} func main() {
start := time.Now().UnixNano()
go calc(0)
go calc(1)
go calc(2) for{
if exits[0] && exits[1] &&exits[2]{
break
}
}
end := time.Now().UnixNano()
fmt.Println("finished,const:%d ms",end-start)
}

这种方法其实比较笨,go为我们提供了锁同步的方式 sync.WaitGroup,演示代码为:

//等待一组goroutine执行完成

package main

import (
"time"
"fmt"
"sync"
) var waitGroup sync.WaitGroup func calc(index int){
for i:=0;i<1000;i++{
time.Sleep(time.Millisecond)
}
//执行完成的时候Done
waitGroup.Done()
} func main() {
start := time.Now().UnixNano()
for i:=0;i<3;i++{
// 每次在调用之前add
waitGroup.Add(1)
go calc(i)
}
//在循环外等待wait
waitGroup.Wait()
end := time.Now().UnixNano()
fmt.Println("finished,const:%d ms",end-start)
}

关于Channel的补充

channel 概念:

类似unix中的管道pipe

先进先出

线程安全,多个goroutine同时访问,不需要加锁

channel是有类型的,一个整数的channel只能存放整数

定时器的补充

//定时器
package main import (
"time"
"fmt"
) func main() {
t := time.NewTicker(time.Second)
for v:= range t.C{
fmt.Println("hello",v)
}
}
// 一次性定时器
package main import (
"time"
"fmt"
) func main() {
select{
case <- time.After(time.Second):
fmt.Println("after")
}
}

超时控制

package main

import (
"time"
"fmt"
) func queryDb(ch chan int){
time.Sleep(time.Second)
ch <- 100
} func main() {
ch := make(chan int)
go queryDb(ch)
t := time.NewTicker(time.Second*4)
select{
case v:=<-ch:
fmt.Println("result:",v)
case <-t.C:
fmt.Println("timeout")

关于单元测试和异常捕获的补充

package main

import (
"time"
"fmt"
) func calc(){
// defer 定义的后面出现错误的都可以捕获到
defer func() {
err := recover()
if err!=nil{
fmt.Println(err)
}
}()
var p *int
*p = 100 }

关于单元测试的简单例子演示:

package test

func add(a,b int)int{
return a+b
}

上面的代码,如果想要做单元测试,我们需要定义一个测试文件,我把上面代码的文件名是calc.go,单元测试代码的文件名calc_test.go

package test

import "testing"

func TestAdd(t *testing.T){
result := add(2,3)
if result != 5{
t.Fatalf("add is not right")
return
}
t.Logf("add is right")
}

这样当我们需要测试add函数的时候只需要在该包下执行go test,效果如下:

如果想要看更详细的可以通过go test -v,效果如下:

Go基础--goroutine和channel的相关教程结束。

《Go基础--goroutine和channel.doc》

下载本文的Word格式文档,以方便收藏与打印。