17370845950

如何通过Golang实现文件复制_Golang文件拷贝操作实践
Go语言中推荐用io.Copy高效复制文件,支持自动缓冲;需手动控制缓冲时可自定义读写循环;保留权限需额外调用os.Chmod。

在Go语言中实现文件复制并不复杂,标准库提供了足够的支持来完成这一常见任务。最核心的方法是使用 osio 包配合操作。下面介绍几种常用的文件拷贝方式,并说明其适用场景和注意事项。

使用 io.Copy 进行高效拷贝

这是推荐的主流做法。通过 os.Open 打开源文件,os.Create 创建目标文件,再用 io.Copy 完成数据传输。该方法内部使用了默认缓冲区(32KB),性能良好且代码简洁。

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

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

    _, err = io.Copy(destination, source)
    return err
}

这种方式自动处理缓冲,无需手动管理读写块大小,适合绝大多数场景。

手动控制缓冲区进行复制

如果你需要自定义缓冲区大小(例如受限内存环境),可以手动实现读写循环:

func copyFileWithBuffer(src, dst string) error {
    buffer := make([]byte, 1024*1024) // 1MB 缓冲

    source, err := os.Open(src)
    if err != nil {
        return err
    }
    defer source.Close()

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

    for {
        n, err := source.Read(buffer)
        if err == io.EOF {
            break
        }
        if err != nil {
            return err
        }

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

这种方法灵活,但通常没必要,除非有特殊性能调优需求。

利用 io.Copy 配合文件信息保留元数据

上述方法不会复制文件权限等元数据。若需保留原始文件的权限模式,可以在拷贝后设置:

func copyFileWithMode(src, dst string) error {
    info, err := os.Stat(src)
    if err != nil {
        return err
    }

    err = copyFile(src, dst) // 使用前面定义的拷贝函数
    if err != nil {
        return err
    }

    return os.Chmod(dst, info.Mode())
}

这样能确保目标文件具有与源文件相同的访问权限。

基本上就这些。对于日常开发,直接使用 io.Copy 就足够了,简单、安全、高效。如果涉及大文件或特殊需求,再考虑定制缓冲或并发策略。不复杂但容易忽略的是关闭文件和错误处理,记得 always defer Close。