您的位置 首页 golang

Go中复制文件的3种技巧

本文将向你展示如何使用 go 编程语言复制文件。尽管在Go中复制文件有三种以上的方法,但本文将介绍三种最常见的方法:使用io.Copy()Go库中的函数调用;一次读取所有输入文件并将其写入另一个文件;并使用缓冲区将文件分小块复制。

方法1:使用io.Copy()

该实用程序的第一个版本将使用io.Copy()标准Go库的功能。该实用程序的逻辑可以在该copy()函数的实现中找到,如下所示:

 func copy(src, dst string) (int64, error) {
        sourceFileStat, err := os.Stat(src)
        if err != nil {
                return 0, err
        }
​
        if !sourceFileStat.Mode().IsRegular() {
                return 0, fmt.Errorf("%s is not a regular file", src)
        }
​
        source, err := os.Open(src)
        if err != nil {
                return 0, err
        }
        defer source. Close ()
​
        destination, err := os.Create(dst)
        if err != nil {
                return 0, err
        }
        defer destination.Close()
        nBytes, err := io.Copy(destination, source)
        return nBytes, err
}  

除了测试将要复制的文件是否存在(os.Stat(src))和是否为常规文件(sourceFileStat.Mode().IsRegular())以便可以打开文件进行读取之外,所有工作均由该io.Copy(destination, source)语句完成。该io.Copy()函数返回复制的字节数和在复制过程中发生的第一条错误消息。在Go中,如果没有错误消息,则错误变量的值为nil。

你可以io.Copy()在io软件包文档页面上了解有关该功能的更多信息。

执行cp1.go将生成下一种输出:

 $ go run cp1.go
Please provide two command line arguments!
$ go run cp1.go fileCP.txt /tmp/fileCPCOPY
Copied 3826 bytes!
$ diff fileCP.txt /tmp/fileCPCOPY  

该技术尽可能简单,但是给开发人员没有灵活性,这并不总是一件坏事。但是,有时开发人员需要或想要决定他们如何读取文件。

方法2:使用ioutil.WriteFile()和ioutil.ReadFile()

复制文件的第二种方法是使用ioutil.ReadFile()和ioutil.WriteFile()函数。第一个函数将整个文件的内容读取到一个字节片中,第二个函数将一个字节片的内容写入一个文件中。

该实用程序的逻辑可以在以下Go代码中找到:

  input , err := ioutil.ReadFile(sourceFile)
        if err != nil {
                fmt.Println(err)
                return
        }
​
        err = ioutil.WriteFile(destinationFile, input, 0644)
        if err != nil {
                fmt.Println("Error creating", destinationFile)
                fmt.Println(err)
                return
        }  

除了这两个if块(它们是Go的工作方式的一部分)之外,你还可以看到在ioutil.ReadFile()and ioutil.WriteFile()语句中可以找到该程序的功能。

执行 cp2 .go将生成下一种输出:

 $ go run cp2.go
Please provide two command line arguments!
$ go run cp2.go fileCP.txt /tmp/copyFileCP
$ diff fileCP.txt /tmp/copyFileCP  

请注意,尽管此技术将复制文件,但是当你要复制大文件时,它可能效率不高,因为返回的字节片ioutil.ReadFile()也将非常大。

方法3:使用os.Read()和os.Write()

在Go中复制文件的第三种方法是使用cp3.go 将在本节中开发的实用程序。它接受三个参数:输入文件的文件名,输出文件的文件名和缓冲区的大小。

的最重要部分cp3.go位于以下for循环中,可以在copy() function:

  buf := make([]byte, BUFFERSIZE)
        for {
                n, err := source.Read(buf)
                if err != nil && err != io.EOF {
                        return err
                }
                if n == 0 {
                        break
                }
​
                if _, err := destination.Write(buf[:n]); err != nil {
                        return err
                }
        }  

这种技术使用os.Read()用于读取输入文件的一小部分到名为缓冲器buf和os.Write()用于写入该缓冲器中的内容到一个文件。当读取错误或到达文件末尾(io.EOF)时,复制过程停止。

执行cp3.go将生成下一种输出:

如你所见,缓冲区的大小会极大地影响的性能cp3.go。

做一些基准测试

本文的最后一部分将尝试cp3.go 使用time(1)命令行实用程序比较这三个程序以及各种缓冲区大小的性能。

下面的输出显示的性能cp1.go,cp2.go以及cp3.go复制一个500MB的文件时:

输出显示所有三个实用程序的性能都非常相似,这意味着标准Go库的功能非常聪明且经过优化。

现在,让我们测试缓冲区大小如何影响的性能cp3.go。cp3.go以10、20和1,000字节的缓冲区大小执行以在相当快的计算机上复制500MB文件将产生以下结果:

生成的输出表明,缓冲区越大,cp3.go实用程序的性能越快,这或多或少是期望的。此外,使用小于20字节的缓冲区大小复制大文件是一个非常缓慢的过程,应避免使用。

文章来源:智云一二三科技

文章标题:Go中复制文件的3种技巧

文章地址:https://www.zhihuclub.com/100771.shtml

关于作者: 智云科技

热门文章

网站地图