Go的程式設計模式一-管道Pipeline

package main
import "fmt"

/*
管道模式
1.PipeFunc 管道函式，管道函式接收一個管道，返回一個新管道
2.Pipeline 引數，1資料來源通道，2管道函式列表
*/
type PipeFunc func(<-chan int) <-chan int
func Sq(ch <-chan int) <-chan int {
    out := make(chan int)
    go func() {
        for val := range ch {
            out <- val * val
        }
        close(out)
    }()
    return out
}
func Odd(ch <-chan int) <-chan int {
    out := make(chan int)
    go func() {
        for val := range ch {
            if val%2 != 0 {
                out <- val
            }
        }
        close(out)
    }()
    return out
}
func Sum(ch <-chan int) <-chan int {
    out := make(chan int)
    go func() {
        sum := 0
        for val := range ch {
            sum += val
        }
        out <- sum
        close(out)
    }()
    return out
}
//引數：1.資料來源 2.管道函式
func Pipeline(dataSource <-chan int, pipeFns ...PipeFunc) <-chan int {
    for i := range pipeFns {
        dataSource = pipeFns[i](dataSource) //每次都是新的管道重新賦值給dataSource
    }
    return dataSource
}
//使用
func main() {
    data := []int{1, 2, 3, 4, 5, 6, 6, 7, 8, 8, 9, 9, 0}
    dataSource := make(chan int)
    go func() {
        for _, item := range data {
            dataSource <- item
        }
        close(dataSource)
    }()
    resCh := Pipeline(dataSource, Sq, Odd, Sum) //返回的是最後結果的管道
    for v := range resCh {
        fmt.Println("result: ", v)
    }
}

本作品採用《CC 協議》，轉載必須註明作者和本文連結