Golang进阶-复制文件

1. 文件复制原理:读取源文件的数据,把读到的数据再写到目标文件中;如果文件比较小,一次读写就可以了;如果文件比较大,则便边读边写

2. 使用io.read和io.write来实现

func main() {
    srcFile := "E:/GoPath/src/a/1.png"
    desFile := "E:/GoPath/src/a/2.png"
    total,err := CopyFile1(srcFile, desFile)
    fmt.Println(total, err)
}

func CopyFile1(srcFile, destFile string)(int, error)  {
    filel,err := os.Open(srcFile)
    if err != nil {
        return 0, err
    }

    file2,err := os.OpenFile(destFile,os.O_WRONLY|os.O_CREATE,os.ModePerm)
    if err != nil {
        return 0, err
    }

    defer filel.Close()
    defer file2.Close()

    //读写
    bs := make([]byte, 1024, 1024)
    n := -1 //读取的数据量
    total := 0
    for{
        n,err = filel.Read(bs)
        if err == io.EOF || n == 0 {
            fmt.Println("拷贝完毕")
            break
        } else if err != nil {
            fmt.Println("报错了:", err)
            return total, err
        }
        total += n
        file2.Write(bs[:n])
    }

    return total,nil
}

如果文件比较大,可以把切片bs的长度写大一些

3. 使用io包的copy方法

func CopyFile2(srcFile, destFile string)(int64, error)  {
    file1,err := os.Open(srcFile)
    if err != nil {
        return 0, err
    }

    file2,err := os.OpenFile(destFile,os.O_WRONLY|os.O_CREATE,os.ModePerm)
    if err != nil {
        return 0, err
    }

    defer file1.Close()
    defer file2.Close()

    return io.Copy(file2, file1)
}

4. 使用ioutil的读写,一次性读取,一次性写入;不适合较大的文件复制,容易内存溢出

func CopyFile3(srcFile, destFile string)(int, error)  {
    bs,err := ioutil.ReadFile(srcFile)
    if err != nil {
        return 0,err
    }
    err = ioutil.WriteFile(destFile,bs,0777)
    if err != nil {
        return 0,err
    }
    return len(bs), nil
}

5. 总结,后两种方法的性能还是不错的;扩展:CopyN(), CopyBuffer();无论哪个copy方法最终都是有copyBuffer这个私有方法实现的

标签: none

添加新评论