file-util.go: new functions CopyFileWithParent(), CopyFile(), CompareFilesBMd5(), CompareLastChangeTime(), and LoadFile()
This commit is contained in:
parent
da3dc390c3
commit
97a7408481
104
file-util.go
104
file-util.go
@ -5,10 +5,13 @@
|
|||||||
package utils
|
package utils
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bytes"
|
||||||
|
"crypto/md5"
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"os"
|
"os"
|
||||||
|
"path"
|
||||||
"strconv"
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
)
|
)
|
||||||
@ -140,3 +143,104 @@ func ParseMemSize(s string) (size int64, err error) {
|
|||||||
}
|
}
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func CopyFileWithParent(src, dst string, perm os.FileMode) (int64, error) {
|
||||||
|
parent := path.Dir(dst)
|
||||||
|
if err := os.MkdirAll(parent, 0775); err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
return CopyFile(src, dst, perm)
|
||||||
|
}
|
||||||
|
|
||||||
|
func CopyFile(src, dst string, perm os.FileMode) (int64, error) {
|
||||||
|
sourceFileStat, err := os.Stat(src)
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
|
||||||
|
if !sourceFileStat.Mode().IsRegular() {
|
||||||
|
return 0, fmt.Errorf("%s is not a regular file", src)
|
||||||
|
}
|
||||||
|
|
||||||
|
source, err := os.Open(src)
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
defer source.Close()
|
||||||
|
|
||||||
|
destination, err := os.Create(dst)
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
defer destination.Close()
|
||||||
|
|
||||||
|
if perm != 0 {
|
||||||
|
os.Chmod(dst, perm)
|
||||||
|
}
|
||||||
|
nBytes, err := io.Copy(destination, source)
|
||||||
|
return nBytes, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func CompareFilesByMd5(filePath1, filePath2 string) (same bool, err error) {
|
||||||
|
var fh1, fh2 *os.File
|
||||||
|
|
||||||
|
if fh1, err = os.Open(filePath1); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer fh1.Close()
|
||||||
|
|
||||||
|
if fh2, err = os.Open(filePath2); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer fh2.Close()
|
||||||
|
|
||||||
|
h1 := md5.New()
|
||||||
|
if _, err = io.Copy(h1, fh1); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
h2 := md5.New()
|
||||||
|
if _, err = io.Copy(h2, fh2); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
s1 := h1.Sum(nil)
|
||||||
|
s2 := h2.Sum(nil)
|
||||||
|
same = bytes.Equal(s1, s2)
|
||||||
|
// same = bytes.Equal(h1.Sum(nil), h2.Sum(nil))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// <0 = first newer than second
|
||||||
|
// =0 = same date
|
||||||
|
// >1 = first older than second
|
||||||
|
func CompareLastChangeTime(first, second string) (result int, err error) {
|
||||||
|
var info os.FileInfo
|
||||||
|
|
||||||
|
if info, err = os.Stat(first); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if info, err = os.Stat(second); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
tFirst := info.ModTime()
|
||||||
|
tSecond := info.ModTime()
|
||||||
|
if tFirst.After(tSecond) {
|
||||||
|
result = -1
|
||||||
|
} else if tFirst.Before(tSecond) {
|
||||||
|
result = 1
|
||||||
|
} else {
|
||||||
|
result = 0
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func LoadFile(filePath string) (data []byte, err error) {
|
||||||
|
data, err = os.ReadFile(filePath)
|
||||||
|
if err != nil {
|
||||||
|
err = fmt.Errorf("error reading file %q: %w", filePath, err)
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user