32. Readers读取数据流

作者: 厚土火焱 | 来源:发表于2017-08-20 09:47 被阅读125次

    go语言的 io 包指定了 io.Reader 接口。go语言标准库包含了这个接口的许多实现,包括文件、网络连接、压缩、加密等等。
    io.Reader 接口有一个 Read 方法:

    func (T) Read(b []byte) (n int, err error)
    

    在程序中,我们使用循环读取数据流,直到 error 返回 io.EOF 。
    我们建立一个每次以 8 个字节读取 strings.Reader 的输出的程序示例。

    package main
    import(
        "fmt"
        "strings"
        "io"
    )
    
    func main() {
        r := strings.NewReader("Hello, Reader!")
        b := make([]byte, 8)    // 8 这里控制每次读取的字节数
        for{
            n, err := r.Read(b)
            fmt.Printf("n = %v err = %v b = %v\n", n, err, b)
            fmt.Printf("b[:n] = %q\n", b[:n])
            if err == io.EOF{
                break
            }
        }
    }
    

    运行结果

    n = 8 err = <nil> b = [72 101 108 108 111 44 32 82]
    b[:n] = "Hello, R"
    n = 6 err = <nil> b = [101 97 100 101 114 33 32 82]
    b[:n] = "eader!"
    n = 0 err = EOF b = [101 97 100 101 114 33 32 82]
    b[:n] = ""
    

    再展示一个例子吧。这个例子,实现一个 Reader 类型,它不断生成 ASCII 字符 ‘A’ 的流。

    package main
    
    import (
        "fmt"
        "io"
        "os"
    )
    
    type MyReader struct{}
    
    func Validate(r io.Reader) {
        b := make([]byte, 1024, 2048)
        i, o := 0, 0
        for ; i < 1<<20 && o < 1<<20; i++ { // test 1mb
            n, err := r.Read(b)
            for i, v := range b[:n] {
                if v != 'A' {
                    fmt.Fprintf(os.Stderr, "got byte %x at offset %v, want 'A'\n", v, o+i)
                    return
                }
            }
            o += n
            if err != nil {
                fmt.Fprintf(os.Stderr, "read error: %v\n", err)
                return
            }
        }
        if o == 0 {
            fmt.Fprintf(os.Stderr, "read zero bytes after %d Read calls\n", i)
            return
        }
        fmt.Println("OK!")
    }
    //实现一个 Reader 类型,它不断生成 ASCII 字符 'A' 的流。
    // TODO: Add a Read([]byte) (int, error) method to MyReader.
    func (mr MyReader) Read(b []byte) (n int, err error)  {
        i := 0
        for ;i < len(b);i++ {
            b[i] = 'A'
        }
        return i, nil
    }
    
    func main() {
        Validate(MyReader{})
    }
    

    代码中的 func Validate(r io.Reader) 可以从 github 上下载。
    golang.org/x/tour/reader

    相关文章

      网友评论

        本文标题:32. Readers读取数据流

        本文链接:https://www.haomeiwen.com/subject/xzotdxtx.html