95 lines
1.8 KiB
Go
95 lines
1.8 KiB
Go
|
package utils
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"crypto/md5"
|
|||
|
|
"encoding/base64"
|
|||
|
|
"encoding/hex"
|
|||
|
|
"fmt"
|
|||
|
|
"git.hpds.cc/Component/logging"
|
|||
|
|
"go.uber.org/zap"
|
|||
|
|
"io"
|
|||
|
|
"os"
|
|||
|
|
"path"
|
|||
|
|
"path/filepath"
|
|||
|
|
"strings"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func CopyFile(src, dst string) error {
|
|||
|
|
sourceFileStat, err := os.Stat(src)
|
|||
|
|
if err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if !sourceFileStat.Mode().IsRegular() {
|
|||
|
|
return fmt.Errorf("%s is not a regular file", src)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
source, err := os.Open(src)
|
|||
|
|
if err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
defer func(source *os.File) {
|
|||
|
|
_ = source.Close()
|
|||
|
|
}(source)
|
|||
|
|
|
|||
|
|
destination, err := os.Create(dst)
|
|||
|
|
if err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
defer func(destination *os.File) {
|
|||
|
|
_ = destination.Close()
|
|||
|
|
}(destination)
|
|||
|
|
_, err = io.Copy(destination, source)
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func PathExists(path string) bool {
|
|||
|
|
_, err := os.Stat(path)
|
|||
|
|
if err == nil {
|
|||
|
|
return true
|
|||
|
|
}
|
|||
|
|
if os.IsNotExist(err) {
|
|||
|
|
return false
|
|||
|
|
}
|
|||
|
|
return false
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ReadFile 读取到file中，再利用ioutil将file直接读取到[]byte中, 这是最优
|
|||
|
|
func ReadFile(fn string) []byte {
|
|||
|
|
f, err := os.Open(fn)
|
|||
|
|
if err != nil {
|
|||
|
|
logging.L().Error("Read File", zap.String("File Name", fn), zap.Error(err))
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
defer func(f *os.File) {
|
|||
|
|
_ = f.Close()
|
|||
|
|
}(f)
|
|||
|
|
|
|||
|
|
fd, err := io.ReadAll(f)
|
|||
|
|
if err != nil {
|
|||
|
|
logging.L().Error("Read File To buff", zap.String("File Name", fn), zap.Error(err))
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return fd
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func GetFileName(fn string) string {
|
|||
|
|
fileType := path.Ext(fn)
|
|||
|
|
return strings.TrimSuffix(fn, fileType)
|
|||
|
|
}
|
|||
|
|
func GetFileNameAndExt(fn string) string {
|
|||
|
|
_, fileName := filepath.Split(fn)
|
|||
|
|
return fileName
|
|||
|
|
}
|
|||
|
|
func GetFileMd5(data []byte) string {
|
|||
|
|
hash := md5.New()
|
|||
|
|
hash.Write(data)
|
|||
|
|
return hex.EncodeToString(hash.Sum(nil))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func FileToBase64(fn string) string {
|
|||
|
|
buff := ReadFile(fn)
|
|||
|
|
return base64.StdEncoding.EncodeToString(buff) // 加密成base64字符串
|
|||
|
|
}
|