/
file.go
80 lines (62 loc) · 1.88 KB
/
file.go
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
package mongo
import (
"context"
"fmt"
"io"
"go.mongodb.org/mongo-driver/mongo/gridfs"
"go.mongodb.org/mongo-driver/mongo/options"
"github.com/infraboard/mcube/v2/exception"
"github.com/infraboard/mcube/v2/ioc/apps/oss"
)
func (s *service) UploadFile(ctx context.Context, req *oss.UploadFileRequest) error {
s.log.Debug().Msgf("bucket name: %s, db file name: %s", req.BucketName, req.FileName)
if err := req.Validate(); err != nil {
return exception.NewBadRequest("valiate upload file request error, %s", err)
}
bucket, err := s.getBucket(req.BucketName)
if err != nil {
return err
}
opts := options.GridFSUpload()
opts.Metadata = req.Meta()
// 清除已有文件
bucket.Delete(req.FileName)
// 上传新文件
uploadStream, err := bucket.OpenUploadStreamWithID(req.FileName, req.FileName, opts)
if err != nil {
return err
}
defer uploadStream.Close()
fileSize, err := io.Copy(uploadStream, req.ReadCloser())
if err != nil {
return err
}
s.log.Debug().Msgf("Write file %s to DB was successful. File size: %d M", req.FileName, fileSize/1024/1024)
return nil
}
func (s *service) Download(ctx context.Context, req *oss.DownloadFileRequest) error {
if err := req.Validate(); err != nil {
return exception.NewBadRequest("valiate upload file request error, %s", err)
}
bucket, err := s.getBucket(req.BucketName)
if err != nil {
return err
}
s.log.Debug().Msgf("start download file: %s ...", req.FileID)
// 下载文件
size, err := bucket.DownloadToStream(req.FileID, req.Writer())
if err != nil {
return err
}
s.log.Debug().Msgf("download file: %s complete, size: %d", req.FileID, size)
return nil
}
func (s *service) getBucket(name string) (*gridfs.Bucket, error) {
opts := options.GridFSBucket()
opts.SetName(name)
bucket, err := gridfs.NewBucket(s.db, opts)
if err != nil {
return nil, fmt.Errorf("new bucket error, %s", err)
}
return bucket, nil
}