go語言的 io 包指定了 io.Reader 接口。go語言標(biāo)準(zhǔn)庫包含了這個接口的許多實(shí)現(xiàn)膏潮，包括文件、網(wǎng)絡(luò)連接满力、壓縮焕参、加密等等轻纪。
io.Reader 接口有一個 Read 方法：
func (T) Read(b []byte) (n int, err error)
在程序中，我們使用循環(huán)讀取數(shù)據(jù)流叠纷，直到 error 返回 io.EOF 刻帚。
我們建立一個每次以 8 個字節(jié)讀取 strings.Reader 的輸出的程序示例。
package main
import(
"fmt"
"strings"
"io"
)
func main() {
r := strings.NewReader("Hello, Reader!")
b := make([]byte, 8) // 8 這里控制每次讀取的字節(jié)數(shù)
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
}
}
}
運(yùn)行結(jié)果
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] = ""
再展示一個例子吧涩嚣。這個例子崇众，實(shí)現(xià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!")
}
//實(shí)現(xiàn)一個 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
