123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130 |
- package example
- import (
- "io/ioutil"
- "strconv"
- "github.com/flipped-aurora/gin-vue-admin/server/global"
- "github.com/flipped-aurora/gin-vue-admin/server/model/common/response"
- exampleRes "github.com/flipped-aurora/gin-vue-admin/server/model/example/response"
- "github.com/flipped-aurora/gin-vue-admin/server/utils"
- "github.com/gin-gonic/gin"
- "go.uber.org/zap"
- )
- func (u *FileUploadAndDownloadApi) BreakpointContinue(c *gin.Context) {
- fileMd5 := c.Request.FormValue("fileMd5")
- fileName := c.Request.FormValue("fileName")
- chunkMd5 := c.Request.FormValue("chunkMd5")
- chunkNumber, _ := strconv.Atoi(c.Request.FormValue("chunkNumber"))
- chunkTotal, _ := strconv.Atoi(c.Request.FormValue("chunkTotal"))
- _, FileHeader, err := c.Request.FormFile("file")
- if err != nil {
- global.GVA_LOG.Error("接收文件失败!", zap.Any("err", err))
- response.FailWithMessage("接收文件失败", c)
- return
- }
- f, err := FileHeader.Open()
- if err != nil {
- global.GVA_LOG.Error("文件读取失败!", zap.Any("err", err))
- response.FailWithMessage("文件读取失败", c)
- return
- }
- defer f.Close()
- cen, _ := ioutil.ReadAll(f)
- if !utils.CheckMd5(cen, chunkMd5) {
- global.GVA_LOG.Error("检查md5失败!", zap.Any("err", err))
- response.FailWithMessage("检查md5失败", c)
- return
- }
- err, file := fileUploadAndDownloadService.FindOrCreateFile(fileMd5, fileName, chunkTotal)
- if err != nil {
- global.GVA_LOG.Error("查找或创建记录失败!", zap.Any("err", err))
- response.FailWithMessage("查找或创建记录失败", c)
- return
- }
- err, pathc := utils.BreakPointContinue(cen, fileName, chunkNumber, chunkTotal, fileMd5)
- if err != nil {
- global.GVA_LOG.Error("断点续传失败!", zap.Any("err", err))
- response.FailWithMessage("断点续传失败", c)
- return
- }
- if err = fileUploadAndDownloadService.CreateFileChunk(file.ID, pathc, chunkNumber); err != nil {
- global.GVA_LOG.Error("创建文件记录失败!", zap.Any("err", err))
- response.FailWithMessage("创建文件记录失败", c)
- return
- }
- response.OkWithMessage("切片创建成功", c)
- }
- func (u *FileUploadAndDownloadApi) FindFile(c *gin.Context) {
- fileMd5 := c.Query("fileMd5")
- fileName := c.Query("fileName")
- chunkTotal, _ := strconv.Atoi(c.Query("chunkTotal"))
- err, file := fileUploadAndDownloadService.FindOrCreateFile(fileMd5, fileName, chunkTotal)
- if err != nil {
- global.GVA_LOG.Error("查找失败!", zap.Any("err", err))
- response.FailWithMessage("查找失败", c)
- } else {
- response.OkWithDetailed(exampleRes.FileResponse{File: file}, "查找成功", c)
- }
- }
- func (b *FileUploadAndDownloadApi) BreakpointContinueFinish(c *gin.Context) {
- fileMd5 := c.Query("fileMd5")
- fileName := c.Query("fileName")
- err, filePath := utils.MakeFile(fileName, fileMd5)
- if err != nil {
- global.GVA_LOG.Error("文件创建失败!", zap.Any("err", err))
- response.FailWithDetailed(exampleRes.FilePathResponse{FilePath: filePath}, "文件创建失败", c)
- } else {
- response.OkWithDetailed(exampleRes.FilePathResponse{FilePath: filePath}, "文件创建成功", c)
- }
- }
- func (u *FileUploadAndDownloadApi) RemoveChunk(c *gin.Context) {
- fileMd5 := c.Query("fileMd5")
- fileName := c.Query("fileName")
- filePath := c.Query("filePath")
- err := utils.RemoveChunk(fileMd5)
- err = fileUploadAndDownloadService.DeleteFileChunk(fileMd5, fileName, filePath)
- if err != nil {
- global.GVA_LOG.Error("缓存切片删除失败!", zap.Any("err", err))
- response.FailWithDetailed(exampleRes.FilePathResponse{FilePath: filePath}, "缓存切片删除失败", c)
- } else {
- response.OkWithDetailed(exampleRes.FilePathResponse{FilePath: filePath}, "缓存切片删除成功", c)
- }
- }
|