go语言学习笔记----模拟实现文件拷贝函数

实例1

//main
package main

import (
	"bufio"
	"flag"
	"fmt"
	"io"
	"os"
	"strings"
)

func fileExists(filename string) bool {
	_, err := os.Stat(filename)
	return err == nil || os.IsExist(err)
}

func copyFile(src, dst string) (w int64, err error) {
	srcFile, err := os.Open(src)
	if err != nil {
		fmt.Println(err.Error())
		return
	}

	defer srcFile.Close()
	dstFile, err := os.Create(dst)

	if err != nil {
		fmt.Println(err.Error())
		return
	}

	defer dstFile.Close()

	return io.Copy(dstFile, srcFile)

}
func copyFileAction(src, dst string, showProgress, force bool) {
	if !force {
		if fileExists(dst) {
			fmt.Printf("%s exists override? y/n\n", dst)
			reader := bufio.NewReader(os.Stdin)
			data, _, _ := reader.ReadLine()

			if strings.TrimSpace(string(data)) != "y" {
				return
			}

		}

	}
	copyFile(src, dst)
	if showProgress {
		fmt.Printf("'%s'->'%s'\n", src, dst)
	}
}

func main() {
	var showProgress, force bool
　　　　//定义命令行参数
	flag.BoolVar(&force, "f", false, "force copy when existing")
	flag.BoolVar(&showProgress, "v", false, "explain what is being done")

	flag.Parse()
　　　　//非法命令行数量检测
	if flag.NArg() < 2 {
		flag.Usage()
		return
	}
       //取第0个参数与第一个参数作为源文件名跟目标文件名
	copyFileAction(flag.Arg(0), flag.Arg(1), showProgress, force)
}

 

 

示例2：Golang实现文件拷贝功能

package main

import (
    "bufio"
    "fmt"
    "io"
    "os"
)

// 编写一个函数，接收两个文件路径:
func copyFile(dstFileName string, srcFileName string) (written int64, err error) {
    srcFile, err := os.Open(srcFileName)
    if err != nil {
        fmt.Printf("open file error = %v\n", err)
    }
    defer srcFile.Close()

    //通过srcFile，获取到READER
    reader := bufio.NewReader(srcFile)

    //打开dstFileName
    dstFile, err := os.OpenFile(dstFileName, os.O_WRONLY|os.O_CREATE, 0666)
    if err != nil {
        fmt.Printf("open file error = %v\n", err)
        return
    }

    //通过dstFile，获取到WRITER
    writer := bufio.NewWriter(dstFile)
    //writer.Flush()

    defer dstFile.Close()

    return io.Copy(writer, reader)
}


func main() {
    //调用copyFile完成文件拷贝
    srcFile := "e:/copyFileTest02.pdf"
    dstFile := "e:/Go/tools/copyFileTest02.pdf"

    _, err := copyFile(dstFile, srcFile)
    if err == nil {
        fmt.Println("拷贝文件正常...")
    } else {
        fmt.Printf("拷贝文件出错了... err =%v\n", err)
    }
}

 说明：

1, 使用了带缓存的 bufio.NewReader(srcFile) 和 bufio.NewWriter(dstFile)，一边读一边写，这样支持拷贝较大的文件；
2，defer dstFile.Close() 切记使用defer 来关闭打开的文件，避免内存泄漏；