003_对go语言中的工作池代码练习的一些思考和改进

在进行工作池的代码练习时候，我发现了一个有趣的事情，首先看下面一段代码：

package main

import "fmt"
import "time"

func worker(id int, jobs <-chan int, results chan<- int) {
	for j := range jobs {
		fmt.Println("worker", id, "processing job", j)
		time.Sleep(time.Second)
		results <- j * 2
	}
}
func main() {
	jobs := make(chan int, 100)
	results := make(chan int, 100)
	for w := 1; w <= 3; w++ {
		go worker(w, jobs, results)
	}
	for j := 1; j <= 9; j++ {
		jobs <- j
	}
	close(jobs)
	for a := 1; a <= 9; a++ {
		<-results
	}
}

　　

上面代码运行结果是下面这样的，会发现3个工人，平均分配了9个工作，每人有3个

worker 3 processing job 1
worker 2 processing job 3
worker 1 processing job 2
worker 3 processing job 4
worker 1 processing job 5
worker 2 processing job 6
worker 1 processing job 8
worker 2 processing job 7
worker 3 processing job 9

　　

但是，如果将代码改造成下面这样，去掉了工作的时间等待：

package main

import "fmt"

func worker(id int, jobs <-chan int, results chan<- int) {
	for j := range jobs {
		fmt.Printf("I am %d working at jobs %d\n", id, j)
		results <- j * 2
	}
}

func main() {
	jobs := make(chan int, 10)
	results := make(chan int, 10)
	for i := 1; i < 4; i++ {
		go worker(i, jobs, results)
	}
	for k := 1; k < 10; k++ {
		jobs <- k
	}
	for v := 1; v < 10; v++ {
		<-results
	}
}

　　

那么运行结果变成了下面这样：

I am 3 working at jobs 1
I am 3 working at jobs 4
I am 3 working at jobs 5
I am 3 working at jobs 6
I am 3 working at jobs 7
I am 3 working at jobs 8
I am 3 working at jobs 9
I am 1 working at jobs 2
I am 2 working at jobs 3

　　

你会发现，第三个工人做了9件工作里面的7件，剩余两个工人闲下来了。经过代码的分析发现，这个例子的逻辑是，先将9件工作全部放在jobs通道中，然后再由工人从jobs通道里取任务，做完后返回结果。但是如果某个工人做的快，还没等剩余的工人取任务时候就完成了，那他就会做很多工作，如果在实际开发中，不考虑类似问题的话，会造成性能浪费，任务分配不平均的问题

经过思考，我决定改变任务分配方式，不能一股脑的将任务全部分配下去，应该做完一件，再分配一件，这样来避免任务分配问题，我在代码中，增加了一个isTrue通道，当第一个任务做完后，将结果发送给isTrue通道，然后任务分配方拿到isTrue里的值后再分配下一个任务，改造后的代码如下：

package main

import "fmt"

func worker(id int, jobs <-chan int, results chan<- int, isTrue chan<- bool) {
	for j := range jobs {
		fmt.Printf("I am %d working at jobs %d\n", id, j)
		results <- j * 2
		isTrue <- true
	}
}

func main() {
	jobs := make(chan int, 10)
	results := make(chan int, 10)
	isTrue := make(chan bool)
	for i := 1; i < 4; i++ {
		go worker(i, jobs, results, isTrue)
	}
	for k := 1; k < 10; k++ {
		jobs <- k
		<-isTrue
	}
	for v := 1; v < 10; v++ {
		<-results
	}
}

　　

代码运行结果如下：

I am 3 working at jobs 1
I am 1 working at jobs 2
I am 2 working at jobs 3
I am 3 working at jobs 4
I am 1 working at jobs 5
I am 2 working at jobs 6
I am 3 working at jobs 7
I am 1 working at jobs 8
I am 2 working at jobs 9

　　

以上就解决了任务分配问题。当然，我这种方式还是有缺点的，因为增加了分配任务的时间，会让任务完成总时间增大，在实际业务中，肯定还会有更加完美的解决方案