首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >厚土Go学习笔记 | 32. Readers读取数据流

厚土Go学习笔记 | 32. Readers读取数据流

作者头像
李海彬
发布2018-03-28 11:58:24
7990
发布2018-03-28 11:58:24
举报
文章被收录于专栏:Golang语言社区Golang语言社区

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 上下载。

Github: https://github.com/golang/tour

本文参与 腾讯云自媒体分享计划,分享自微信公众号。
原始发表:2017-12-28,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 Golang语言社区 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档