63 lines
1.7 KiB
Go
63 lines
1.7 KiB
Go
|
package file
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"mini_server/internal/config"
|
|||
|
|
"mini_server/internal/dao"
|
|||
|
|
"mini_server/internal/server/help"
|
|||
|
|
"net/http"
|
|||
|
|
|
|||
|
|
"github.com/gin-gonic/gin"
|
|||
|
|
"github.com/sirupsen/logrus"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type DownloadURI struct {
|
|||
|
|
FileId string `uri:"file_id" binding:"len=32"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 文件下载接口
|
|||
|
|
func Download(c *gin.Context) {
|
|||
|
|
var param DownloadURI
|
|||
|
|
if err := c.ShouldBindUri(¶m); err != nil {
|
|||
|
|
c.JSON(http.StatusBadRequest, fmt.Sprintf("invalid download uri,err:%v", err))
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
fileId := param.FileId
|
|||
|
|
var data *dao.File
|
|||
|
|
var err error
|
|||
|
|
var ok bool
|
|||
|
|
|
|||
|
|
if config.FileCacheEnable { // 从文件缓存中查询
|
|||
|
|
if data, ok = help.GetFileCache(fileId); ok {
|
|||
|
|
logrus.Tracef("从内存缓存中获取文件数据，file_uuid:%v", fileId)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if data == nil { // 未获取到文件缓存数据
|
|||
|
|
logrus.Tracef("从msyql中获取文件数据，file_uuid:%v", fileId)
|
|||
|
|
data, err = dao.File_Query(param.FileId)
|
|||
|
|
if err != nil {
|
|||
|
|
logrus.Errorf("文件下载失败,fileId:%s,err:%v", param.FileId, err)
|
|||
|
|
c.JSON(http.StatusInternalServerError, "文件下载失败")
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
if data == nil { // 文件不存在
|
|||
|
|
logrus.Errorf("文件下载失败，文件file_id:%v不存在", param.FileId)
|
|||
|
|
c.JSON(http.StatusNotFound, "文件不存在")
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if config.FileCacheEnable { // 更新到文件缓存中
|
|||
|
|
help.SetFileCache(fileId, data)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
logrus.Tracef("文件%s下载成功，文件大小:%d，数据库中blob实际大小:%d", data.FileName, data.FileSize, len(data.FileContent))
|
|||
|
|
c.Header("Content-Disposition", "attachment;filename="+data.FileName)
|
|||
|
|
c.Header("Content-Type", data.FileType)
|
|||
|
|
c.Header("Accept-Length", fmt.Sprintf("%d", data.FileSize))
|
|||
|
|
c.Writer.WriteHeader(http.StatusOK)
|
|||
|
|
c.Writer.Write(data.FileContent)
|
|||
|
|
}
|