Golang io.LimitReader()函数的用法及示例

在Golang中，io.LimitReader()函数是用于返回一个io.Reader接口的实现，该实现只允许读取一定数量的字节。这个函数主要用于限制reader读取操作的范围，防止出现过度读取的情况。

下面是使用io.LimitReader()函数的一些示例：

示例一

package main

import (
    "fmt"
    "io"
    "strings"
)

func main() {
    // 定义reader
    reader := strings.NewReader("Hello, world!")

    // 限制reader读取前5个字节
    lr := io.LimitReader(reader, 5)

    // 将结果输出到控制台
    io.CopyN(io.Stdout, lr, 5)
}

在示例一中，我们定义一个字符串reader，并使用io.LimitReader()函数将reader限制为只能读取前5个字节。最终，控制台输出的结果为“Hello”。

示例二

package main

import (
    "fmt"
    "io"
    "strings"
)

func main() {
    // 定义reader
    reader := strings.NewReader("Hello, world!")

    // 限制reader读取前5个字节
    lr := io.LimitReader(reader, 5)

    // 使用byte数组保存读取的结果
    buf := make([]byte, 5)
    io.ReadFull(lr, buf)

    // 将结果输出到控制台
    fmt.Println(string(buf))
}

在示例二中，我们同样使用io.LimitReader()函数将reader限制为只能读取前5个字节。不过，这次我们使用io.ReadFull()函数读取5个字节到一个byte数组中。最终，控制台输出的结果为“Hello”。

示例三

package main

import (
    "fmt"
    "io"
    "strings"
)

func main() {
    // 定义reader
    reader := strings.NewReader("Hello, world!")

    // 限制reader读取前100个字节（实际上只有13个字节可读取）
    lr := io.LimitReader(reader, 100)

    // 使用byte数组保存读取的结果
    buf := make([]byte, 100)
    n, _ := io.ReadFull(lr, buf)

    // 将结果输出到控制台
    fmt.Println(string(buf[:n]))
}

在示例三中，我们将reader限制为只能读取前100个字节，但实际上只有13个字节可读取。我们同样使用io.ReadFull()函数读取所有可读的字节。最终，控制台输出的结果为“Hello, world!”。

总的来说，使用io.LimitReader()函数可以很方便地限制reader的读取操作，保证读取的字节数不超过限制的范围。