Go基础--goroutine和channel

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,效果如下:

 

posted @ 2018-02-03 00:28  fan-tastic  阅读(1866)  评论(0编辑  收藏  举报