Golang 中的 IO 包详解:结构体类型介绍

开发 后端
实现了Io.Reader、Io.ReaderAt 和 Io.Seeker 接口的类型,用于在一个 Reader 中只读取某部分的数据。在使用 Io.SectionReader 时,通常是将其作为参数传递给其他需要 ReaderAt 或 Seeker 接口的函数,并在该函数中使用 ReadAt 或 Seek 方法来访问数据。

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
}

实现了 io.Reader 接口,并且进行了功能扩展。R 表示 io.Reader 对象,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)
	}
}

当读取的字节数超过限制时,LimitedReader 会自动终止读取并返回一个 io.EOF 错误,表示已经达到了总字节数的限制。

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
}

实现了 io.Reader、io.ReaderAt 和 io.Seeker 接口的类型,用于在一个 Reader 中只读取某部分的数据。在使用 io.SectionReader 时,通常是将其作为参数传递给其他需要 ReaderAt 或 Seeker 接口的函数,并在该函数中使用 ReadAt 或 Seek 方法来访问数据。简单示例如下:

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
}

实现了 io.Reader 和 io.Writer 接口的类型,可以将输入流的内容复制到一个指定的输出流中。简单示例如下:

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 中读取数据的类型。io.Pipe 实际上是一个管道,可以用于在同一个进程中的不同 goroutine 之间传输数据。PipeReader 实际上是通过 io.Pipe 返回的读取端实例。使用起来非常简单,可以通过 io.Pipe 函数创建一个 Pipe 实例,io.Pipe 函数返回的是两个值分别是 io.PipeReader 和 io.PipeWriter 类型的指针,前者用于从管道中读取数据,后者用于向管道中写入数据。简单示例如下:

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 中写入数据的类型。io.Pipe 实际上是一个管道,可以用于在同一个进程中的不同 goroutine 之间传输数据。PipeWriter 实际上是通过 io.Pipe 返回的写入端实例。io.Pipe 使用起来非常简单,可以通过 io.Pipe 函数创建一个 Pipe 实例,io.Pipe 函数返回的是两个值,分别是 io.PipeReader 和 io.PipeWriter 类型的指针,前者用于从管道中读取数据,后者用于向管道中写入数据。

责任编辑:姜华 来源: 今日头条
相关推荐

2023-08-03 08:48:07

Golang接口

2023-08-02 09:07:27

Golangio 包

2023-06-09 08:16:09

GolangStruct Tag

2023-08-31 09:28:12

Golang可导出函数

2023-08-08 14:51:29

2023-08-07 09:18:32

Golang偏移量接口

2022-01-09 23:04:19

语言打印结构体

2023-11-27 15:02:37

BytesGolang

2023-10-18 08:22:38

BufioGolang

2023-10-31 09:10:39

2023-05-12 09:40:53

ContextGolang

2024-01-18 09:07:04

Errors函数Golang

2023-09-05 08:22:44

Golangstrings 包

2023-10-07 09:08:32

Golangbufio

2023-11-03 08:53:15

StrconvGolang

2023-10-10 08:57:44

Golangbufio

2023-11-07 09:02:07

Golangbytes

2023-09-07 07:35:54

GolangBufio

2023-09-06 09:10:04

Golang字符串

2023-09-04 08:17:37

Golangstrings 包
点赞
收藏

51CTO技术栈公众号