Go 中复制文件的三种方式

在 Go 语言中，复制文件是一项常见的操作。以下将介绍三种常见的复制文件的方式。

方式一：使用缓冲读取和写入

通过创建缓冲区来逐块读取源文件，并将数据写入目标文件。这种方式可以有效地控制内存使用，并在处理大文件时避免一次性将整个文件读入内存。

package main

import (
    "fmt"
    "io"
    "os"
)

func copyFileBuffered(src, dst string) error {
    srcFile, err := os.Open(src)
    if err!= nil {
        return err
    }
    defer srcFile.Close()

    dstFile, err := os.Create(dst)
    if err!= nil {
        return err
    }
    defer dstFile.Close()

    buf := make([]byte, 4096)
    for {
        n, err := srcFile.Read(buf)
        if err!= nil && err!= io.EOF {
            return err
        }
        if n == 0 {
            break
        }

        _, err = dstFile.Write(buf[:n])
        if err!= nil {
            return err
        }
    }

    return nil
}

方式二：使用 io.Copy 函数

io.Copy 函数会自动处理缓冲区和数据的读写，使代码更加简洁。

package main

import (
    "fmt"
    "io"
    "os"
)

func copyFileUsingIoCopy(src, dst string) error {
    srcFile, err := os.Open(src)
    if err!= nil {
        return err
    }
    defer srcFile.Close()

    dstFile, err := os.Create(dst)
    if err!= nil {
        return err
    }
    defer dstFile.Close()

    _, err = io.Copy(dstFile, srcFile)
    return err
}

方式三：逐字节复制

这种方式虽然简单直观，但在性能上可能不如前两种方式，特别是对于大文件。

package main

import (
    "fmt"
    "io"
    "os"
)

func copyFileByteByByte(src, dst string) error {
    srcFile, err := os.Open(src)
    if err!= nil {
        return err
    }
    defer srcFile.Close()

    dstFile, err := os.Create(dst)
    if err!= nil {
        return err
    }
    defer dstFile.Close()

    var b byte
    for {
        _, err := srcFile.Read(&b)
        if err!= nil && err!= io.EOF {
            return err
        }
        if err == io.EOF {
            break
        }
        err = dstFile.Write([]byte{b})
        if err!= nil {
            return err
        }
    }

    return nil
}

在实际应用中，根据文件的大小、性能要求和具体场景，选择合适的复制方式可以提高程序的效率和稳定性。缓冲读取和写入以及使用 io.Copy 函数通常是更推荐的方式，特别是在处理较大文件时。

以上就是 Go 中复制文件的三种常见方式，希望对您在实际开发中有所帮助。