120 lines
2.8 KiB
Go
120 lines
2.8 KiB
Go
package utils
|
||
|
||
import (
|
||
"crypto/md5"
|
||
"encoding/hex"
|
||
"fmt"
|
||
"git.hpds.cc/Component/logging"
|
||
"go.uber.org/zap"
|
||
"hpds-iot-web/pkg/minio"
|
||
"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 DownloadMinioFileToLocalPath(accessUrl, dstPath, fileName, protocol, endpoint, bucket, accessKeyId, secretAccessKey string,
|
||
logger *logging.Logger) {
|
||
|
||
if !PathExists(path.Join(dstPath, fileName)) {
|
||
dPath := strings.Replace(accessUrl, fmt.Sprintf("%s://%s/", protocol, endpoint), "", 1)
|
||
|
||
dPath = strings.Replace(dPath, bucket, "", 1)
|
||
minioCli := minio.NewClient(accessKeyId, secretAccessKey, endpoint, false, logger)
|
||
|
||
imgByte, err := minioCli.GetObject(dPath, bucket)
|
||
if err != nil {
|
||
logger.With(zap.String("源文件名", accessUrl)).
|
||
With(zap.String("文件名", path.Join(dstPath, fileName))).
|
||
Error("文件下载", zap.Error(err))
|
||
}
|
||
err = os.MkdirAll(dstPath, os.ModePerm)
|
||
if err != nil {
|
||
logger.With(zap.String("源文件名", accessUrl)).
|
||
With(zap.String("文件名", path.Join(dstPath, fileName))).
|
||
Error("创建文件下载目录", zap.Error(err))
|
||
}
|
||
err = os.WriteFile(path.Join(dstPath, fileName), imgByte, os.ModePerm)
|
||
if err != nil {
|
||
logger.With(zap.String("源文件名", accessUrl)).
|
||
With(zap.String("文件名", path.Join(dstPath, fileName))).
|
||
Error("文件写入", zap.Error(err))
|
||
}
|
||
}
|
||
}
|