golang--交替打印字符

    摘要:今天面试中,面试官让写使用 goroutine 交替打印字符 ABC,没有写出来,所以今天总结一下:

  在学习操作系统信号量章节的时候,我们知道用信号量可以实现同步和互斥问题,比如著名的生产者和消费者问题,读者写者问题以及哲学家就餐问题。一般情况下:

    把初始化信号量设置为0,用于解决互斥问题

    把初始化信号量设置为1,用于解决同步问题

 

  三个协程交替打印 ABC 的操作,显然属于同步问题,因为 B 必须在打印 A 之后打印,所以想要将初始化信号量设置为 1,这是一个典型的生产者消费者模型(B协程作为A协程的消费者,同时作为C协程的生产者)。

package main

import (
    "fmt"
    "sync"
)

func PrintA(wg *sync.WaitGroup, chA, chB chan struct{}) {
    for i := 0; i < 10; i++ {
        <-chA
        fmt.Println("A")
        chB <- struct{}{}
    }
    wg.Done()
}

func PrintB(wg *sync.WaitGroup, chB, chC chan struct{}) {
    for i := 0; i < 10; i++ {
        <-chB
        fmt.Println("B")
        chC <- struct{}{}
    }
    wg.Done()
}

func PrintC(wg *sync.WaitGroup, chC, chA chan struct{}) {
    for i := 0; i < 10; i++ {
        <-chC
        fmt.Println("C")
        chA <- struct{}{}
    }
    wg.Done()
}

func main() {
    wg := sync.WaitGroup{}
    chA := make(chan struct{}, 1)
    chB := make(chan struct{}, 1)
    chC := make(chan struct{}, 1)

    chA <- struct{}{}

    wg.Add(1)
    go PrintA(&wg, chA, chB)

    wg.Add(1)
    go PrintB(&wg, chB, chC)

    wg.Add(1)
    go PrintC(&wg, chC, chA)

    wg.Wait()
    fmt.Println("Success.")
}

 

posted on 2020-09-24 17:31  爱笑的张飞  阅读(817)  评论(0编辑  收藏  举报

导航