怎样使用go语言将文件复制成另一文件?go语言标准包io里提供一个简单的函数copy实现了这个功能,下面是一个例子。

package main

import (
    "fmt"
    "io"
    "os"
)
func main() {
    CopyFile(os.Args[1], os.Args[2]) // os.Args[1]为目标文件,os.Args[2]为源文件
    fmt.Println("复制完成",)
}
func CopyFile(dstName, srcName string) (written int64, err error) {
    src, err := os.Open(srcName)
    if err != nil {
        return
    }
    defer src.Close()
    dst, err := os.OpenFile(dstName, os.O_WRONLY|os.O_CREATE, 0644)
    if err != nil {
        return
    }
    defer dst.Close()
    return io.Copy(dst, src)
}
e:\go_source\Go_program_code\temptest>temptest.exe abc.log ex14061310.log
e:\gosource\Goprogram_code\temptest>ls -al
total 3947
drwxr-xr-x 2 sharp Administ 0 Oct 24 10:23 .
drwxr-xr-x 25 sharp Administ 8192 Oct 20 17:20 ..
-rw-r--r-- 1 sharp Administ 2917891 Oct 24 11:05 abc.log
-rw-r--r-- 1 sharp Administ 2917891 Jun 13 10:00 ex14061310.log
-rwxr-xr-x 1 sharp Administ 2236928 Oct 24 11:05 temptest.exe
-rw-r--r-- 1 sharp Administ 520 Oct 24 11:05 temptest.go

注意这里defer的使用,在源文件和目的文件打开后都跟了一个defer延迟关闭文件。如果目标文件后面没有使用defer dst.Close(),那么一旦创建目标文件调试失败,它将直接返回错误,那么,会导致源文件一直保持打开状态,这样资源就得不到释放。因此,在go语言里,记得open一个文件做了错误判断后要紧跟一个defer close延迟调用。

io.Copy的函数原型如下
func Copy(dst Writer, src Reader) (written int64, err error)
Copy函数从源到目的的复制,直到读到源的EOF或者出现其它的错误,它返回所复制的字节数及复制过程出现的第一个错误。如果成功复制,Cop返回的是 err ==nil,因为Copy被定义为从源复制直到遇到EOF,它不把EOF当作一个错误。
io包里还有一个io.CopyN的函数,原型如下
func CopyN(dst Writer, src Reader, n int64) (written int64, err error)
CopyN将从源文件里复制n字节(或者遇到错误中断)到目标文件,并返到实际复制的节数
package main

import (
    "fmt"
    "io"
    "os"
)
func main() {
    counter,  := CopyFile(os.Args[1], os.Args[2], os.Args[3]) // os.Args[1]为目标文件,os.Args[2]为源文件
     fmt.Println("复制", counter, "字节!")
}
func CopyFile(dstName, srcName string, n int64) (written int64, err error) {
    src, err := os.Open(srcName)
    if err != nil {
        return
    }
    defer src.Close()
    dst, err := os.OpenFile(dstName, os.OWRONLY|os.OCREATE, 0644)
     if err != nil {
        return
    }
    defer dst.Close()
    return io.CopyN(dst, src, n) //
}
e:\go_source\Go_program_code\temptest>temptest.exe test.go temptest.go 1000
复制 526 字节!

e:\go_source\Go_program_code\temptest>ls -al
total 3948
drwxr-xr-x 2 sharp Administ 0 Oct 24 11:10 .
drwxr-xr-x 25 sharp Administ 8192 Oct 20 17:20 ..
-rw-r--r-- 1 sharp Administ 2917891 Oct 24 11:05 abc.log
-rw-r--r-- 1 sharp Administ 2917891 Jun 13 10:00 ex14061310.log
-rwxr-xr-x 1 sharp Administ 2236928 Oct 24 11:05 temptest.exe
-rw-r--r-- 1 sharp Administ 526 Oct 24 11:10 temptest.go
-rw-r--r-- 1 sharp Administ 526 Oct 24 11:10 test.go

有疑问加站长微信联系(非本文作者)