84 lines
2.0 KiB
Go
84 lines
2.0 KiB
Go
package file
|
||
|
||
import (
|
||
"bytes"
|
||
"encoding/hex"
|
||
"io"
|
||
"log"
|
||
"mini_server/internal/dao"
|
||
"mini_server/internal/server/help"
|
||
"mini_server/internal/server/httpreply"
|
||
"net/http"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
"github.com/google/uuid"
|
||
"github.com/sirupsen/logrus"
|
||
)
|
||
|
||
type UploadReply struct {
|
||
FileId string `json:"file_id"` // 文件uuid，返回file_uuid
|
||
}
|
||
|
||
// 文件上传接口
|
||
|
||
func Upload(c *gin.Context) {
|
||
reply := httpreply.NewDefaultReplyData()
|
||
defer httpreply.Reply(c, reply)
|
||
|
||
userId, ok := help.GetUserId(c)
|
||
if !ok {
|
||
reply.Status = http.StatusNotFound
|
||
reply.Message = "无法获取用户信息"
|
||
return
|
||
}
|
||
|
||
file, err := c.FormFile("file")
|
||
if err != nil {
|
||
reply.Status = http.StatusBadRequest
|
||
reply.Message = "无法获取文件信息"
|
||
return
|
||
}
|
||
|
||
// 读取文件中的数据，并写入到数据库中
|
||
reader, _ := file.Open()
|
||
defer reader.Close()
|
||
|
||
// 添加文件大小限制，目前使用mediumblob，有16M存储限制
|
||
if file.Size >= int64(1024*1024*16) {
|
||
reply.Status = http.StatusRequestEntityTooLarge
|
||
reply.Message = "上传文件超过限制，目前最多支持16M文件上传"
|
||
return
|
||
}
|
||
|
||
// 拷贝数据
|
||
buf := bytes.NewBuffer(nil)
|
||
if _, err := io.Copy(buf, reader); err != nil {
|
||
reply.Status = http.StatusInternalServerError
|
||
reply.Message = "copy file failed"
|
||
return
|
||
}
|
||
fileType := http.DetectContentType(buf.Bytes())
|
||
log.Printf("文件上传请求，文件名:%v, 文件大小:%d, 文件类型:%v", file.Filename, file.Size, fileType)
|
||
|
||
// 将数据写入数据库中
|
||
bytesUUID, _ := uuid.NewUUID()
|
||
uuid := hex.EncodeToString(bytesUUID[:])
|
||
data := &dao.File{
|
||
UUID: uuid, // 作为唯一标识
|
||
UserId: userId,
|
||
FileType: fileType,
|
||
FileName: file.Filename,
|
||
FileSize: uint32(file.Size),
|
||
FileContent: buf.Bytes(),
|
||
}
|
||
|
||
if _, err = dao.File_Insert(data); err != nil {
|
||
logrus.Errorf("dao.File_Insert failed,err:%v", err)
|
||
reply.Status = http.StatusInternalServerError
|
||
reply.Message = "文件上传保存失败"
|
||
return
|
||
}
|
||
|
||
reply.Data = &UploadReply{FileId: uuid}
|
||
}
|