协程间的三种通信方式，你学会了吗？

最新推荐文章于 2024-10-14 17:05:04 发布

天帅风ztm

最新推荐文章于 2024-10-14 17:05:04 发布

阅读量1.5k

点赞数 2

分类专栏： golang 文章标签：开发语言

本文链接：https://blog.youkuaiyun.com/qq_17199495/article/details/125787317

版权

golang 专栏收录该内容

14 篇文章

订阅专栏

本文介绍了Go语言中三种重要的协程通信方式：1) 使用通道(channel)实现数据传递；2) 通过Context控制协程的生命周期；3) 利用sync.Cond进行条件同步。示例代码详细展示了每种方法的用法，帮助读者深入理解Go协程的灵活运用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

golang中要想灵活运用协程来解决问题，协程间的通信一定要掌握。这里列举了3种方式。sync.waitgroup 其实也应该掌握

1、channel

import (
   "fmt"
   "time"
)

func main() {
   ch := make(chan string)

   go sendData(ch)
   go getData(ch)

   time.Sleep(1e9)
}

func sendData(ch chan string){
  ch <- "golang"
}

func getData(ch chan string){
  fmt.Println(<- ch)
}

2、context

// 使用 Context 控制多个 goroutine
func contextPart4() {
    ctx, cancel := context.WithCancel(context.Background())
    go watch(ctx, "task 1")
    go watch(ctx, "task 2")
    go watch(ctx, "task 3")
 
    time.Sleep(10 * time.Second)
    log.Println("可以通知任务结束")
    cancel()
    time.Sleep(5 * time.Second)
}
 
func watch(ctx context.Context, name string) {
    for {
        select {
        case <-ctx.Done():
            log.Println(name + " 任务即将要退出了...")
            return
        default:
            log.Println(name + " goroutine 继续处理任务中...")
            time.Sleep(2 * time.Second)
        }
    }
}

3、sync.cond

var locker sync.Mutex 
var cond = sync.NewCond(&locker) 
 
// NewCond(l Locker)里面定义的是一个接口,拥有lock和unlock方法。 
// 看到sync.Mutex的方法,func (m *Mutex) Lock(),可以看到是指针有这两个方法,所以应该传递的是指针 
func main() { 
    // 启动多个协程 
    for i := 0; i < 10; i++ { 
        gofunc(x int) { 
            cond.L.Lock()          // 获取锁 
            defer cond.L.Unlock()  // 释放锁 
           
            cond.Wait()   // 等待通知，阻塞当前 goroutine 
           
            // 通知到来的时候, cond.Wait()就会结束阻塞, do something. 这里仅打印 
            fmt.Println(x) 
        }(i) 
    } 
   
    time.Sleep(time.Second * 1) // 睡眠 1 秒，等待所有 goroutine 进入 Wait 阻塞状态 
    fmt.Println("Signal...") 
    cond.Signal()               // 1 秒后下发一个通知给已经获取锁的 goroutine 
   
    time.Sleep(time.Second * 1) 
    fmt.Println("Signal...") 
    cond.Signal()               // 1 秒后下发下一个通知给已经获取锁的 goroutine 
   
    time.Sleep(time.Second * 1) 
    cond.Broadcast()            // 1 秒后下发广播给所有等待的goroutine 
    fmt.Println("Broadcast...") 
    time.Sleep(time.Second * 1) // 等待所有 goroutine 执行完毕 
}