go语言实现文件复制、移动（cp、mv命令）

目录

复制文件

方式1：通过内置包文件操作实现

方式2：通过exec包实现

移动文件

方式1：通过内置包文件操作实现

方式2：通过exec包实现

提示

复制文件

方式1：通过内置包文件操作实现

func CopyFile(src, dst string) (size int64, err error) {
?? ?fmt.Printf("ready copy '%s' to '%s'\n", src, dst)

?? ?srcFile, err := os.OpenFile(src, os.O_RDONLY, 0644)
?? ?if err != nil {
?? ??? ?err = fmt.Errorf("failed to open source file: %v", err)
?? ??? ?return
?? ?}
?? ?defer srcFile.Close()

?? ?info, err := srcFile.Stat()
?? ?if err != nil {
?? ??? ?err = fmt.Errorf("failed to get file info: %v\n", err)
?? ??? ?return
?? ?}

?? ?size = info.Size()

?? ?if info.IsDir() {
?? ??? ?err = fmt.Errorf("source is a directory, does not support copying directories recursively.")
?? ??? ?return
?? ?}

?? ?dstFile, err := createFile(dst, info)
?? ?if err != nil {
?? ??? ?err = errors.New(fmt.Sprintf("failed to create destination file: %v", err))
?? ??? ?return
?? ?}
?? ?defer dstFile.Close()

?? ?_, err = io.Copy(dstFile, srcFile)
?? ?if err != nil {
?? ??? ?err = fmt.Errorf(fmt.Sprintf("failed to copy file: %v\n", err))
?? ??? ?return
?? ?}

?? ?fmt.Printf("successfully copied '%s' to '%s'\n", src, dst)
?? ?return
}

func createFile(dst string, info os.FileInfo) (*os.File, error) {
?? ?dir := filepath.Dir(dst)
?? ?
?? ?err := os.MkdirAll(dir, os.ModePerm)
?? ?if err != nil {
?? ??? ?return nil, fmt.Errorf("failed to create destination directory: %w", err)
?? ?}

?? ?dstFile, err := os.Create(dst)
?? ?if err != nil {
?? ??? ?return nil, fmt.Errorf("failed to create destination file: %w", err)
?? ?}

?? ?err = dstFile.Chmod(info.Mode())
?? ?if err != nil {
?? ??? ?dstFile.Close()
?? ??? ?return nil, fmt.Errorf("failed to set destination file permissions: %w", err)
?? ?}

?? ?return dstFile, nil
}

方式2：通过exec包实现

封装执行简单命令的函数：

func Exec(name string, args ...string) (outString string, err error) {
?? ?var out bytes.Buffer
?? ?var stderr bytes.Buffer

?? ?cmd := exec.Command(name, args...)
?? ?cmd.Stdout = &out
?? ?cmd.Stderr = &stderr

?? ?err = cmd.Run()
?? ?outString = out.String()
?? ?if err != nil {
?? ??? ?err = fmt.Errorf("exec failed: %v, stderr=%s. name=%s, args=%v.", err, stderr.String(), name, args)
?? ?}

?? ?return
}

使用：

_, err = Exec("cp", src, dst)

即可完成文件复制。

移动文件

方式1：通过内置包文件操作实现

func Move(src, dst string) error {
?? ?return os.Rename(src, dst)
}

方式2：通过exec包实现

_, err = Exec("mv", src, dst)

提示

目标目录不存在时，通过exec.Command的方式执行cp、mv会报错，因此需要提前准备好目的目录。
?