d40c815760
* 媒体库增加分类，图库多选择时优化。 * 重构 JWT token 生成，使用 `New()` 函数代替直接创建实例 * 上传组件支持查看大图 (#1982) * 将参数缓存，媒体库增加分类，图库多选择时优化。 (#1978) * 媒体库增加分类，图库多选择时优化。 *修复文件上传进度显示bug&按钮样式优化 (#1986) * fix:添加内部 iframe 展示网页，优化 permission 代码 * 俩个uuid库合并一个，更新库到当前版本。 * 优化关于我们界面 * feat: 个人中心头像调整，媒体库兼容性调整。 * feat: 自动化代码前端页面美化，多余按钮收入专家模式 * feat: 增加单独生成server功能 * feat: 限制单独生成前后端的情况下的细节配置 * feat: 修复全选失败报错的问题 --------- Co-authored-by: task <121913992@qq.com> Co-authored-by: Feng.YJ <32027253+huiyifyj@users.noreply.github.com> Co-authored-by: will0523 <dygsunshine@163.com> Co-authored-by: task <ms.yangdan@gmail.com> Co-authored-by: sslee <57312216+GIS142857@users.noreply.github.com> Co-authored-by: bypanghu <bypanghu@163.com> Co-authored-by: Azir <2075125282@qq.com> Co-authored-by: piexlMax(奇淼 <qimiaojiangjizhao@gmail.com> Co-authored-by: krank <emosick@qq.com>
113 lines
3.0 KiB
Go
113 lines
3.0 KiB
Go
package utils
|
|
|
|
import (
|
|
"errors"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
// 前端传来文件片与当前片为什么文件的第几片
|
|
// 后端拿到以后比较次分片是否上传 或者是否为不完全片
|
|
// 前端发送每片多大
|
|
// 前端告知是否为最后一片且是否完成
|
|
|
|
const (
|
|
breakpointDir = "./breakpointDir/"
|
|
finishDir = "./fileDir/"
|
|
)
|
|
|
|
//@author: [piexlmax](https://github.com/piexlmax)
|
|
//@function: BreakPointContinue
|
|
//@description: 断点续传
|
|
//@param: content []byte, fileName string, contentNumber int, contentTotal int, fileMd5 string
|
|
//@return: error, string
|
|
|
|
func BreakPointContinue(content []byte, fileName string, contentNumber int, contentTotal int, fileMd5 string) (string, error) {
|
|
path := breakpointDir + fileMd5 + "/"
|
|
err := os.MkdirAll(path, os.ModePerm)
|
|
if err != nil {
|
|
return path, err
|
|
}
|
|
pathC, err := makeFileContent(content, fileName, path, contentNumber)
|
|
return pathC, err
|
|
}
|
|
|
|
//@author: [piexlmax](https://github.com/piexlmax)
|
|
//@function: CheckMd5
|
|
//@description: 检查Md5
|
|
//@param: content []byte, chunkMd5 string
|
|
//@return: CanUpload bool
|
|
|
|
func CheckMd5(content []byte, chunkMd5 string) (CanUpload bool) {
|
|
fileMd5 := MD5V(content)
|
|
if fileMd5 == chunkMd5 {
|
|
return true // 可以继续上传
|
|
} else {
|
|
return false // 切片不完整，废弃
|
|
}
|
|
}
|
|
|
|
//@author: [piexlmax](https://github.com/piexlmax)
|
|
//@function: makeFileContent
|
|
//@description: 创建切片内容
|
|
//@param: content []byte, fileName string, FileDir string, contentNumber int
|
|
//@return: string, error
|
|
|
|
func makeFileContent(content []byte, fileName string, FileDir string, contentNumber int) (string, error) {
|
|
if strings.Contains(fileName, "..") || strings.Contains(FileDir, "..") {
|
|
return "", errors.New("文件名或路径不合法")
|
|
}
|
|
path := FileDir + fileName + "_" + strconv.Itoa(contentNumber)
|
|
f, err := os.Create(path)
|
|
if err != nil {
|
|
return path, err
|
|
} else {
|
|
_, err = f.Write(content)
|
|
if err != nil {
|
|
return path, err
|
|
}
|
|
}
|
|
defer f.Close()
|
|
return path, nil
|
|
}
|
|
|
|
//@author: [piexlmax](https://github.com/piexlmax)
|
|
//@function: makeFileContent
|
|
//@description: 创建切片文件
|
|
//@param: fileName string, FileMd5 string
|
|
//@return: error, string
|
|
|
|
func MakeFile(fileName string, FileMd5 string) (string, error) {
|
|
rd, err := os.ReadDir(breakpointDir + FileMd5)
|
|
if err != nil {
|
|
return finishDir + fileName, err
|
|
}
|
|
_ = os.MkdirAll(finishDir, os.ModePerm)
|
|
fd, err := os.OpenFile(finishDir+fileName, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0o644)
|
|
if err != nil {
|
|
return finishDir + fileName, err
|
|
}
|
|
defer fd.Close()
|
|
for k := range rd {
|
|
content, _ := os.ReadFile(breakpointDir + FileMd5 + "/" + fileName + "_" + strconv.Itoa(k))
|
|
_, err = fd.Write(content)
|
|
if err != nil {
|
|
_ = os.Remove(finishDir + fileName)
|
|
return finishDir + fileName, err
|
|
}
|
|
}
|
|
return finishDir + fileName, nil
|
|
}
|
|
|
|
//@author: [piexlmax](https://github.com/piexlmax)
|
|
//@function: RemoveChunk
|
|
//@description: 移除切片
|
|
//@param: FileMd5 string
|
|
//@return: error
|
|
|
|
func RemoveChunk(FileMd5 string) error {
|
|
err := os.RemoveAll(breakpointDir + FileMd5)
|
|
return err
|
|
}
|