115 lines
2.8 KiB
Go
115 lines
2.8 KiB
Go
package amazons3
|
|
|
|
import (
|
|
"bytes"
|
|
"server/utility/oss"
|
|
|
|
"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/gogf/gf/crypto/gmd5"
|
|
"github.com/gogf/gf/errors/gerror"
|
|
"github.com/gogf/gf/frame/g"
|
|
"github.com/gogf/gf/net/ghttp"
|
|
"github.com/gogf/gf/os/gfile"
|
|
"github.com/gogf/gf/os/glog"
|
|
"github.com/gogf/gf/os/gtime"
|
|
"github.com/gogf/gf/util/grand"
|
|
)
|
|
|
|
type AmazonS3Client struct{}
|
|
|
|
func (c *AmazonS3Client) initSess() (*session.Session, error) {
|
|
key := g.Config().GetString("aws.s3.key")
|
|
secret := g.Config().GetString("aws.s3.secret")
|
|
region := g.Config().GetString("aws.s3.region")
|
|
if len(key) == 0 || len(secret) == 0 {
|
|
return nil, gerror.New("aws s3 配置错误")
|
|
}
|
|
config := &aws.Config{
|
|
Region: aws.String(region),
|
|
Credentials: credentials.NewStaticCredentials(key, secret, ""),
|
|
}
|
|
sess := session.Must(session.NewSession(config))
|
|
return sess, nil
|
|
}
|
|
|
|
func (c *AmazonS3Client) Upload(file interface{}) (string, error) {
|
|
uploadFile, ok := file.(*ghttp.UploadFile)
|
|
if !ok {
|
|
return "", gerror.New("参数类型错误")
|
|
}
|
|
bucket := g.Config().GetString("aws.s3.bucket")
|
|
if len(bucket) == 0 {
|
|
glog.Warning("bucket为空:", bucket)
|
|
return "", gerror.New("aws s3 配置错误")
|
|
}
|
|
sess, err := c.initSess()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
uploader := s3manager.NewUploader(sess)
|
|
f, err := uploadFile.Open()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
defer f.Close()
|
|
body := make([]byte, uploadFile.Size)
|
|
_, err = f.Read(body)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
key, err := gmd5.Encrypt(uploadFile.Filename + gtime.Datetime() + grand.Digits(6))
|
|
ext := gfile.Ext(uploadFile.Filename)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
r, err := uploader.Upload(&s3manager.UploadInput{
|
|
Bucket: aws.String(bucket),
|
|
Key: aws.String(key + ext),
|
|
Body: bytes.NewReader(body),
|
|
})
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return r.Location, nil
|
|
}
|
|
|
|
func (c *AmazonS3Client) UploadLocalFile(path, name string) (string, error) {
|
|
bucket := g.Config().GetString("aws.s3.bucket")
|
|
if len(bucket) == 0 {
|
|
glog.Warning("bucket为空:", bucket)
|
|
return "", gerror.New("aws s3 配置错误")
|
|
}
|
|
sess, err := c.initSess()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
uploader := s3manager.NewUploader(sess)
|
|
f, err := gfile.Open(path)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
defer f.Close()
|
|
body := make([]byte, gfile.Size(path))
|
|
_, err = f.Read(body)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
r, err := uploader.Upload(&s3manager.UploadInput{
|
|
Bucket: aws.String(bucket),
|
|
Key: aws.String(name),
|
|
Body: bytes.NewReader(body),
|
|
})
|
|
if err != nil {
|
|
glog.Debug(err)
|
|
return "", err
|
|
}
|
|
return r.Location, nil
|
|
}
|
|
|
|
func init() {
|
|
oss.RegisterClient("amazon_s3", &AmazonS3Client{})
|
|
}
|