Golang网课-worker pool练习

package main

/* 使用goroutine和channel实现一个计算int64随机数各位数和的程序
1.开启一个goroutine循环生成int64类型的随机数，发送到jobChan
2.开启24个goroutine从jobChan中取出随机数计算各位数的和，将结果发送到resultChan
3.主goroutine从resultChan取出结果并打印到终端输出 */

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

type Job struct {
    num int64
}

type Result struct {
    jobPtr *Job
    sum    int64
}

var jobChan = make(chan *Job, 100)    //传输随机数的 channel
var resChan = make(chan *Result, 100) //传输求和结果的 channel

//1.开启一个goroutine循环生成int64类型的随机数，发送到jobChan
func generate(jobChan chan<- *Job) {
    for {
        x := rand.Int63() //产生随机数x

        newJob := &Job{num: x} //产生Job结构体指针用于在通道jobChan中传输

        jobChan <- newJob

        time.Sleep(time.Millisecond * 500)
    }
}

//2.开启24个goroutine从jobChan中取出随机数计算各位数的和，将结果发送到resultChan
func calc(jobChan <-chan *Job, resChan chan<- *Result) {
    for {
        //取值
        temp_job := <-jobChan
        n := temp_job.num
        sum := int64(0)
        //计算
        for n > 0 {
            sum += n % 10
            n = n / 10
        }
        //把结果发送到resultChan
        var temp_res = &Result{sum: sum, jobPtr: temp_job}
        resChan <- temp_res
    }
}

func main() {
    go generate(jobChan)

    for i := 0; i < 24; i++ {
        go calc(jobChan, resChan)
    }

    //3.主goroutine从resultChan取出结果并打印到终端输出
    for res := range resChan {
        fmt.Printf("num is: %d, sum is: %d\n", res.jobPtr.num, res.sum)
    }
}