mirror of
https://gitee.com/xiangheng/x_admin.git
synced 2025-10-06 00:27:00 +08:00
98 lines
2.9 KiB
Go
98 lines
2.9 KiB
Go
package commonController
|
|
|
|
import (
|
|
"fmt"
|
|
"net/url"
|
|
"os"
|
|
"regexp"
|
|
"x_admin/core/response"
|
|
"x_admin/service/commonService"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func UploadChunkRoute(rg *gin.RouterGroup) {
|
|
handle := uploadChunkHandler{
|
|
uploadPath: "./uploads",
|
|
tmpPath: "./tmp",
|
|
}
|
|
os.MkdirAll(handle.uploadPath, 0755)
|
|
os.MkdirAll(handle.tmpPath, 0755)
|
|
|
|
rg = rg.Group("/common")
|
|
rg.GET("/uploadChunk/CheckFileExist", handle.CheckFileExist)
|
|
rg.GET("/uploadChunk/CheckChunkExist", handle.CheckChunkExist)
|
|
rg.POST("/uploadChunk/UploadChunk", handle.UploadChunk)
|
|
rg.POST("/uploadChunk/MergeChunk", handle.MergeChunk)
|
|
}
|
|
|
|
type uploadChunkHandler struct {
|
|
uploadPath string
|
|
tmpPath string
|
|
}
|
|
|
|
func (uh uploadChunkHandler) CheckFileExist(c *gin.Context) {
|
|
var fileMd5 = c.Query("fileMd5")
|
|
var fileName = url.QueryEscape(c.Query("fileName"))
|
|
// 正则检查MD5
|
|
reg := regexp.MustCompile(`^[a-fA-F0-9]{32}$`)
|
|
if !reg.MatchString(fileMd5) {
|
|
response.FailWithMsg(c, response.SystemError, "MD5格式错误")
|
|
return
|
|
}
|
|
var filePath = fmt.Sprintf("%s/%s_%s", uh.uploadPath, fileMd5, fileName)
|
|
// 检查文件是否存在
|
|
if commonService.UploadChunkService.CheckFileExist(filePath) {
|
|
response.OkWithData(c, filePath)
|
|
return
|
|
}
|
|
response.OkWithData(c, nil)
|
|
}
|
|
|
|
// 检查chunk是否存在
|
|
func (uh uploadChunkHandler) CheckChunkExist(c *gin.Context) {
|
|
fileMd5 := c.Query("fileMd5") // 上传文件的md5
|
|
index := c.Query("index") // 分片序号
|
|
chunkPath := fmt.Sprintf("%s/%s/%s", uh.tmpPath, fileMd5, index)
|
|
if commonService.UploadChunkService.CheckFileExist(chunkPath) {
|
|
response.Ok(c)
|
|
return
|
|
}
|
|
response.FailWithMsg(c, response.SystemError, "分片不存在")
|
|
}
|
|
|
|
// UploadChunk 上传分片
|
|
func (uh uploadChunkHandler) UploadChunk(c *gin.Context) {
|
|
chunk, _ := c.FormFile("chunk") // 分片文件
|
|
index := c.PostForm("index") // 分片序号
|
|
fileMd5 := c.PostForm("fileMd5") // 上传文件的md5
|
|
|
|
chunkDir := fmt.Sprintf("%s/%s", uh.tmpPath, fileMd5)
|
|
chunkPath := fmt.Sprintf("%s/%s", chunkDir, index)
|
|
err := commonService.UploadChunkService.UploadChunk(chunkDir, chunkPath, chunk)
|
|
if err != nil {
|
|
response.FailWithMsg(c, response.SystemError, err.Error())
|
|
return
|
|
}
|
|
response.Ok(c)
|
|
}
|
|
func (uh uploadChunkHandler) MergeChunk(c *gin.Context) {
|
|
var MergeChunk struct {
|
|
FileMd5 string `json:"fileMd5"` // 上传文件的md5
|
|
FileName string `json:"fileName"` // 文件名
|
|
ChunkCount int `json:"chunkCount"` // 分片数量
|
|
}
|
|
bindErr := c.ShouldBindJSON(&MergeChunk)
|
|
if bindErr != nil {
|
|
response.FailWithMsg(c, response.SystemError, bindErr.Error())
|
|
return
|
|
}
|
|
var filePath = fmt.Sprintf("%s/%s_%s", uh.uploadPath, MergeChunk.FileMd5, url.QueryEscape(MergeChunk.FileName))
|
|
err := commonService.UploadChunkService.MergeChunk(MergeChunk.FileMd5, filePath, MergeChunk.ChunkCount)
|
|
if err != nil {
|
|
response.FailWithMsg(c, response.SystemError, err.Error())
|
|
return
|
|
}
|
|
response.OkWithData(c, filePath)
|
|
}
|