9012069bd7
* chore: execute goimports to format the code Signed-off-by: promalert <promalert@outlook.com> * goimports -w . --------- Signed-off-by: promalert <promalert@outlook.com> Co-authored-by: Chris Lu <chris.lu@gmail.com>
52 lines
1.5 KiB
Go
52 lines
1.5 KiB
Go
package s3
|
|
|
|
import (
|
|
"fmt"
|
|
"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"
|
|
"github.com/seaweedfs/seaweedfs/weed/pb/remote_pb"
|
|
"github.com/seaweedfs/seaweedfs/weed/remote_storage"
|
|
"github.com/seaweedfs/seaweedfs/weed/util"
|
|
)
|
|
|
|
func init() {
|
|
remote_storage.RemoteStorageClientMakers["tencent"] = new(TencentRemoteStorageMaker)
|
|
}
|
|
|
|
type TencentRemoteStorageMaker struct{}
|
|
|
|
func (s TencentRemoteStorageMaker) HasBucket() bool {
|
|
return true
|
|
}
|
|
|
|
func (s TencentRemoteStorageMaker) Make(conf *remote_pb.RemoteConf) (remote_storage.RemoteStorageClient, error) {
|
|
client := &s3RemoteStorageClient{
|
|
supportTagging: true,
|
|
conf: conf,
|
|
}
|
|
accessKey := util.Nvl(conf.TencentSecretId, os.Getenv("COS_SECRETID"))
|
|
secretKey := util.Nvl(conf.TencentSecretKey, os.Getenv("COS_SECRETKEY"))
|
|
|
|
config := &aws.Config{
|
|
Endpoint: aws.String(conf.TencentEndpoint),
|
|
Region: aws.String("us-west-2"),
|
|
S3ForcePathStyle: aws.Bool(true),
|
|
S3DisableContentMD5Validation: aws.Bool(true),
|
|
}
|
|
if accessKey != "" && secretKey != "" {
|
|
config.Credentials = credentials.NewStaticCredentials(accessKey, secretKey, "")
|
|
}
|
|
|
|
sess, err := session.NewSession(config)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("create tencent session: %w", err)
|
|
}
|
|
sess.Handlers.Build.PushFront(skipSha256PayloadSigning)
|
|
client.conn = s3.New(sess)
|
|
return client, nil
|
|
}
|