2019-01-13 12:45:25 +08:00
|
|
|
package common
|
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
|
|
|
"io"
|
|
|
|
"os"
|
|
|
|
)
|
|
|
|
|
|
|
|
// CopyFile copy file
|
|
|
|
func CopyFile(source string, dest string) (err error) {
|
|
|
|
sourcefile, err := os.Open(source)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
defer sourcefile.Close()
|
|
|
|
|
|
|
|
destfile, err := os.Create(dest)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
defer destfile.Close()
|
|
|
|
|
|
|
|
_, err = io.Copy(destfile, sourcefile)
|
|
|
|
if err == nil {
|
|
|
|
sourceinfo, err := os.Stat(source)
|
|
|
|
if err != nil {
|
|
|
|
_ = os.Chmod(dest, sourceinfo.Mode())
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
// CopyDir recursive copy of directory
|
|
|
|
func CopyDir(source string, dest string) (err error) {
|
|
|
|
|
|
|
|
// get properties of source dir
|
|
|
|
sourceinfo, err := os.Stat(source)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
// create dest dir
|
|
|
|
|
|
|
|
err = os.MkdirAll(dest, sourceinfo.Mode())
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
directory, _ := os.Open(source)
|
|
|
|
|
|
|
|
objects, err := directory.Readdir(-1)
|
|
|
|
|
|
|
|
for _, obj := range objects {
|
|
|
|
|
|
|
|
sourcefilepointer := source + "/" + obj.Name()
|
|
|
|
|
|
|
|
destinationfilepointer := dest + "/" + obj.Name()
|
|
|
|
|
|
|
|
if obj.IsDir() {
|
|
|
|
// create sub-directories - recursively
|
|
|
|
err = CopyDir(sourcefilepointer, destinationfilepointer)
|
|
|
|
if err != nil {
|
|
|
|
fmt.Println(err)
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
// perform copy
|
|
|
|
err = CopyFile(sourcefilepointer, destinationfilepointer)
|
|
|
|
if err != nil {
|
|
|
|
fmt.Println(err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
2019-01-16 09:41:02 +08:00
|
|
|
return err
|
2019-01-13 12:45:25 +08:00
|
|
|
}
|