当前位置：首页 > news >正文

go make function_Go中复制文件的3种技巧

news 来源：原创 2024/5/8 14:35:31

本文将向你展示如何使用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.goPlease provide two command line arguments!$ go run cp1.go fileCP.txt /tmp/fileCPCOPYCopied 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.goPlease 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将生成下一种输出：