init

wangpengfei

2023-08-24 487b20629178159396a93460f7355f5ebee7a8a0

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
package upload
 
import (
    "errors"
    "io"
    "mime/multipart"
    "os"
    "path"
    "strings"
    "time"
 
    "github.com/flipped-aurora/gin-vue-admin/server/global"
    "github.com/flipped-aurora/gin-vue-admin/server/utils"
    "go.uber.org/zap"
)
 
type Local struct{}
 
//@author: [piexlmax](https://github.com/piexlmax)
//@author: [ccfish86](https://github.com/ccfish86)
//@author: [SliverHorn](https://github.com/SliverHorn)
//@object: *Local
//@function: UploadFile
//@description: 上传文件
//@param: file *multipart.FileHeader
//@return: string, string, error
 
func (*Local) UploadFile(file *multipart.FileHeader) (string, string, error) {
    // 读取文件后缀
    ext := path.Ext(file.Filename)
    // 读取文件名并加密
    name := strings.TrimSuffix(file.Filename, ext)
    name = utils.MD5V([]byte(name))
    // 拼接新文件名
    filename := name + "_" + time.Now().Format("20060102150405") + ext
    // 尝试创建此路径
    mkdirErr := os.MkdirAll(global.GVA_CONFIG.Local.StorePath, os.ModePerm)
    if mkdirErr != nil {
        global.GVA_LOG.Error("function os.MkdirAll() failed", zap.Any("err", mkdirErr.Error()))
        return "", "", errors.New("function os.MkdirAll() failed, err:" + mkdirErr.Error())
    }
    // 拼接路径和文件名
    p := global.GVA_CONFIG.Local.StorePath + "/" + filename
    filepath := global.GVA_CONFIG.Local.Path + "/" + filename
 
    f, openError := file.Open() // 读取文件
    if openError != nil {
        global.GVA_LOG.Error("function file.Open() failed", zap.Any("err", openError.Error()))
        return "", "", errors.New("function file.Open() failed, err:" + openError.Error())
    }
    defer f.Close() // 创建文件 defer 关闭
 
    out, createErr := os.Create(p)
    if createErr != nil {
        global.GVA_LOG.Error("function os.Create() failed", zap.Any("err", createErr.Error()))
 
        return "", "", errors.New("function os.Create() failed, err:" + createErr.Error())
    }
    defer out.Close() // 创建文件 defer 关闭
 
    _, copyErr := io.Copy(out, f) // 传输（拷贝）文件
    if copyErr != nil {
        global.GVA_LOG.Error("function io.Copy() failed", zap.Any("err", copyErr.Error()))
        return "", "", errors.New("function io.Copy() failed, err:" + copyErr.Error())
    }
    return filepath, filename, nil
}
 
//@author: [piexlmax](https://github.com/piexlmax)
//@author: [ccfish86](https://github.com/ccfish86)
//@author: [SliverHorn](https://github.com/SliverHorn)
//@object: *Local
//@function: DeleteFile
//@description: 删除文件
//@param: key string
//@return: error
 
func (*Local) DeleteFile(key string) error {
    p := global.GVA_CONFIG.Local.StorePath + "/" + key
    if strings.Contains(p, global.GVA_CONFIG.Local.StorePath) {
        if err := os.Remove(p); err != nil {
            return errors.New("本地文件删除失败, err:" + err.Error())
        }
    }
    return nil
}