go 协程和管道实战

快来打我* 2021-07-24 17:50 504阅读 0赞

# 一 需求 #

1 开启一个 writeData协程，向管道 intChan 中写入 50 个整数。

2 开启一个 readData协程，从管道 intChan 中读取 writeData 写入的数据。

3 注意：writeData 和 readData操作的是同一个管道。

4 主线程需要等待 wrtiteDate 和 readDate 协程都完成工作才能退出管道。

# 二 图解 #

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2NoZW5ncWl1bWluZw_size_16_color_FFFFFF_t_70][]

# 三 代码 #

package main
    
    import (
       "fmt"
       "time"
    )
    
    // write Data
    func writeData(intChan chan int) {
       for i := 1; i <= 50; i++ {
          // 放入数据
          intChan <- i
          fmt.Println("writeData ", i)
          // time.Sleep(time.Second)
       }
       close(intChan) // 关闭
    }
    
    // read data
    func readData(intChan chan int, exitChan chan bool) {
       for {
          v, ok := <-intChan
          if !ok {
             break
          }
          time.Sleep(time.Second)
          fmt.Printf("readData 读到数据=%v\n", v)
       }
       // readData 读取完数据后，即任务完成
       exitChan <- true
       close(exitChan)
    }
    
    
    func main() {
       // 创建两个管道
       intChan := make(chan int, 10)
       exitChan := make(chan bool, 1)
       go writeData(intChan)
       go readData(intChan, exitChan)
       // time.Sleep(time.Second * 10)
       for {
          _, ok := <-exitChan
          if !ok {
             break
          }
       }
    }

# 四 测试 #

writeData  1
    writeData  2
    writeData  3
    writeData  4
    writeData  5
    writeData  6
    writeData  7
    writeData  8
    writeData  9
    writeData  10
    writeData  11
    readData 读到数据=1
    writeData  12
    readData 读到数据=2
    writeData  13
    readData 读到数据=3
    writeData  14
    readData 读到数据=4
    writeData  15
    readData 读到数据=5
    writeData  16
    readData 读到数据=6
    writeData  17
    readData 读到数据=7
    writeData  18
    readData 读到数据=8
    writeData  19
    readData 读到数据=9
    writeData  20
    readData 读到数据=10
    writeData  21
    readData 读到数据=11
    writeData  22
    readData 读到数据=12
    writeData  23
    readData 读到数据=13
    writeData  24
    readData 读到数据=14
    writeData  25
    readData 读到数据=15
    writeData  26
    readData 读到数据=16
    writeData  27
    readData 读到数据=17
    writeData  28
    readData 读到数据=18
    writeData  29
    readData 读到数据=19
    writeData  30
    readData 读到数据=20
    writeData  31
    readData 读到数据=21
    writeData  32
    readData 读到数据=22
    writeData  33
    readData 读到数据=23
    writeData  34
    readData 读到数据=24
    writeData  35
    readData 读到数据=25
    writeData  36
    readData 读到数据=26
    writeData  37
    readData 读到数据=27
    writeData  38
    readData 读到数据=28
    writeData  39
    readData 读到数据=29
    writeData  40
    readData 读到数据=30
    writeData  41
    readData 读到数据=31
    writeData  42
    readData 读到数据=32
    writeData  43
    readData 读到数据=33
    writeData  44
    readData 读到数据=34
    writeData  45
    readData 读到数据=35
    writeData  46
    readData 读到数据=36
    writeData  47
    readData 读到数据=37
    writeData  48
    readData 读到数据=38
    writeData  49
    readData 读到数据=39
    writeData  50
    readData 读到数据=40
    readData 读到数据=41
    readData 读到数据=42
    readData 读到数据=43
    readData 读到数据=44
    readData 读到数据=45
    readData 读到数据=46
    readData 读到数据=47
    readData 读到数据=48
    readData 读到数据=49
    readData 读到数据=50

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2NoZW5ncWl1bWluZw_size_16_color_FFFFFF_t_70]: /images/20210724/1790afed87f441a199491c0e793a8964.png