mirror of https://github.com/rocboss/paopao-ce
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
68 lines
1.6 KiB
68 lines
1.6 KiB
package dao
|
|
|
|
import (
|
|
"io"
|
|
"net/url"
|
|
"strings"
|
|
|
|
"github.com/aliyun/aliyun-oss-go-sdk/oss"
|
|
"github.com/rocboss/paopao-ce/internal/conf"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
func newAliossServent() *aliossServant {
|
|
client, err := oss.New(conf.AliOSSSetting.Endpoint, conf.AliOSSSetting.AccessKeyID, conf.AliOSSSetting.AccessKeySecret)
|
|
if err != nil {
|
|
logrus.Fatalf("alioss.New err: %v", err)
|
|
}
|
|
|
|
bucket, err := client.Bucket(conf.AliOSSSetting.Bucket)
|
|
if err != nil {
|
|
logrus.Fatalf("client.Bucket err: %v", err)
|
|
}
|
|
|
|
return &aliossServant{
|
|
bucket: bucket,
|
|
domain: getOssDomain(),
|
|
}
|
|
}
|
|
|
|
func (s *aliossServant) PutObject(objectKey string, reader io.Reader, objectSize int64, contentType string) (string, error) {
|
|
err := s.bucket.PutObject(objectKey, reader, oss.ContentLength(objectSize), oss.ContentType(contentType))
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return s.domain + objectKey, nil
|
|
}
|
|
|
|
func (s *aliossServant) SignURL(objectKey string, expiredInSec int64) (string, error) {
|
|
signedURL, err := s.bucket.SignURL(objectKey, oss.HTTPGet, expiredInSec)
|
|
if err != nil {
|
|
logrus.Errorf("client.SignURL err: %v", err)
|
|
return "", err
|
|
}
|
|
|
|
ur, err := url.Parse(signedURL)
|
|
if err != nil {
|
|
logrus.Errorf("url.Parse err: %v", err)
|
|
return "", err
|
|
}
|
|
|
|
epath, err := url.PathUnescape(ur.Path)
|
|
if err != nil {
|
|
logrus.Errorf("url.PathUnescape err: %v", err)
|
|
return "", err
|
|
}
|
|
|
|
ur.Path, ur.RawPath = epath, epath
|
|
return ur.String(), nil
|
|
}
|
|
|
|
func (s *aliossServant) ObjectURL(objetKey string) string {
|
|
return s.domain + objetKey
|
|
}
|
|
|
|
func (s *aliossServant) ObjectKey(objectUrl string) string {
|
|
return strings.Replace(objectUrl, s.domain, "", -1)
|
|
}
|