-
Notifications
You must be signed in to change notification settings - Fork 0
/
utils.go
80 lines (61 loc) · 1.71 KB
/
utils.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 s3
import (
"fmt"
"net/url"
"os"
"github.com/aws/aws-sdk-go/aws"
"github.com/aws/aws-sdk-go/aws/credentials"
"github.com/aws/aws-sdk-go/aws/session"
"github.com/aws/aws-sdk-go/service/s3/s3manager"
"github.com/rosberry/storage/common"
)
func (s *S3Storage) getSession() *session.Session {
return session.Must(session.NewSession(&aws.Config{
Region: aws.String(s.cfg.Region),
Credentials: credentials.NewStaticCredentials(s.cfg.AccessKeyID, s.cfg.SecretAccessKey, ""),
}))
}
func (s *S3Storage) storeByPath(filePath string, path string) (cLink string, err error) {
err = s.storeByInternalPath(filePath, common.PathToInternalPath(s.cfg.Prefix, path))
if err != nil {
return "", fmt.Errorf("failed to store %s: %w", path, err)
}
cLink = s.GetCLink(path)
return
}
func (s *S3Storage) storeByInternalPath(filePath, internalPath string) (err error) {
uploader := s3manager.NewUploader(s.getSession())
f, _ := os.Open(filePath)
defer f.Close()
mimetype := common.GetFileContentType(f)
_, err = uploader.Upload(&s3manager.UploadInput{
Bucket: aws.String(s.cfg.BucketName),
Key: aws.String(internalPath),
Body: f,
ContentType: aws.String(mimetype),
})
if err != nil {
return fmt.Errorf("failed to upload file: %w", err)
}
return nil
}
func (s *S3Storage) prepareURL(cLink string) (u *url.URL, err error) {
var uc *url.URL
uc, err = url.Parse(cLink)
if err != nil {
return
}
if uc.Scheme != s.cfg.StorageKey {
err = ErrStorageKeyNotMatch
return
}
u = &url.URL{}
u.Scheme = s.scheme
u.Host = fmt.Sprintf(S3HostTemplate, s.cfg.BucketName)
u.Path = uc.Path
if uc.Opaque != "" {
u.Path = "/" + uc.Opaque
}
u.Path = s.cfg.Prefix + u.Path
return
}