优化上传与下载文件示例的上传与删除的api层与service层代码

server/service/exa_file_upload_download.go

@@ -1,9 +1,14 @@
 package service
 
 import (
+	"errors"
 	"gin-vue-admin/global"
 	"gin-vue-admin/model"
 	"gin-vue-admin/model/request"
+	"gin-vue-admin/utils"
+	"mime/multipart"
+	"os"
+	"strings"
 )
 
 // @title    Upload
@@ -35,8 +40,24 @@ func FindFile(id uint) (error, model.ExaFileUploadAndDownload) {
 // @param     file            model.ExaFileUploadAndDownload
 // @return                    error
 
-func DeleteFile(file model.ExaFileUploadAndDownload) error {
-	err := global.GVA_DB.Where("id = ?", file.ID).Unscoped().Delete(file).Error
+func DeleteFile(file model.ExaFileUploadAndDownload) (err error) {
+	var fileFromDb model.ExaFileUploadAndDownload
+	err, fileFromDb = FindFile(file.ID)
+	if err != nil {
+		return errors.New("文件不存在")
+	}
+	if global.GVA_CONFIG.LocalUpload.Local { // 删除本地文件
+		if strings.Contains(fileFromDb.Url, global.GVA_CONFIG.LocalUpload.FilePath) {
+			if err = os.Remove(fileFromDb.Url); err != nil {
+				err = errors.New("本地文件删除失败, err:" + err.Error())
+			}
+		}
+	} else {
+		if err = utils.DeleteFile(file.Key); err != nil { // 删除七牛云文件
+			err = errors.New("七牛云文件删除失败, err:" + err.Error())
+		}
+	}
+	err = global.GVA_DB.Where("id = ?", file.ID).Unscoped().Delete(file).Error
 	return err
 }
 
@@ -57,3 +78,30 @@ func GetFileRecordInfoList(info request.PageInfo) (err error, list interface{},
 	err = db.Limit(limit).Offset(offset).Order("updated_at desc").Find(&fileLists).Error
 	return err, fileLists, total
 }
+
+// @title    UploadFile
+// @description   根据配置文件判断是文件上传到本地或者七牛云
+// @auth                     （2020/04/05  20:22）
+// @param     header          *multipart.FileHeader
+// @param     noSave          string
+// @return    err             error
+// @return    file            file model.ExaFileUploadAndDownload
+
+func UploadFile(header *multipart.FileHeader, noSave string) (err error, file model.ExaFileUploadAndDownload) {
+	var filePath, key string
+	var f model.ExaFileUploadAndDownload
+	if global.GVA_CONFIG.LocalUpload.Local { // 本地上传
+		err, filePath, key = utils.UploadFileLocal(header)
+	} else { // 七牛云上传
+		err, filePath, key = utils.UploadRemote(header)
+	}
+	if noSave == "0" {
+		f.Url = filePath
+		f.Name = header.Filename
+		s := strings.Split(f.Name, ".")
+		f.Tag = s[len(s)-1]
+		f.Key = key
+		return Upload(f), f
+	}
+	return
+}