完善功能

utility/oss/amazon_s3/amazon-s3.go

@@ -0,0 +1,114 @@
+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{})
+}