golang面试题：Sum of Squares

 

package main

import "fmt"

func SumOfSquares(c, quit chan int) {
	y := 1
	for {
		select {
		case c <- y * y:
			y++
		case <-quit:
			return
		}
	}
}
func main() {
	sumChan := make(chan int)
	quitChan := make(chan int)

	sum := 0

	go func() {
		for i := 1; i <= 5; i++ {
			sum += <-sumChan
		}
		fmt.Println(sum)
		quitChan <- 0
	}()

	SumOfSquares(sumChan, quitChan)
}