采购负责人

zhangqian

2023-08-29 e668092cf0bca471e665580c255c04a6ffc3cb8f

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
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.Index(fileName, "..") > -1 || strings.Index(FileDir, "..") > -1 {
        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
}