003 Golang-channel-practice

第三题

给定一个int slice，将里面每个值的平方数计算并输出。

直接按照上一题的模板！

package main

import (
	"fmt"
	"sync"
)

func main() {
	sliceInt := []int{2, 5, 4, 8, 6, 33, 54}
	c := generator2(sliceInt)
	receiver2(c)
}

func generator2(slice []int) <-chan int {
	c := make(chan int, len(slice))
	go func() {
		for _, val := range slice {
			c <- val
		}
		close(c) // 关闭channel，表示没有更多的数据将被发送
	}()
	return c
}

func receiver2(c <-chan int) {
	var wg sync.WaitGroup

	for i := range c {
		wg.Add(1)
		go func(val int) {
			defer wg.Done()
			fmt.Println(val * val)
		}(i)
	}

	wg.Wait() // 等待所有goroutine完成
}