V1 Add

package main

import (
    "fmt"
    "sync"
)

func main() {
    var wg sync.WaitGroup
    wg.Add(3)

    catCh := make(chan struct{})
    dogCh := make(chan struct{})
    fishCh := make(chan struct{})

    // 打印cat的协程
    go func() {
        defer wg.Done()
        for i := 0; i < 10; i++ {
            <-catCh
            fmt.Println("cat")
            dogCh <- struct{}{}
        }
    }()

    // 打印dog的协程
    go func() {
        defer wg.Done()
        for i := 0; i < 10; i++ {
            <-dogCh
            fmt.Println("dog")
            fishCh <- struct{}{}
        }
    }()

    // 打印fish的协程
    go func() {
        defer wg.Done()
        for i := 0; i < 10; i++ {
            <-fishCh
            fmt.Println("fish")
            if i < 9 {
                catCh <- struct{}{}
            } else {
                // 9 退出
                break
            }
        }
    }()

    // 启动第一个协程
    catCh <- struct{}{}

    wg.Wait()
}

V2 update 2023-12-21

package main

import (
    "fmt"
    "sync"
    "time"
)

var wg sync.WaitGroup

func main() {
    ch1 := make(chan struct{}, 1)
    ch2 := make(chan struct{}, 1)
    ch3 := make(chan struct{}, 1)

    wg.Add(3)
    start := time.Now().Unix()
    // 这里的 三个必须都是10，否则有的 channel 关闭，再进行写入报错
    total := 10
    // 触发动作
    ch1 <- struct{}{}
    go handle("CAT", total, ch1, ch2)
    go handle("DOG", total, ch2, ch3)
    go handle("FISH", total, ch3, ch1)
    wg.Wait()
    end := time.Now().Unix()
    fmt.Printf("cost:%d", end-start)
}

// go goroutine
func handle(name string, nums int, inputChan chan struct{}, outputChan chan struct{}) {
    defer wg.Done()
    for i := 0; i < nums; i++ {
        // 这里休眠可以感受到 3个 goroutine 的并发执行，而不是串行，1秒中，输出三个
        time.Sleep(1 * time.Second)
        select {
        case <-inputChan:
            fmt.Printf("%s \n", name)
            outputChan <- struct{}{}
        }
    }
}