17370845950

如何在 Golang 中实现文件复制_Golang 文件操作与异常处理示例
使用 io.Copy 结合 os.Open 和 os.Create 是 Go 中高效复制文件的推荐方法,示例代码展示了打开源文件、创建目标文件、调用 io.Copy 执行复制、同步数据到磁盘的完整流程;为保留权限,可读取源文件元信息并用 Chmod 应用到目标文件;通过 defer 确保资源释放,逐项检查错误以实现健壮的异常处理。

在 Golang 中实现文件复制并不复杂,但需要兼顾性能和异常处理。Go 标准库提供了足够的工具来完成这一任务,关键在于正确使用 osio 包,并对可能出现的错误进行合理捕获与反馈。

使用 io.Copy 实现高效文件复制

最推荐的方式是结合 os.Openos.Create 使用 io.Copy。这种方式利用底层优化的复制逻辑,避免手动管理缓冲区带来的性能损耗。

注意:复制大文件时应避免一次性读入内存。

示例代码如下:

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)
    if err != nil {
        return err
    }

    // 显式提交写入内容到磁盘
    err = destination.Sync()
    if err != nil {
        return err
    }

    return nil
}

该方法自动处理分块读取,适合大多数场景。

保留文件元信息(如权限)

如果需要目标文件保持源文件的权限和模式,可以在复制后设置对应属性。

获取源文件信息并应用到目标文件:

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

    info, err := source.Stat()
    if err != nil {
        return err
    }

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

    _, err = io.Copy(destination, source)
    if err != nil {
        return err
    }

    err = destination.Chmod(info.Mode())
    if err != nil {
        return err
    }

    return destination.Sync()
}

这样可以确保复制后的文件权限一致,适用于配置文件或可执行脚本的复制。

完整异常处理与资源释放

Go 的 defer 能保证文件句柄关闭,但仍需注意多个错误来源。例如打开源文件失败、创建目标文件失败、复制过程出错、同步失败等。

建议统一错误处理流程:

  • 每个 I/O 操作后检查 error
  • 使用 defer file.Close() 防止资源泄露
  • Sync() 的调用增加健壮性,防止缓存未写入

还可以封装返回带上下文的错误信息,便于排查问题:

if err := copyFile("a.txt", "b.txt"); err != nil {
    log.Fatalf("文件复制失败: %v", err)
}

基本上就这些。Golang 的文件操作简洁高效,只要注意错误处理和资源管理,就能写出稳定可靠的复制逻辑。