Golang 中的 io.CopyN()函数示例

在 Go 语言中， io包为 I/O 原语提供基本接口。它的主要工作是封装这种原语之王的持续实现。 Go 语言中的CopyN()函数用于将“n”个字节从源复制到目标。如果dst是由ReaderFrom接口实现的，那么副本就是利用它来实现的。而且，这个函数是在io包下定义的。在这里，您需要导入“io”包才能使用这些功能。

句法：

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

这里，“dst”是目的地，“src”是将内容复制到目的地的来源，“n”是要从来源复制的字节数限制。

返回值：它返回复制到“dst”的 int64 类型的字节总数，并返回从 src 复制到 dst 时遇到的第一个错误（如果有）。此外，当且仅当错误为“nil”时，返回的字节数为“n”。

下面的例子说明了上述方法的使用：

示例 1：

// Golang program to illustrate the usage of
// io.CopyN() function
  
// Including main package
package main
  
// Importing fmt, io, os, and strings
import (
    "fmt"
    "io"
    "os"
    "strings"
)
  
// Calling main
func main() {
  
    // Defining source
    src := strings.NewReader("GeeksforGeeks\n")
  
    // Defining destination using Stdout
    dst := os.Stdout
  
    // Calling CopyN method with its parameters
    bytes, err := io.CopyN(dst, src, 5)
  
    // If error is not nil then panics
    if err != nil {
        panic(err)
    }
  
    // Prints output
    fmt.Printf("\nThe number of bytes are: %d\n", bytes)
}

输出：

Geeks
The number of bytes are: 5

这里，返回的字节数等于“n”，因为错误是“nil”。

示例 2：

// Golang program to illustrate the usage of
// io.CopyN() function
  
// Including main package
package main
  
// Importing fmt, io, os, and strings
import (
    "fmt"
    "io"
    "os"
    "strings"
)
  
// Calling main
func main() {
  
    // Defining source
    src := strings.NewReader("CS-Portal\n")
  
    // Defining destination using Stdout
    dst := os.Stdout
  
    // Calling CopyN method with its parameters
    bytes, err := io.CopyN(dst, src, 20)
  
    // If error is not nil then panics
    if err != nil {
        panic(err)
    }
  
    // Prints output
    fmt.Printf("The number of bytes are: %d\n", bytes)
}

输出：

CS-Portal
panic: EOF

goroutine 1 [running]:
main.main()
    /tmp/sandbox691649995/prog.go:29 +0x137

在这里，在上面的示例中，字符串的NewReader() 方法用于从要复制的内容读取的位置。此处使用“Stdout”以创建写入复制内容的默认文件描述符。此外，这里返回的字节数小于“n”，因此上面会抛出 EOF 错误。