一文了解Go语言io.Copy函数

func Copy(dst io.Writer, src io.Reader) (written int64, err error)

package main
import (
        "fmt"
        "io"
        "os"
)
func main() {
        fmt.Print("请输入一个字符串：")
        src := readString()
        // 通过io.Copy 函数能够将 src 的全部数据 拷贝到 控制台上输出
        written, err := io.Copy(os.Stdout, src)
        if err != nil {
                fmt.Println("复制过程中发生错误：", err)
                return
        }
        fmt.Printf("\n成功复制了 %d 个字节。\n", written)
}
func readString() io.Reader {
   buffer := make([]byte, 1024)
   n, _ := os.Stdin.Read(buffer)
   // 如果实际读取的字节数少于切片长度，则截取切片
   if n < len(buffer) {
      buffer = buffer[:n]
   }
   return strings.NewReader(string(buffer))
}

func Copy(dst Writer, src Reader) (written int64, err error) {
   // Copy 函数 调用了 copyBuffer 函数来实现
   return copyBuffer(dst, src, nil)
}
func copyBuffer(dst Writer, src Reader, buf []byte) (written int64, err error) {
   // 如果 源Reader 实现了 WriterTo 接口,直接调用该方法 将数据写入到 目标Writer 当中
   if wt, ok := src.(WriterTo); ok {
      return wt.WriteTo(dst)
   }
   // 同理，如果 目标Writer 实现了 ReaderFrom 接口,直接调用ReadFrom方法
   if rt, ok := dst.(ReaderFrom); ok {
      return rt.ReadFrom(src)
   }
   // 如果没有传入缓冲区，此时默认 创建一个 缓冲区
   if buf == nil {
      // 默认缓冲区 大小为 32kb
      size := 32 * 1024
      // 如果源Reader 为LimitedReader, 此时比较 可读数据数 和 默认缓冲区，取较小那个
      if l, ok := src.(*LimitedReader); ok && int64(size) > l.N {
         if l.N < 1 {
            size = 1
         } else {
            size = int(l.N)
         }
      }
      buf = make([]byte, size)
   }
   for {
      // 调用Read方法 读取数据
      nr, er := src.Read(buf)
      if nr > 0 {
         // 将数据写入到 目标Writer 当中
         nw, ew := dst.Write(buf[0:nr])
         // 判断写入是否 出现了 错误
         if nw < 0 || nr < nw {
            nw = 0
            if ew == nil {
               ew = errInvalidWrite
            }
         }
         // 累加 总写入数据
         written += int64(nw)
         if ew != nil {
            err = ew
            break
         }
         // 写入字节数 小于 读取字节数,此时报错
         if nr != nw {
            err = ErrShortWrite
            break
         }
      }
      if er != nil {
         if er != EOF {
            err = er
         }
         break
      }
   }
   return written, err
}

package main
import (
        "fmt"
        "io"
        "os"
)
func main() {
        sourceFile := "source.txt"
        destinationFile := "destination.txt"
        // 打开源文件
        src, err := os.Open(sourceFile)
        if err != nil {
                fmt.Println("无法打开源文件:", err)
                return
        }
        // 调用Close方法
        defer src.Close()
        // 创建目标文件
        dst, err := os.Create(destinationFile)
        if err != nil {
                fmt.Println("无法创建目标文件:", err)
                return
        }
        // 调用Close 方法
        defer dst.Close()
        // 执行文件复制
        _, err = io.Copy(dst, src)
        if err != nil {
                fmt.Println("复制文件出错:", err)
                return
        }
        fmt.Println("文件复制成功!")
}

func BenchmarkCopyWithBufferSize(b *testing.B) {
   // 本地运行时, 文件大小为 100 M
   filePath := "largefile.txt"
   bufferSizes := []int{32 * 1024, 64 * 1024, 128 * 1024} // 不同的缓冲区大小
   for _, bufferSize := range bufferSizes {
      b.Run(fmt.Sprintf("BufferSize-%d", bufferSize), func(b *testing.B) {
         for n := 0; n < b.N; n++ {
            src, _ := os.Open(filePath)
            dst, _ := os.Create("destination.txt")
            buffer := make([]byte, bufferSize)
            _, _ = io.CopyBuffer(dst, src, buffer)
            _ = src.Close()
            _ = dst.Close()
            _ = os.Remove("destination.txt")
         }
      })
   }
}