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

在 Go 语言中， io包为 I/O 原语提供基本接口。它的主要工作是封装这种原语之王的持续实现。 Go 语言中的CopyBuffer()函数与Copy() 方法相同，但唯一的例外是它通过提供的缓冲区显示是否需要，而不是分配临时缓冲区。如果 src 由 WriterTo 实现或 dst 由 ReaderFrom 实现，则不会使用缓冲区来执行复制操作。而且，这个函数是在io包下定义的。在这里，您需要导入“io”包才能使用这些功能。

句法：

func CopyBuffer(dst Writer, src Reader, buf []byte) (written int64, err error)

这里，“dst”是目的地，“src”是内容被复制到目的地的来源，“buf”是允许内存中永久空间的缓冲区。

返回值：它返回复制到“dst”的 int64 类型的字节总数，并返回从 src 复制到 dst 时遇到的第一个错误（如果有）。但是，如果缓冲区为零，则分配一个，否则，如果缓冲区的长度为零，则 CopyBuffer 会发生混乱。

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

示例 1：

// Golang program to illustrate the usage of
// io.CopyBuffer() 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("GfG\n")
  
    // Defining destination using Stdout
    dst := os.Stdout
  
    // Defining buffer of length one using
    // make keyword
    buffer := make([]byte, 1)
  
    // Calling CopyBuffer method with its parameters
    bytes, err := io.CopyBuffer(dst, src, buffer)
  
    // If error is not nil then panics
    if err != nil {
        panic(err)
    }
  
    // Prints output
    fmt.Printf("The number of bytes are: %d\n", bytes)
}

输出：

GfG
The number of bytes are: 4

示例 2：

// Golang program to illustrate the usage of
// io.CopyBuffer() function
  
// Including main package
package main
  
// Importing fmt, io, os, and strings
import (
    "fmt"
    "io"
    "os"
    "strings"
)
  
// Calling main
func main() {
  
    // Defining two sources
    src1 := strings.NewReader("GfG\n")
    src2 := strings.NewReader("GeeksforGeeks is a CS-Portal\n")
  
    // Defining destination using Stdout
    dst := os.Stdout
  
    // Defining buffer of length one using
    // make keyword
    buffer := make([]byte, 1)
  
    // Calling CopyBuffer method with its parameters
    bytes1, err := io.CopyBuffer(dst, src1, buffer)
    bytes2, err := io.CopyBuffer(dst, src2, buffer)
  
    // If error is not nil then panics
    if err != nil {
        panic(err)
    }
  
    // Prints output
    fmt.Printf("The number of bytes are: %d\n", bytes1)
    fmt.Printf("The number of bytes are: %d\n", bytes2)
}

输出：

GfG
GeeksforGeeks is a CS-Portal
The number of bytes are: 4
The number of bytes are: 29

在这里，在上面的示例中，字符串的NewReader() 方法用于从要复制的内容读取的位置。此处使用“Stdout”以创建写入复制内容的默认文件描述符。而且，上面在调用 CopyBuffer() 方法时重用了相同的缓冲区，不需要额外的缓冲区分配。