Go语言有缓冲和无缓冲通道实现样例

时间:2024-01-11 16:05:26

感觉可以,但不好用。

应该有封装程序更高的包包吧。

package main

import (
	"math/rand"
	"fmt"
	"time"
	"sync"
)

const (
	numberGoroutines = 4
	taskLoad = 10
)

var (
	wg sync.WaitGroup
	wg2 sync.WaitGroup
	wg3 sync.WaitGroup
)

func init() {
	rand.Seed(time.Now().Unix())
}

func main() {
	court := make(chan int)
	wg.Add(2)

	go player("Nadal", court)
	go player("Djokovic", court)

	court <- 1
	wg.Wait()

	baton := make(chan int)
	wg2.Add(1)
	go Runner(baton)

	baton <- 1
	wg2.Wait()

	tasks := make(chan string, taskLoad)

	wg3.Add(numberGoroutines)
	for gr := 1; gr <= numberGoroutines; gr++ {
		go worker(tasks, gr)
	}

	for post := 1; post <= taskLoad; post++ {
		tasks <- fmt.Sprintf("Task: %d", post)
	}

	close(tasks)
	wg3.Wait()
}

func player(name string, court chan int) {
	defer wg.Done()

	for {
		ball, ok := <- court
		if !ok {
			fmt.Printf("Player %s Won\n", name)
			return
		}

		n := rand.Intn(100)
		if n % 13 == 0 {
			fmt.Printf("Player %s Missed\n", name)

			close(court)
			return
		}

		fmt.Printf("Player %s Hit %d\n", name, ball)
		ball++

		court <- ball
	}
}

func Runner(baton chan int) {
	var newRunner int
	runner := <-baton
	fmt.Printf("Runner %d Running With Baton\n", runner)
	if runner != 4 {
		newRunner = runner + 1
		fmt.Printf("Runner %d To The Line\n", newRunner)
		go Runner(baton)
	}

	time.Sleep(100 * time.Millisecond)

	if runner == 4 {
		fmt.Printf("Runner %d Finished, Race Over\n", runner)
		wg2.Done()
		return
	}

	fmt.Printf("Runner %d Exchange With Runner %d\n",
		runner,
		newRunner)

	baton <- newRunner
}

func worker(tasks chan string, worker int) {
	defer wg3.Done()

	for {
		task, ok := <- tasks
		if !ok {
			fmt.Printf("Worker: %d: Shutting Down\n", worker)
			return
		}

		fmt.Printf("Worker: %d: Started %s\n", worker, task)

		sleep := rand.Int63n(100)
		time.Sleep(time.Duration(sleep) * time.Millisecond)

		fmt.Printf("Worker: %d: Completed %s\n", worker, task)
	}
}

  Go语言有缓冲和无缓冲通道实现样例