怎样使用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:\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