You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

59 lines
2.1 KiB

package upload_file
import (
"errors"
"fmt"
"github.com/gin-gonic/gin"
"goskeleton/app/global/my_errors"
"goskeleton/app/global/variable"
"goskeleton/app/utils/md5_encrypt"
"os"
"path"
"strings"
"time"
)
func Upload(context *gin.Context, savePath string) (r bool, finnalSavePath interface{}) {
newSavePath, newReturnPath := generateYearMonthPath(savePath)
// 1.获取上传的文件名(参数验证器已经验证完成了第一步错误,这里简化)
file, _ := context.FormFile(variable.ConfigYml.GetString("FileUploadSetting.UploadFileField")) // file 是一个文件结构体(文件对象)
// 保存文件原始文件名进行全局唯一编码加密、md5 加密,保证在后台存储不重复
var saveErr error
if sequence := variable.SnowFlake.GetId(); sequence > 0 {
saveFileName := fmt.Sprintf("%d%s", sequence, file.Filename)
saveFileName = md5_encrypt.MD5(saveFileName) + path.Ext(saveFileName)
if saveErr = context.SaveUploadedFile(file, newSavePath+saveFileName); saveErr == nil {
// 上传成功,返回资源的相对路径,这里请根据实际返回绝对路径或者相对路径
finnalSavePath = gin.H{
"path": strings.ReplaceAll(newReturnPath+saveFileName, variable.BasePath, ""),
}
return true, finnalSavePath
}
} else {
saveErr = errors.New(my_errors.ErrorsSnowflakeGetIdFail)
variable.ZapLog.Error("文件保存出错:" + saveErr.Error())
}
return false, nil
}
// 文件上传可以设置按照 xxx年-xx月 格式存储
func generateYearMonthPath(savePathPre string) (string, string) {
returnPath := variable.BasePath + variable.ConfigYml.GetString("FileUploadSetting.UploadFileReturnPath")
curYearMonth := time.Now().Format("2006_01")
newSavePathPre := savePathPre + curYearMonth
newReturnPathPre := returnPath + curYearMonth
// 相关路径不存在,创建目录
if _, err := os.Stat(newSavePathPre); err != nil {
if err = os.MkdirAll(newSavePathPre, os.ModePerm); err != nil {
variable.ZapLog.Error("文件上传创建目录出错" + err.Error())
return "", ""
}
}
return newSavePathPre + "/", newReturnPathPre + "/"
}