Go 练习

教你上传本地代码到github

程序1

package main

import (
    "fmt"
    "sort/pipeline"
)

func main() {
    p := pipeline.ArraySource(3, 2, 6, 7, 4)
    for {
        if num, ok := <-p; ok {
            fmt.Println(num)
        } else {
            break
        }
    }
}

package pipeline
func ArraySource(a ...int) <-chan int {
    out := make(chan int)
    go func() {
        for _, v := range a {
            out <- v
        }
        close(out)
    }()
    return out
}

程序2

main.go

package main
import (
    "fmt"
    "os"
    "sort/pipeline"
)

func main() {
    file, err := os.Create("small.in")
    if err != nil {
        panic(err)
    }
    p := pipeline.RandomSource(10)
    pipeline.WriterSink(file, p)

    file, err = os.Open("small.in")
    if err != nil {
        panic(err)
    }
    p = pipeline.ReaderSource(file)
    for v := range p {
        fmt.Println(v)
    }
    mergeDemo()

}

func mergeDemo() {
    p := pipeline.Merge(
        pipeline.InMemSort(pipeline.ArraySource(3, 2, 6, 7, 4)),
        pipeline.InMemSort(pipeline.ArraySource(7, 4, 0, 3, 2, 13, 8)))
    for v := range p {
        fmt.Println(v)
    }
}

pipeline.go

package pipeline

import (
    "encoding/binary"
    "io"
    "math/rand"
    "sort"
)

func ArraySource(a ...int) <-chan int {
    out := make(chan int)
    go func() {
        for _, v := range a {
            out <- v
        }
        close(out)
    }()
    return out
}

func InMemSort(in <-chan int) <-chan int {
    out := make(chan int)
    go func() {
        //Read into memory
        a := []int{}
        for v := range in {
            a = append(a, v)
        }
        //Sort
        sort.Ints(a)
        //Output
        for _, v := range a {
            out <- v
        }
        close(out)
    }()
    return out
}

func Merge(in1, in2 <-chan int) <-chan int {
    out := make(chan int)
    go func() {
        v1, ok1 := <-in1
        v2, ok2 := <-in2
        for ok1 || ok2 {
            if !ok2 || (ok1 && v1 <= v2) {
                out <- v1
                v1, ok1 = <-in1
            } else {
                out <- v2
                v2, ok2 = <-in2
            }
        }
        close(out)
    }()
    return out
}

func ReaderSource(reader io.Reader) <-chan int {
    out := make(chan int)
    go func() {
        buffer := make([]byte, 8)
        for {
            n, err := reader.Read(buffer)
            if n > 0 {
                v := int(
                    binary.BigEndian.Uint64(buffer))
                out <- v
            }
            if err != nil {
                break
            }
        }
        close(out)
    }()
    return out
}

func WriterSink(writer io.Writer, in <-chan int) {
    for v := range in {
        buffer := make([]byte, 8)
        binary.BigEndian.PutUint64(buffer, uint64(v))
        writer.Write(buffer)
    }
}

func RandomSource(count int) <-chan int {
    out := make(chan int)
    go func() {
        for i := 0; i < count; i++ {
            out <- rand.Int()
        }
        close(out)
    }()
    return out
}