You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
109 lines
3.3 KiB
Go
109 lines
3.3 KiB
Go
11 months ago
|
package example
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"mime/multipart"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"miniapp/global"
|
||
|
|
"miniapp/model/common/request"
|
||
|
|
"miniapp/model/example"
|
||
|
|
"miniapp/utils/upload"
|
||
|
|
)
|
||
|
|
|
||
|
|
//@author: [piexlmax](https://github.com/piexlmax)
|
||
|
|
//@function: Upload
|
||
|
|
//@description: 创建文件上传记录
|
||
|
|
//@param: file model.ExaFileUploadAndDownload
|
||
|
|
//@return: error
|
||
|
|
|
||
|
|
func (e *FileUploadAndDownloadService) Upload(file example.ExaFileUploadAndDownload) error {
|
||
|
|
return global.GVA_DB.Create(&file).Error
|
||
|
|
}
|
||
|
|
|
||
|
|
//@author: [piexlmax](https://github.com/piexlmax)
|
||
|
|
//@function: FindFile
|
||
|
|
//@description: 查询文件记录
|
||
|
|
//@param: id uint
|
||
|
|
//@return: model.ExaFileUploadAndDownload, error
|
||
|
|
|
||
|
|
func (e *FileUploadAndDownloadService) FindFile(id uint) (example.ExaFileUploadAndDownload, error) {
|
||
|
|
var file example.ExaFileUploadAndDownload
|
||
|
|
err := global.GVA_DB.Where("id = ?", id).First(&file).Error
|
||
|
|
return file, err
|
||
|
|
}
|
||
|
|
|
||
|
|
//@author: [piexlmax](https://github.com/piexlmax)
|
||
|
|
//@function: DeleteFile
|
||
|
|
//@description: 删除文件记录
|
||
|
|
//@param: file model.ExaFileUploadAndDownload
|
||
|
|
//@return: err error
|
||
|
|
|
||
|
|
func (e *FileUploadAndDownloadService) DeleteFile(file example.ExaFileUploadAndDownload) (err error) {
|
||
|
|
var fileFromDb example.ExaFileUploadAndDownload
|
||
|
|
fileFromDb, err = e.FindFile(file.ID)
|
||
|
|
if err != nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
oss := upload.NewOss()
|
||
|
|
if err = oss.DeleteFile(fileFromDb.Key); err != nil {
|
||
|
|
return errors.New("文件删除失败")
|
||
|
|
}
|
||
|
|
err = global.GVA_DB.Where("id = ?", file.ID).Unscoped().Delete(&file).Error
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
// EditFileName 编辑文件名或者备注
|
||
|
|
func (e *FileUploadAndDownloadService) EditFileName(file example.ExaFileUploadAndDownload) (err error) {
|
||
|
|
var fileFromDb example.ExaFileUploadAndDownload
|
||
|
|
return global.GVA_DB.Where("id = ?", file.ID).First(&fileFromDb).Update("name", file.Name).Error
|
||
|
|
}
|
||
|
|
|
||
|
|
//@author: [piexlmax](https://github.com/piexlmax)
|
||
|
|
//@function: GetFileRecordInfoList
|
||
|
|
//@description: 分页获取数据
|
||
|
|
//@param: info request.PageInfo
|
||
|
|
//@return: list interface{}, total int64, err error
|
||
|
|
|
||
|
|
func (e *FileUploadAndDownloadService) GetFileRecordInfoList(info request.PageInfo) (list interface{}, total int64, err error) {
|
||
|
|
limit := info.PageSize
|
||
|
|
offset := info.PageSize * (info.Page - 1)
|
||
|
|
keyword := info.Keyword
|
||
|
|
db := global.GVA_DB.Model(&example.ExaFileUploadAndDownload{})
|
||
|
|
var fileLists []example.ExaFileUploadAndDownload
|
||
|
|
if len(keyword) > 0 {
|
||
|
|
db = db.Where("name LIKE ?", "%"+keyword+"%")
|
||
|
|
}
|
||
|
|
err = db.Count(&total).Error
|
||
|
|
if err != nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
err = db.Limit(limit).Offset(offset).Order("updated_at desc").Find(&fileLists).Error
|
||
|
|
return fileLists, total, err
|
||
|
|
}
|
||
|
|
|
||
|
|
//@author: [piexlmax](https://github.com/piexlmax)
|
||
|
|
//@function: UploadFile
|
||
|
|
//@description: 根据配置文件判断是文件上传到本地或者七牛云
|
||
|
|
//@param: header *multipart.FileHeader, noSave string
|
||
|
|
//@return: file model.ExaFileUploadAndDownload, err error
|
||
|
|
|
||
|
|
func (e *FileUploadAndDownloadService) UploadFile(header *multipart.FileHeader, noSave string) (file example.ExaFileUploadAndDownload, err error) {
|
||
|
|
oss := upload.NewOss()
|
||
|
|
filePath, key, uploadErr := oss.UploadFile(header)
|
||
|
|
if uploadErr != nil {
|
||
|
|
panic(uploadErr)
|
||
|
|
}
|
||
|
|
s := strings.Split(header.Filename, ".")
|
||
|
|
f := example.ExaFileUploadAndDownload{
|
||
|
|
Url: filePath,
|
||
|
|
Name: header.Filename,
|
||
|
|
Tag: s[len(s)-1],
|
||
|
|
Key: key,
|
||
|
|
}
|
||
|
|
if noSave == "0" {
|
||
|
|
return f, e.Upload(f)
|
||
|
|
}
|
||
|
|
return f, nil
|
||
|
|
}
|