golang 复制文件文件夹

野性酷女 2021-09-02 03:12 771阅读 0赞

在golang 中，复制文件是比较容易的，直接操作"io/ioutil"包就可以了，但是复制文件夹就有点麻烦了，需要一层层检查是否存在这个文件夹，不存在就创建文件夹，

实例如下：

package copy
    
    import (
        "fmt"
        "io"
        "io/ioutil"
        "os"
        "path"
    )
    
    // File copies a single file from src to dst
    func File(src, dst string) error {
        var err error
        var srcfd *os.File
        var dstfd *os.File
        var srcinfo os.FileInfo
    
        if srcfd, err = os.Open(src); err != nil {
            return err
        }
        defer srcfd.Close()
    
        if dstfd, err = os.Create(dst); err != nil {
            return err
        }
        defer dstfd.Close()
    
        if _, err = io.Copy(dstfd, srcfd); err != nil {
            return err
        }
        if srcinfo, err = os.Stat(src); err != nil {
            return err
        }
        return os.Chmod(dst, srcinfo.Mode())
    }
    
    
    // Dir copies a whole directory recursively
    func Dir(src string, dst string) error {
        var err error
        var fds []os.FileInfo
        var srcinfo os.FileInfo
    
        if srcinfo, err = os.Stat(src); err != nil {
            return err
        }
    
        if err = os.MkdirAll(dst, srcinfo.Mode()); err != nil {
            return err
        }
    
        if fds, err = ioutil.ReadDir(src); err != nil {
            return err
        }
        for _, fd := range fds {
            srcfp := path.Join(src, fd.Name())
            dstfp := path.Join(dst, fd.Name())
    
            if fd.IsDir() {
                if err = Dir(srcfp, dstfp); err != nil {
                    fmt.Println(err)
                }
            } else {
                if err = File(srcfp, dstfp); err != nil {
                    fmt.Println(err)
                }
            }
        }
        return nil
    }