Golang文件拷贝与移动操作实践

os.Rename

os.Rename

package main

import (
    "fmt"
    "io"
    "os"
    "path/filepath"
)

// moveFileSameFS 尝试在同一文件系统内移动文件
func moveFileSameFS(src, dst string) error {
    err := os.Rename(src, dst)
    if err != nil {
        // 检查是否是跨文件系统错误，如果是，可能需要回退到copy-then-delete
        linkErr, ok := err.(*os.LinkError)
        if ok && linkErr.Op == "rename" && linkErr.Err.Error() == "cross-device link" {
            return fmt.Errorf("os.Rename failed due to cross-device link, consider copy-then-delete: %w", err)
        }
        return fmt.Errorf("failed to move file with os.Rename: %w", err)
    }
    return nil
}

// copyFile 拷贝文件，处理权限和错误
func copyFile(src, dst string) error {
    sourceFileStat, err := os.Stat(src)
    if err != nil {
        return fmt.Errorf("failed to stat source file %s: %w", src, err)
    }

    if !sourceFileStat.Mode().IsRegular() {
        return fmt.Errorf("source %s is not a regular file", src)
    }

    source, err := os.Open(src)
    if err != nil {
        return fmt.Errorf("failed to open source file %s: %w", src, err)
    }
    defer source.Close()

    destination, err := os.Create(dst)
    if err != nil {
        return fmt.Errorf("failed to create destination file %s: %w", dst, err)
    }
    defer destination.Close()

    _, err = io.Copy(destination, source)
    if err != nil {
        return fmt.Errorf("failed to copy content from %s to %s: %w", src, dst, err)
    }

    // 尝试复制文件权限
    err = os.Chmod(dst, sourceFileStat.Mode())
    if err != nil {
        fmt.Printf("Warning: failed to set permissions for %s: %v\n", dst, err)
    }

    return nil
}

// moveFileRobust 实现一个健壮的文件移动操作，支持跨文件系统
func moveFileRobust(src, dst string) error {
    // 尝试原子性移动
    err := moveFileSameFS(src, dst)
    if err == nil {
        return nil // 成功原子移动
    }

    // 如果原子移动失败（可能是跨文件系统），则回退到拷贝-删除策略
    fmt.Printf("Atomic move failed for %s to %s, attempting copy-then-delete: %v\n", src, dst, err)

    // 拷贝文件
    err = copyFile(src, dst)
    if err != nil {
        return fmt.Errorf("failed to copy file during robust move: %w", err)
    }

    // 拷贝成功后删除源文件
    err = os.Remove(src)
    if err != nil {
        // 这里需要特别注意，如果删除失败，就意味着目标文件已存在，但源文件未删除，需要手动处理
        return fmt.Errorf("successfully copied %s to %s, but failed to remove source file: %w", src, dst, err)
    }

    return nil
}

func main() {
    // 示例用法
    // 创建一些测试文件
    os.WriteFile("test_src.txt", []byte("Hello, Go!"), 0644)
    os.MkdirAll("temp_dir", 0755)
    os.WriteFile("temp_dir/another_src.txt", []byte("Another file."), 0644)

    // 1. 同一文件系统内的移动
    fmt.Println("--- Test Same FS Move ---")
    err := moveFileRobust("test_src.txt", "test_dst.txt")
    if err != nil {
        fmt.Println("Error moving test_src.txt:", err)
    } else {
        fmt.Println("Moved test_src.txt to test_dst.txt successfully.")
    }
    // 清理
    os.Remove("test_dst.txt")

    // 2. 模拟跨文件系统移动（实际操作中，这需要两个不同的挂载点）
    // 这里我们通过先删除目标，再用 copy-then-delete 来模拟
    fmt.Println("\n--- Test Cross FS Move Simulation ---")
    // 假设 "temp_dir/another_src.txt" 在一个不同的文件系统上
    // 实际场景中，dstPath 可能是 /mnt/usb/another_src.txt
    dstPath := filepath.Join("temp_dir", "moved_another_src.txt")
    err = moveFileRobust("temp_dir/another_src.txt", dstPath)
    if err != nil {
        fmt.Println("Error moving another_src.txt:", err)
    } else {
        fmt.Println("Moved temp_dir/another_src.txt to", dstPath, "successfully.")
    }

    // 清理
    os.RemoveAll("temp_dir")
}

os.Rename

rename

rename

EXDEV

syscall.EXDEV

cross-device link

rename

os.Rename

moveFileRobust

os.Rename

io.Copy

io.Reader

io.Writer

io.Copy

os.Stat

FileInfo

os.Chmod

os.Chtimes

os.Chown

io.Copy

copyFile

defer source.Close()

defer destination.Close()

os.Create

os.Stat

io.Copy

io.Copy

os.FileMode

0644