go语言的io包指定了io.Reader接口。go语言标准库包含了这个接口的许多实现,包括文件、网络连接、压缩、加密等等。
io.Reader接口有一个Read方法:
func (T) Read(b []byte) (n int, err error)
在程序中,我们使用循环读取数据流,直到error返回io.EOF。
我们建立一个每次以8个字节读取strings.Reader的输出的程序示例
package mainimport("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 mainimport ("fmt""io""os")type MyReader struct{}func Validate(r io.Reader) {b := make([]byte, 1024, 2048)i, o := 0, 0for ; i < 1<<20 && o < 1<<20; i++ { // test 1mbn, 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 += nif 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 := 0for ;i < len(b);i++ {b[i] = 'A'}return i, nil}func main() {Validate(MyReader{})}

