Golang 中的 IO 包詳解：結(jié)構(gòu)體類型介紹
io.LimitedReader
// A LimitedReader reads from R but limits the amount of
// data returned to just N bytes. Each call to Read
// updates N to reflect the new amount remaining.
// Read returns EOF when N <= 0 or when the underlying R returns EOF.
type LimitedReader struct {
R Reader // underlying reader
N int64 // max bytes remaining
}實(shí)現(xiàn)了 io.Reader 接口，并且進(jìn)行了功能擴(kuò)展。R 表示 io.Reader 對(duì)象，N 表示最多允許讀取的字節(jié)數(shù)。簡(jiǎn)單示例如下所示：
package main
import (
"bytes"
"fmt"
"io"
)
func main() {
data := []byte("hello, world!")
reader := io.LimitReader(bytes.NewReader(data), 5)
buf := make([]byte, 10)
n, err := reader.Read(buf)
if err == nil {
fmt.Printf("%s\n", buf[:n])
} else {
fmt.Printf("read error: %s\n", err)
}
}當(dāng)讀取的字節(jié)數(shù)超過限制時(shí)，LimitedReader 會(huì)自動(dòng)終止讀取并返回一個(gè) io.EOF 錯(cuò)誤，表示已經(jīng)達(dá)到了總字節(jié)數(shù)的限制。
io.SectionReader
// SectionReader implements Read, Seek, and ReadAt on a section
// of an underlying ReaderAt.
type SectionReader struct {
r ReaderAt
base int64
off int64
limit int64
}實(shí)現(xiàn)了 io.Reader、io.ReaderAt 和 io.Seeker 接口的類型，用于在一個(gè) Reader 中只讀取某部分的數(shù)據(jù)。在使用 io.SectionReader 時(shí)，通常是將其作為參數(shù)傳遞給其他需要 ReaderAt 或 Seeker 接口的函數(shù)，并在該函數(shù)中使用 ReadAt 或 Seek 方法來訪問數(shù)據(jù)。簡(jiǎn)單示例如下：
package main
import (
"bytes"
"fmt"
"io"
)
func main() {
data := []byte("hello, world!")
reader := bytes.NewReader(data)
sectionReader := io.NewSectionReader(reader, 0, 6)
buf := make([]byte, 5)
n, err := sectionReader.ReadAt(buf, 0)
if err == nil {
fmt.Printf("%s\n", buf[:n])
} else {
fmt.Printf("read error: %s\n", err)
}
}io.teeReader
type teeReader struct {
r Reader
w Writer
}實(shí)現(xiàn)了 io.Reader 和 io.Writer 接口的類型，可以將輸入流的內(nèi)容復(fù)制到一個(gè)指定的輸出流中。簡(jiǎn)單示例如下：
package main
import (
"bytes"
"fmt"
"io"
)
func main() {
data := []byte("hello, world!")
buf1 := bytes.NewBuffer(nil)
buf2 := bytes.NewBuffer(nil)
reader := bytes.NewReader(data)
tee := io.TeeReader(reader, io.MultiWriter(buf1, buf2))
buf := make([]byte, 10)
n, err := tee.Read(buf)
if err == nil {
fmt.Printf("%s\n", buf[:n])
fmt.Printf("%s\n", buf1.Bytes())
fmt.Printf("%s\n", buf2.Bytes())
} else {
fmt.Printf("read error: %s\n", err)
}
}io.PipeReader
// A PipeReader is the read half of a pipe.
type PipeReader struct {
p *pipe
}io.PipeReader 用于從 io.Pipe 中讀取數(shù)據(jù)的類型。io.Pipe 實(shí)際上是一個(gè)管道，可以用于在同一個(gè)進(jìn)程中的不同 goroutine 之間傳輸數(shù)據(jù)。PipeReader 實(shí)際上是通過 io.Pipe 返回的讀取端實(shí)例。使用起來非常簡(jiǎn)單，可以通過 io.Pipe 函數(shù)創(chuàng)建一個(gè) Pipe 實(shí)例，io.Pipe 函數(shù)返回的是兩個(gè)值分別是 io.PipeReader 和 io.PipeWriter 類型的指針，前者用于從管道中讀取數(shù)據(jù)，后者用于向管道中寫入數(shù)據(jù)。簡(jiǎn)單示例如下：
package main
import (
"bufio"
"fmt"
"io"
)
func main() {
pr, pw := io.Pipe()
go func() {
pw.Write([]byte("hello, world!"))
pw.Close()
}()
br := bufio.NewReader(pr)
line, isPrefix, err := br.ReadLine()
fmt.Println(line, isPrefix, err)
}io.PipeWriter
// A PipeWriter is the write half of a pipe.
type PipeWriter struct {
p *pipe
}io.PipeWriter 是用于向 io.Pipe 中寫入數(shù)據(jù)的類型。io.Pipe 實(shí)際上是一個(gè)管道，可以用于在同一個(gè)進(jìn)程中的不同 goroutine 之間傳輸數(shù)據(jù)。PipeWriter 實(shí)際上是通過 io.Pipe 返回的寫入端實(shí)例。io.Pipe 使用起來非常簡(jiǎn)單，可以通過 io.Pipe 函數(shù)創(chuàng)建一個(gè) Pipe 實(shí)例，io.Pipe 函數(shù)返回的是兩個(gè)值，分別是 io.PipeReader 和 io.PipeWriter 類型的指針，前者用于從管道中讀取數(shù)據(jù)，后者用于向管道中寫入數(shù)據(jù)。
