正在显示
3 个修改的文件
包含
1 行增加
和
180 行删除
| @@ -11,6 +11,7 @@ require ( | @@ -11,6 +11,7 @@ require ( | ||
| 11 | github.com/satori/go.uuid v1.2.0 | 11 | github.com/satori/go.uuid v1.2.0 |
| 12 | github.com/sony/sonyflake v1.0.0 | 12 | github.com/sony/sonyflake v1.0.0 |
| 13 | gitlab.fjmaimaimai.com/mmm-go/gocomm v0.0.1 | 13 | gitlab.fjmaimaimai.com/mmm-go/gocomm v0.0.1 |
| 14 | + golang.org/x/image v0.0.0-20191009234506-e7c1f5e7dbb8 // indirect | ||
| 14 | google.golang.org/appengine v1.6.2 // indirect | 15 | google.golang.org/appengine v1.6.2 // indirect |
| 15 | ) | 16 | ) |
| 16 | 17 |
services/filestore/file.go
已删除
100755 → 0
| 1 | -package filestore | ||
| 2 | - | ||
| 3 | -import ( | ||
| 4 | - "errors" | ||
| 5 | - "fmt" | ||
| 6 | - "io" | ||
| 7 | - "mime/multipart" | ||
| 8 | - "os" | ||
| 9 | - "path" | ||
| 10 | - | ||
| 11 | - "github.com/sony/sonyflake" | ||
| 12 | -) | ||
| 13 | - | ||
| 14 | -var ( | ||
| 15 | - ImageFilePath string | ||
| 16 | - VoiceFilePath string | ||
| 17 | - VideoFilePath string | ||
| 18 | -) | ||
| 19 | - | ||
| 20 | -func init() { | ||
| 21 | - //TODO 初始化文件存储路径,建议手动创建 | ||
| 22 | - ImageFilePath = "" | ||
| 23 | - VoiceFilePath = "" | ||
| 24 | - VideoFilePath = "" | ||
| 25 | - // ValidatePath(ImageFilePath) | ||
| 26 | - // ValidatePath(VoiceFilePath) | ||
| 27 | - // ValidatePath(VideoFilePath) | ||
| 28 | -} | ||
| 29 | - | ||
| 30 | -//IFileStore 文件存储相关 | ||
| 31 | -type IFileStore interface { | ||
| 32 | - StoreFile() (string, error) //实际存储文件操作 | ||
| 33 | -} | ||
| 34 | - | ||
| 35 | -type IFileReader interface { | ||
| 36 | - ReadFileSrc(id int64) (string, error) //实际获取文件链接 | ||
| 37 | -} | ||
| 38 | - | ||
| 39 | -func ValidateFileExt(fileExt string) bool { | ||
| 40 | - allowExt := map[string]bool{ | ||
| 41 | - "jpg": true, | ||
| 42 | - "png": true, | ||
| 43 | - "mp3": true, | ||
| 44 | - } | ||
| 45 | - if _, ok := allowExt[fileExt]; ok { | ||
| 46 | - return true | ||
| 47 | - } | ||
| 48 | - return false | ||
| 49 | -} | ||
| 50 | - | ||
| 51 | -func NewFileStore(fHeader *multipart.FileHeader) (IFileStore, error) { | ||
| 52 | - filename := fHeader.Filename | ||
| 53 | - fileExt := path.Ext(filename) | ||
| 54 | - if ok := ValidateFileExt(fileExt); !ok { | ||
| 55 | - return nil, errors.New("ValidateFileExt fail") | ||
| 56 | - } | ||
| 57 | - var filestore IFileStore | ||
| 58 | - switch fileExt { | ||
| 59 | - case "jpg", "png": | ||
| 60 | - filestore = NewImageFileStore(fHeader, fileExt) | ||
| 61 | - case "mp3": | ||
| 62 | - filestore = NewVideoFileStore(fHeader, fileExt) | ||
| 63 | - } | ||
| 64 | - return filestore, nil | ||
| 65 | -} | ||
| 66 | - | ||
| 67 | -func NewReader(filetype string) (IFileReader, error) { | ||
| 68 | - var filestore IFileReader | ||
| 69 | - switch filetype { | ||
| 70 | - case "imgage": | ||
| 71 | - filestore = &ImageFileStore{} | ||
| 72 | - case "mp3": | ||
| 73 | - filestore = &VideoFileStore{} | ||
| 74 | - default: | ||
| 75 | - return nil, fmt.Errorf("err:filetype is %s", filetype) | ||
| 76 | - } | ||
| 77 | - return filestore, nil | ||
| 78 | -} | ||
| 79 | - | ||
| 80 | -//ImageFileStore 图片文件 | ||
| 81 | -type ImageFileStore struct { | ||
| 82 | - fHeader *multipart.FileHeader | ||
| 83 | - fileType string | ||
| 84 | - size int | ||
| 85 | -} | ||
| 86 | - | ||
| 87 | -var ( | ||
| 88 | - _ IFileStore = &ImageFileStore{} | ||
| 89 | -) | ||
| 90 | - | ||
| 91 | -func NewImageFileStore(fHeader *multipart.FileHeader, filetype string) *ImageFileStore { | ||
| 92 | - return &ImageFileStore{ | ||
| 93 | - fHeader: fHeader, | ||
| 94 | - fileType: filetype, | ||
| 95 | - } | ||
| 96 | -} | ||
| 97 | -func (f ImageFileStore) StoreFile() (string, error) { | ||
| 98 | - storePath := fmt.Sprintf("%s%s.%s", ImageFilePath, randomName(), f.fileType) | ||
| 99 | - tempFile, err := f.fHeader.Open() | ||
| 100 | - if err != nil { | ||
| 101 | - return "", err | ||
| 102 | - } | ||
| 103 | - defer tempFile.Close() | ||
| 104 | - if err = SaveToFile(storePath, tempFile); err != nil { | ||
| 105 | - return "", err | ||
| 106 | - } | ||
| 107 | - //数据库操作 | ||
| 108 | - return storePath, nil | ||
| 109 | -} | ||
| 110 | - | ||
| 111 | -func (f ImageFileStore) ReadFileSrc(id int64) (string, error) { | ||
| 112 | - //数据库操作 | ||
| 113 | - return "", nil | ||
| 114 | -} | ||
| 115 | - | ||
| 116 | -//VoiceFileStore 音频文件 | ||
| 117 | -type VideoFileStore struct { | ||
| 118 | - fHeader *multipart.FileHeader | ||
| 119 | - size int | ||
| 120 | - fileType string | ||
| 121 | -} | ||
| 122 | - | ||
| 123 | -var ( | ||
| 124 | - _ IFileStore = &VideoFileStore{} | ||
| 125 | -) | ||
| 126 | - | ||
| 127 | -func NewVideoFileStore(fHeader *multipart.FileHeader, filetype string) *VideoFileStore { | ||
| 128 | - return &VideoFileStore{ | ||
| 129 | - fHeader: fHeader, | ||
| 130 | - fileType: filetype, | ||
| 131 | - } | ||
| 132 | -} | ||
| 133 | - | ||
| 134 | -func (f VideoFileStore) StoreFile() (string, error) { | ||
| 135 | - storePath := fmt.Sprintf("%s%s.%s", ImageFilePath, randomName(), f.fileType) | ||
| 136 | - tempFile, err := f.fHeader.Open() | ||
| 137 | - if err != nil { | ||
| 138 | - return "", err | ||
| 139 | - } | ||
| 140 | - defer tempFile.Close() | ||
| 141 | - if err = SaveToFile(storePath, tempFile); err != nil { | ||
| 142 | - return "", err | ||
| 143 | - } | ||
| 144 | - //数据库操作 | ||
| 145 | - return storePath, nil | ||
| 146 | -} | ||
| 147 | - | ||
| 148 | -func (f VideoFileStore) ReadFileSrc(id int64) (string, error) { | ||
| 149 | - //数据库操作 | ||
| 150 | - return "", nil | ||
| 151 | -} | ||
| 152 | - | ||
| 153 | -var sf = sonyflake.NewSonyflake(sonyflake.Settings{}) | ||
| 154 | - | ||
| 155 | -func randomName() string { | ||
| 156 | - num, _ := sf.NextID() | ||
| 157 | - return fmt.Sprintf("%d", num) | ||
| 158 | -} | ||
| 159 | - | ||
| 160 | -func ValidatePath(s string) error { | ||
| 161 | - info, err := os.Stat(s) | ||
| 162 | - if err != nil { | ||
| 163 | - return err | ||
| 164 | - } | ||
| 165 | - if ok := info.IsDir(); ok { | ||
| 166 | - return nil | ||
| 167 | - } | ||
| 168 | - return errors.New("path must be directory") | ||
| 169 | -} | ||
| 170 | - | ||
| 171 | -func SaveToFile(storePath string, rd io.Reader) error { | ||
| 172 | - openf, err := os.OpenFile(storePath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0666) | ||
| 173 | - if err != nil { | ||
| 174 | - return err | ||
| 175 | - } | ||
| 176 | - defer openf.Close() | ||
| 177 | - io.Copy(openf, rd) | ||
| 178 | - return nil | ||
| 179 | -} |
services/filestore/file_test.go
已删除
100644 → 0
| 1 | -package filestore |
-
请 注册 或 登录 后发表评论