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

感觉可以,但不好用。

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

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语言有缓冲和无缓冲通道实现样例_第1张图片

你可能感兴趣的:(Go语言有缓冲和无缓冲通道实现样例)