首页 > 代码库 > 使用go语言复制文件

使用go语言复制文件

怎样使用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
复制  2917891  字节!

e:\go_source\Go_program_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.O_WRONLY|os.O_CREATE, 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


使用go语言复制文件