Golang io.CopyN 用法详解 - 复制指定字节数 - 常规

Golang io.CopyN 用法详解：复制指定字节数

在 Golang 中，io.CopyN 函数可以帮助我们方便地将数据从 Reader 复制到 Writer 中，特别适用于需要复制指定字节数的情况。

函数签名

func CopyN(dst Writer, src Reader, n int64) (written int64, err error)

参数说明

dst: 目标 Writer，数据将被复制到这里。
src: 源 Reader，数据将从这里读取。
n: 要复制的字节数。

返回值

written: 实际复制的字节数。
err: 可能发生的错误。

代码示例

下面的示例演示了如何使用 io.CopyN 函数将一个文件的前 1024 个字节复制到另一个文件：

package main

import (
    'io'
    'os'
)

func main() {
    // 打开源文件
    src, err := os.Open('src.txt')
    if err != nil {
        panic(err)
    }
    defer src.Close()

    // 创建目标文件
    dst, err := os.Create('dst.txt')
    if err != nil {
        panic(err)
    }
    defer dst.Close()

    // 复制 1024 个字节
    written, err := io.CopyN(dst, src, 1024)
    if err != nil {
        panic(err)
    }

    println('Written:', written)
}

代码解释：

首先，使用 os.Open 函数打开源文件 src.txt。
然后，使用 os.Create 函数创建目标文件 dst.txt。
调用 io.CopyN 函数将源文件的前 1024 个字节复制到目标文件。
最后，输出实际复制的字节数。

注意事项：

使用 io.CopyN 函数后，需要手动关闭源 Reader 和目标 Writer。
如果源文件的长度小于指定的字节数，io.CopyN 函数将复制整个文件内容，并返回实际复制的字节数。

希望本文能够帮助您理解和使用 Golang 中的 io.CopyN 函数。