Golang 文件的复制 手动实现 与 库函数实现

手动实现

思路就是:先创建一个切片 然后通过Read方法操作源文件把数据读到切片中,然后再通过Write方法把切片里面的数据写到目标文件里面。

具体怎么使用呢?
举个例子:
如果我有一个文件test.txt,然后我想要把这个文件复制到destination.txt,就可以这样写:

package main

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

func main() {
    // 拷贝文件
    srcFile := "/Users/liberhome/GolandProjects/awesomeProject/I-package/April28_Go_io/test.txt"
    destFile := "destination.txt"
    total, err := copyFile1(srcFile, destFile) //函数的具体实现在下面
    fmt.Println(total, err)
}

func copyFile1(srcFile, destFile string) (int, error) { //返回值是完成拷贝数据的字节数&err
    file1, err := os.Open(srcFile) //这里读 所以用open就足够了
    if err != nil {
        return 0, err
    }
    file2, err := os.OpenFile(destFile, os.O_WRONLY|os.O_CREATE, os.ModePerm) //这里要写文件 所以需要用到OpenFile
    if err != nil {
        return 0, err
    }
    // 关闭文件
    defer file1.Close()
    defer file2.Close()
    //读写文件
    bs := make([]byte, 1024, 1024)//用make创造切片
    n := -1    //每次读取的字节数
    total := 0 //读取的总数量
    for {
        n, err = file1.Read(bs)
        if err == io.EOF || n == 0 {
            fmt.Println("copy is complete")
            break
        } else if err != nil {
            fmt.Println("error happen")
            return total, err
        }
        total += n
        file2.Write(bs[:n])
    }
    return total, nil
}

当然了,这段代码不仅可以复制txt文件,其他类型,比如png什么的都不在话下,毕竟是通过字节传递的。

以上就是手动实现copy,当然这太麻烦了,实际工作中很少这么用,之所以放在这里,是为了更深层的理解他的实现原理,万变不离其宗,这就是宗。

接下来介绍一下实际常用的

库函数实现

io.Copy()方法实现

func Copy(dst Writer, src Reader) (written int64, err error)  {}

这个函数需要两个参数,第一个是向外写的目的对象,第二个是读进来的对象;返回值有两个,一个是复制的总数据量,一个是错误。需要注意的是,如果成功拷贝,返回值是nil 不是EOF。

具体怎么使用呢?
还是复制上面例子中的文件,代码如下

package main

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

func main() {
    // 拷贝文件
    srcFile := "/Users/liberhome/GolandProjects/awesomeProject/I-package/April28_Go_io/test.txt"
    destFile := "destination2.txt"
    //total, err := copyFile1(srcFile, destFile) //函数的具体实现在下面
    total, err := copyFile2(srcFile, destFile) //函数的具体实现在下面
    fmt.Println(total, err)
}

func copyFile1(srcFile, destFile string) (int, error) { //返回值是完成拷贝数据的字节数&err
    file1, err := os.Open(srcFile) //这里读 所以用open就足够了
    if err != nil {
        return 0, err
    }
    file2, err := os.OpenFile(destFile, os.O_WRONLY|os.O_CREATE, os.ModePerm) //这里要写文件 所以需要用到OpenFile
    if err != nil {
        return 0, err
    }
    // 关闭文件
    defer file1.Close()
    defer file2.Close()
    //读写文件
    bs := make([]byte, 1024, 1024)
    n := -1    //每次读取的字节数
    total := 0 //读取的总数量
    for {
        n, err = file1.Read(bs)
        if err == io.EOF || n == 0 {
            fmt.Println("copy is complete")
            break
        } else if err != nil {
            fmt.Println("error happen")
            return total, err
        }
        total += n
        file2.Write(bs[:n])
    }
    return total, nil
}

func copyFile2(srcFile, destFile string) (int64, error) {
    file1, err := os.Open(srcFile) //这里读 所以用open就足够了
    if err != nil {
        return 0, err
    }
    file2, err := os.OpenFile(destFile, os.O_WRONLY|os.O_CREATE, os.ModePerm) //这里要写文件 所以需要用到OpenFile
    if err != nil {
        return 0, err
    }
    // 关闭文件
    defer file1.Close()
    defer file2.Close()
    //就是在这里直接调用库函数~
    return io.Copy(file2, file1)
}

除Copy之外,io包还提供了CopyN(dst, src, n)【复制src中n个字节到dst】以及copyBuffer(dst, src, buf)【指定一个buf区域,以后都已这个buf大小完全复制】,
他们之间的底层实现逻辑如下:
Golang 文件的复制 手动实现 与 库函数实现_第1张图片
所以,本质上都是基于copyBuffer,当不传buf的时候默认缓冲区buf是32*1024。

ioutil包的方法实现

但是由于ioutil包中的writeFile和readFile方法都是一次性读取再一次性写入,对大文件并不友好,这里不再展开。

参考资料来源:bilibili

你可能感兴趣的:(golang复制文件)