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

Golang实现文件复制

news 2025/6/27 4:44:20

方法：三种

package zdpgo_fileimport ("errors""io""os"
)// CopyFile 使用io.Copy进行文件的复制，同时也会复制文件的所有权限
// @param src 复制文件
// @param des 目标文件
// @return error 错误信息
func CopyFile(src, des string) error {// 判断文件是否存在if !IsExist(src) {return errors.New("要复制的文件不存在")}// 打开文件srcFile, err := os.Open(src)if err != nil {return err}defer srcFile.Close()// 获取源文件的权限fi, _ := srcFile.Stat()perm := fi.Mode()// 创建目标文件，并复制源文件的所有权限desFile, err := os.OpenFile(des, os.O_RDWR|os.O_CREATE|os.O_TRUNC, perm)if err != nil {return err}defer desFile.Close()// 进行复制_, err = io.Copy(desFile, srcFile)// 返回return err
}// CopyFileByWriteFile 使用ioutil.WriteFile()和ioutil.ReadFile()
// @param src 复制文件
// @param des 目标文件
// @return error 错误信息
func CopyFileByWriteFile(src, des string) error {if !IsExist(src) {return errors.New("要复制的文件不存在")}// 获取源文件的权限srcFile, err := os.Open(src)if err != nil {return err}fi, _ := srcFile.Stat()perm := fi.Mode()defer srcFile.Close()// 读取文件input, err := os.ReadFile(src)if err != nil {return err}// 将源文件的内容和权限写入创建的文件err = os.WriteFile(des, input, perm)if err != nil {return err}// 返回return nil
}// CopyFileByWriteBuffer 使用os.Read()和os.Write()
// @param src 复制文件
// @param des 目标文件
// @param bufSize 缓冲区大小，小于或者等于0的时候，会使用默认1M的缓冲区大小
// @return error 错误信息
func CopyFileByWriteBuffer(src, des string, bufSize int) error {// 设置缓冲区大小if bufSize <= 0 {bufSize = 1 * 1024 * 1024 //1M}// 构造缓冲区buf := make([]byte, bufSize)// 打开文件srcFile, err := os.Open(src)if err != nil {return err}defer srcFile.Close()// 获取源文件的权限fi, _ := srcFile.Stat()perm := fi.Mode()// 创建目标文件并写入复制的权限desFile, err := os.OpenFile(des, os.O_CREATE|os.O_RDWR|os.O_TRUNC, perm)if err != nil {return err}defer desFile.Close()// 进行缓冲写入var n intfor {n, err = srcFile.Read(buf)if err != nil && err != io.EOF {return err}if n == 0 {break}if _, err = desFile.Write(buf[:n]); err != nil {return err}}// 返回return nil
}

测试

package zdpgo_fileimport "testing"func TestCopyFile(t *testing.T) {if err := AddDir("test"); err != nil {t.Fatal(err)}data1 := []struct {Src stringDst string}{{"copy_file.go", "test/test1.go"},{"copy_file.go", "test/test2.go"},{"copy_file.go", "test/test3.go"},}for _, tt := range data1 {v := CopyFile(tt.Src, tt.Dst)if v != nil {t.Error(v)}}data2 := []struct {Src stringDst string}{{"copy_file.go", "test/test11.go"},{"copy_file.go", "test/test22.go"},{"copy_file.go", "test/test33.go"},}for _, tt := range data2 {v := CopyFileByWriteFile(tt.Src, tt.Dst)if v != nil {t.Error(v)}}data3 := []struct {Src stringDst string}{{"copy_file.go", "test/test111.go"},{"copy_file.go", "test/test222.go"},{"copy_file.go", "test/test333.go"},}for _, tt := range data3 {v := CopyFileByWriteBuffer(tt.Src, tt.Dst, 0)if v != nil {t.Error(v)}}
}