revoke(oss): downgrade to SDK v1, switch sign method to v4

pull/2978/head
Darren Yu 12 months ago
parent e29237d593
commit 56cf990645
No known key found for this signature in database
GPG Key ID: 2D69AA5646405984

@ -6,7 +6,7 @@ require (
entgo.io/ent v0.13.0 entgo.io/ent v0.13.0
github.com/Masterminds/semver/v3 v3.3.1 github.com/Masterminds/semver/v3 v3.3.1
github.com/abslant/gzip v0.0.9 github.com/abslant/gzip v0.0.9
github.com/aliyun/alibabacloud-oss-go-sdk-v2 v1.3.0 github.com/aliyun/aliyun-oss-go-sdk v3.0.2+incompatible
github.com/aws/aws-sdk-go v1.31.5 github.com/aws/aws-sdk-go v1.31.5
github.com/bodgit/sevenzip v1.6.0 github.com/bodgit/sevenzip v1.6.0
github.com/cloudflare/cfssl v1.6.1 github.com/cloudflare/cfssl v1.6.1

@ -100,8 +100,8 @@ github.com/alecthomas/template v0.0.0-20190718012654-fb15b899a751/go.mod h1:LOuy
github.com/alecthomas/units v0.0.0-20151022065526-2efee857e7cf/go.mod h1:ybxpYRFXyAe+OPACYpWeL0wqObRcbAqCMya13uyzqw0= github.com/alecthomas/units v0.0.0-20151022065526-2efee857e7cf/go.mod h1:ybxpYRFXyAe+OPACYpWeL0wqObRcbAqCMya13uyzqw0=
github.com/alecthomas/units v0.0.0-20190717042225-c3de453c63f4/go.mod h1:ybxpYRFXyAe+OPACYpWeL0wqObRcbAqCMya13uyzqw0= github.com/alecthomas/units v0.0.0-20190717042225-c3de453c63f4/go.mod h1:ybxpYRFXyAe+OPACYpWeL0wqObRcbAqCMya13uyzqw0=
github.com/alecthomas/units v0.0.0-20190924025748-f65c72e2690d/go.mod h1:rBZYJk541a8SKzHPHnH3zbiI+7dagKZ0cgpgrD7Fyho= github.com/alecthomas/units v0.0.0-20190924025748-f65c72e2690d/go.mod h1:rBZYJk541a8SKzHPHnH3zbiI+7dagKZ0cgpgrD7Fyho=
github.com/aliyun/alibabacloud-oss-go-sdk-v2 v1.3.0 h1:wQlqotpyjYPjJz+Noh5bRu7Snmydk8SKC5Z6u1CR20Y= github.com/aliyun/aliyun-oss-go-sdk v3.0.2+incompatible h1:8psS8a+wKfiLt1iVDX79F7Y6wUM49Lcha2FMXt4UM8g=
github.com/aliyun/alibabacloud-oss-go-sdk-v2 v1.3.0/go.mod h1:FTzydeQVmR24FI0D6XWUOMKckjXehM/jgMn1xC+DA9M= github.com/aliyun/aliyun-oss-go-sdk v3.0.2+incompatible/go.mod h1:T/Aws4fEfogEE9v+HPhhw+CntffsBHJ8nXQCwKr0/g8=
github.com/andybalholm/brotli v1.1.2-0.20250424173009-453214e765f3 h1:8PmGpDEZl9yDpcdEr6Odf23feCxK3LNUNMxjXg41pZQ= github.com/andybalholm/brotli v1.1.2-0.20250424173009-453214e765f3 h1:8PmGpDEZl9yDpcdEr6Odf23feCxK3LNUNMxjXg41pZQ=
github.com/andybalholm/brotli v1.1.2-0.20250424173009-453214e765f3/go.mod h1:05ib4cKhjx3OQYUY22hTVd34Bc8upXjOLL2rKwwZBoA= github.com/andybalholm/brotli v1.1.2-0.20250424173009-453214e765f3/go.mod h1:05ib4cKhjx3OQYUY22hTVd34Bc8upXjOLL2rKwwZBoA=
github.com/anmitsu/go-shlex v0.0.0-20161002113705-648efa622239/go.mod h1:2FmKhYUyUczH0OGQWaF5ceTx0UBShxjsH6f8oGKYe2c= github.com/anmitsu/go-shlex v0.0.0-20161002113705-648efa622239/go.mod h1:2FmKhYUyUczH0OGQWaF5ceTx0UBShxjsH6f8oGKYe2c=

@ -11,7 +11,7 @@ import (
"strings" "strings"
"time" "time"
"github.com/aliyun/alibabacloud-oss-go-sdk-v2/oss" "github.com/aliyun/aliyun-oss-go-sdk/oss"
"github.com/cloudreve/Cloudreve/v4/pkg/filemanager/driver" "github.com/cloudreve/Cloudreve/v4/pkg/filemanager/driver"
"github.com/cloudreve/Cloudreve/v4/pkg/mediameta" "github.com/cloudreve/Cloudreve/v4/pkg/mediameta"
"github.com/cloudreve/Cloudreve/v4/pkg/request" "github.com/cloudreve/Cloudreve/v4/pkg/request"
@ -266,14 +266,13 @@ func (handler *Driver) extractImageMeta(ctx context.Context, path string) ([]dri
// extractMediaInfo Sends API calls to OSS IMM service to extract media info. // extractMediaInfo Sends API calls to OSS IMM service to extract media info.
func (handler *Driver) extractMediaInfo(ctx context.Context, path string, category string, forceSign bool) (string, error) { func (handler *Driver) extractMediaInfo(ctx context.Context, path string, category string, forceSign bool) (string, error) {
mediaOption := []oss.Option{oss.Process(category)}
mediaInfoExpire := time.Now().Add(mediaInfoTTL) mediaInfoExpire := time.Now().Add(mediaInfoTTL)
thumbURL, err := handler.signSourceURL( thumbURL, err := handler.signSourceURL(
ctx, ctx,
path, path,
&mediaInfoExpire, &mediaInfoExpire,
&oss.GetObjectRequest{ mediaOption,
Process: oss.Ptr(category),
},
forceSign, forceSign,
) )
if err != nil { if err != nil {

@ -15,8 +15,7 @@ import (
"strings" "strings"
"time" "time"
"github.com/aliyun/alibabacloud-oss-go-sdk-v2/oss" "github.com/aliyun/aliyun-oss-go-sdk/oss"
"github.com/aliyun/alibabacloud-oss-go-sdk-v2/oss/credentials"
"github.com/cloudreve/Cloudreve/v4/ent" "github.com/cloudreve/Cloudreve/v4/ent"
"github.com/cloudreve/Cloudreve/v4/inventory/types" "github.com/cloudreve/Cloudreve/v4/inventory/types"
"github.com/cloudreve/Cloudreve/v4/pkg/boolset" "github.com/cloudreve/Cloudreve/v4/pkg/boolset"
@ -53,6 +52,7 @@ type Driver struct {
policy *ent.StoragePolicy policy *ent.StoragePolicy
client *oss.Client client *oss.Client
bucket *oss.Bucket
settings setting.Provider settings setting.Provider
l logging.Logger l logging.Logger
config conf.ConfigProvider config conf.ConfigProvider
@ -65,8 +65,12 @@ type Driver struct {
type key int type key int
const ( const (
chunkRetrySleep = time.Duration(5) * time.Second chunkRetrySleep = time.Duration(5) * time.Second
maxDeleteBatch = 1000 uploadIdParam = "uploadId"
partNumberParam = "partNumber"
callbackParam = "callback"
completeAllHeader = "x-oss-complete-all"
maxDeleteBatch = 1000
// MultiPartUploadThreshold 服务端使用分片上传的阈值 // MultiPartUploadThreshold 服务端使用分片上传的阈值
MultiPartUploadThreshold int64 = 5 * (1 << 30) // 5GB MultiPartUploadThreshold int64 = 5 * (1 << 30) // 5GB
@ -98,27 +102,21 @@ func New(ctx context.Context, policy *ent.StoragePolicy, settings setting.Provid
// CORS 创建跨域策略 // CORS 创建跨域策略
func (handler *Driver) CORS() error { func (handler *Driver) CORS() error {
_, err := handler.client.PutBucketCors(context.Background(), &oss.PutBucketCorsRequest{ return handler.client.SetBucketCORS(handler.policy.BucketName, []oss.CORSRule{
Bucket: &handler.policy.BucketName, {
CORSConfiguration: &oss.CORSConfiguration{ AllowedOrigin: []string{"*"},
CORSRules: []oss.CORSRule{ AllowedMethod: []string{
{ "GET",
AllowedOrigins: []string{"*"}, "POST",
AllowedMethods: []string{ "PUT",
"GET", "DELETE",
"POST", "HEAD",
"PUT",
"DELETE",
"HEAD",
},
ExposeHeaders: []string{},
AllowedHeaders: []string{"*"},
MaxAgeSeconds: oss.Ptr(int64(3600)),
},
}, },
}}) ExposeHeader: []string{},
AllowedHeader: []string{"*"},
return err MaxAgeSeconds: 3600,
},
})
} }
// InitOSSClient 初始化OSS鉴权客户端 // InitOSSClient 初始化OSS鉴权客户端
@ -127,28 +125,38 @@ func (handler *Driver) InitOSSClient(forceUsePublicEndpoint bool) error {
return errors.New("empty policy") return errors.New("empty policy")
} }
opt := make([]oss.ClientOption, 0)
// 决定是否使用内网 Endpoint // 决定是否使用内网 Endpoint
endpoint := handler.policy.Server endpoint := handler.policy.Server
useCname := false
if handler.policy.Settings.ServerSideEndpoint != "" && !forceUsePublicEndpoint { if handler.policy.Settings.ServerSideEndpoint != "" && !forceUsePublicEndpoint {
endpoint = handler.policy.Settings.ServerSideEndpoint endpoint = handler.policy.Settings.ServerSideEndpoint
} else if handler.policy.Settings.UseCname { } else if handler.policy.Settings.UseCname {
useCname = true opt = append(opt, oss.UseCname(true))
} }
if !strings.HasPrefix(endpoint, "http://") && !strings.HasPrefix(endpoint, "https://") { if !strings.HasPrefix(endpoint, "http://") && !strings.HasPrefix(endpoint, "https://") {
endpoint = "https://" + endpoint endpoint = "https://" + endpoint
} }
cfg := oss.LoadDefaultConfig(). // 设置签名版本 & 存储桶 region
WithCredentialsProvider(credentials.NewStaticCredentialsProvider(handler.policy.AccessKey, handler.policy.SecretKey, "")). opt = append(opt, oss.AuthVersion(oss.AuthV4))
WithEndpoint(endpoint). opt = append(opt, oss.Region(handler.policy.Settings.Region))
WithRegion(handler.policy.Settings.Region).
WithUseCName(useCname)
// 初始化客户端 // 初始化客户端
client := oss.NewClient(cfg) client, err := oss.New(endpoint, handler.policy.AccessKey, handler.policy.SecretKey, opt...)
if err != nil {
return err
}
handler.client = client handler.client = client
// 初始化存储桶
bucket, err := client.Bucket(handler.policy.BucketName)
if err != nil {
return err
}
handler.bucket = bucket
return nil return nil
} }
@ -162,40 +170,38 @@ func (handler *Driver) List(ctx context.Context, base string, onProgress driver.
var ( var (
delimiter string delimiter string
marker string
objects []oss.ObjectProperties objects []oss.ObjectProperties
commons []oss.CommonPrefix commons []string
) )
if !recursive { if !recursive {
delimiter = "/" delimiter = "/"
} }
p := handler.client.NewListObjectsPaginator(&oss.ListObjectsRequest{ for {
Bucket: &handler.policy.BucketName, subRes, err := handler.bucket.ListObjects(oss.Marker(marker), oss.Prefix(base),
Prefix: &base, oss.MaxKeys(1000), oss.Delimiter(delimiter))
MaxKeys: 1000,
Delimiter: &delimiter,
})
for p.HasNext() {
page, err := p.NextPage(ctx)
if err != nil { if err != nil {
return nil, err return nil, err
} }
objects = append(objects, subRes.Objects...)
objects = append(objects, page.Contents...) commons = append(commons, subRes.CommonPrefixes...)
commons = append(commons, page.CommonPrefixes...) marker = subRes.NextMarker
if marker == "" {
break
}
} }
// 处理列取结果 // 处理列取结果
res := make([]fs.PhysicalObject, 0, len(objects)+len(commons)) res := make([]fs.PhysicalObject, 0, len(objects)+len(commons))
// 处理目录 // 处理目录
for _, object := range commons { for _, object := range commons {
rel, err := filepath.Rel(base, *object.Prefix) rel, err := filepath.Rel(base, object)
if err != nil { if err != nil {
continue continue
} }
res = append(res, fs.PhysicalObject{ res = append(res, fs.PhysicalObject{
Name: path.Base(*object.Prefix), Name: path.Base(object),
RelativePath: filepath.ToSlash(rel), RelativePath: filepath.ToSlash(rel),
Size: 0, Size: 0,
IsDir: true, IsDir: true,
@ -206,17 +212,17 @@ func (handler *Driver) List(ctx context.Context, base string, onProgress driver.
// 处理文件 // 处理文件
for _, object := range objects { for _, object := range objects {
rel, err := filepath.Rel(base, *object.Key) rel, err := filepath.Rel(base, object.Key)
if err != nil { if err != nil {
continue continue
} }
res = append(res, fs.PhysicalObject{ res = append(res, fs.PhysicalObject{
Name: path.Base(*object.Key), Name: path.Base(object.Key),
Source: *object.Key, Source: object.Key,
RelativePath: filepath.ToSlash(rel), RelativePath: filepath.ToSlash(rel),
Size: object.Size, Size: object.Size,
IsDir: false, IsDir: false,
LastModify: *object.LastModified, LastModify: object.LastModified,
}) })
} }
onProgress(len(res)) onProgress(len(res))
@ -243,34 +249,25 @@ func (handler *Driver) Put(ctx context.Context, file *fs.UploadRequest) error {
// 是否允许覆盖 // 是否允许覆盖
overwrite := file.Mode&fs.ModeOverwrite == fs.ModeOverwrite overwrite := file.Mode&fs.ModeOverwrite == fs.ModeOverwrite
forbidOverwrite := oss.Ptr(strconv.FormatBool(!overwrite)) options := []oss.Option{
exipires := oss.Ptr(time.Now().Add(credentialTTL * time.Second).Format(time.RFC3339)) oss.WithContext(ctx),
oss.Expires(time.Now().Add(credentialTTL * time.Second)),
oss.ForbidOverWrite(!overwrite),
oss.ContentType(mimeType),
}
// 小文件直接上传 // 小文件直接上传
if file.Props.Size < MultiPartUploadThreshold { if file.Props.Size < MultiPartUploadThreshold {
_, err := handler.client.PutObject(ctx, &oss.PutObjectRequest{ return handler.bucket.PutObject(file.Props.SavePath, file, options...)
Bucket: &handler.policy.BucketName,
Key: &file.Props.SavePath,
Body: file,
ForbidOverwrite: forbidOverwrite,
ContentType: oss.Ptr(mimeType),
})
return err
} }
// 超过阈值时使用分片上传 // 超过阈值时使用分片上传
imur, err := handler.client.InitiateMultipartUpload(ctx, &oss.InitiateMultipartUploadRequest{ imur, err := handler.bucket.InitiateMultipartUpload(file.Props.SavePath, options...)
Bucket: &handler.policy.BucketName,
Key: &file.Props.SavePath,
ContentType: oss.Ptr(mimeType),
ForbidOverwrite: forbidOverwrite,
Expires: exipires,
})
if err != nil { if err != nil {
return fmt.Errorf("failed to initiate multipart upload: %w", err) return fmt.Errorf("failed to initiate multipart upload: %w", err)
} }
parts := make([]*oss.UploadPartResult, 0) parts := make([]oss.UploadPart, 0)
chunks := chunk.NewChunkGroup(file, handler.chunkSize, &backoff.ConstantBackoff{ chunks := chunk.NewChunkGroup(file, handler.chunkSize, &backoff.ConstantBackoff{
Max: handler.settings.ChunkRetryLimit(ctx), Max: handler.settings.ChunkRetryLimit(ctx),
@ -278,13 +275,7 @@ func (handler *Driver) Put(ctx context.Context, file *fs.UploadRequest) error {
}, handler.settings.UseChunkBuffer(ctx), handler.l, handler.settings.TempPath(ctx)) }, handler.settings.UseChunkBuffer(ctx), handler.l, handler.settings.TempPath(ctx))
uploadFunc := func(current *chunk.ChunkGroup, content io.Reader) error { uploadFunc := func(current *chunk.ChunkGroup, content io.Reader) error {
part, err := handler.client.UploadPart(ctx, &oss.UploadPartRequest{ part, err := handler.bucket.UploadPart(imur, content, current.Length(), current.Index()+1, oss.WithContext(ctx))
Bucket: &handler.policy.BucketName,
Key: &file.Props.SavePath,
UploadId: imur.UploadId,
PartNumber: int32(current.Index() + 1),
Body: content,
})
if err == nil { if err == nil {
parts = append(parts, part) parts = append(parts, part)
} }
@ -293,27 +284,14 @@ func (handler *Driver) Put(ctx context.Context, file *fs.UploadRequest) error {
for chunks.Next() { for chunks.Next() {
if err := chunks.Process(uploadFunc); err != nil { if err := chunks.Process(uploadFunc); err != nil {
handler.cancelUpload(*imur) handler.cancelUpload(imur)
return fmt.Errorf("failed to upload chunk #%d: %w", chunks.Index(), err) return fmt.Errorf("failed to upload chunk #%d: %w", chunks.Index(), err)
} }
} }
_, err = handler.client.CompleteMultipartUpload(ctx, &oss.CompleteMultipartUploadRequest{ _, err = handler.bucket.CompleteMultipartUpload(imur, parts, oss.ForbidOverWrite(!overwrite), oss.WithContext(ctx))
Bucket: &handler.policy.BucketName,
Key: imur.Key,
UploadId: imur.UploadId,
CompleteMultipartUpload: &oss.CompleteMultipartUpload{
Parts: lo.Map(parts, func(part *oss.UploadPartResult, i int) oss.UploadPart {
return oss.UploadPart{
PartNumber: int32(i + 1),
ETag: part.ETag,
}
}),
},
ForbidOverwrite: oss.Ptr(strconv.FormatBool(!overwrite)),
})
if err != nil { if err != nil {
handler.cancelUpload(*imur) handler.cancelUpload(imur)
} }
return err return err
@ -328,12 +306,7 @@ func (handler *Driver) Delete(ctx context.Context, files ...string) ([]string, e
for index, group := range groups { for index, group := range groups {
handler.l.Debug("Process delete group #%d: %v", index, group) handler.l.Debug("Process delete group #%d: %v", index, group)
// 删除文件 // 删除文件
delRes, err := handler.client.DeleteMultipleObjects(ctx, &oss.DeleteMultipleObjectsRequest{ delRes, err := handler.bucket.DeleteObjects(group)
Bucket: &handler.policy.BucketName,
Objects: lo.Map(group, func(v string, i int) oss.DeleteObject {
return oss.DeleteObject{Key: &v}
}),
})
if err != nil { if err != nil {
failed = append(failed, group...) failed = append(failed, group...)
lastError = err lastError = err
@ -341,14 +314,7 @@ func (handler *Driver) Delete(ctx context.Context, files ...string) ([]string, e
} }
// 统计未删除的文件 // 统计未删除的文件
failed = append( failed = append(failed, util.SliceDifference(files, delRes.DeletedObjects)...)
failed,
util.SliceDifference(files,
lo.Map(delRes.DeletedObjects, func(v oss.DeletedInfo, i int) string {
return *v.Key
}),
)...,
)
} }
if len(failed) > 0 && lastError == nil { if len(failed) > 0 && lastError == nil {
@ -381,14 +347,12 @@ func (handler *Driver) Thumb(ctx context.Context, expire *time.Time, ext string,
thumbParam += fmt.Sprintf("/format,%s", enco.Format) thumbParam += fmt.Sprintf("/format,%s", enco.Format)
} }
req := &oss.GetObjectRequest{ thumbOption := []oss.Option{oss.Process(thumbParam)}
Process: oss.Ptr(thumbParam),
}
thumbURL, err := handler.signSourceURL( thumbURL, err := handler.signSourceURL(
ctx, ctx,
e.Source(), e.Source(),
expire, expire,
req, thumbOption,
false, false,
) )
if err != nil { if err != nil {
@ -410,11 +374,11 @@ func (handler *Driver) Source(ctx context.Context, e fs.Entity, args *driver.Get
} }
// 添加各项设置 // 添加各项设置
req := &oss.GetObjectRequest{} var signOptions = make([]oss.Option, 0, 2)
if args.IsDownload { if args.IsDownload {
encodedFilename := url.PathEscape(args.DisplayName) encodedFilename := url.PathEscape(args.DisplayName)
req.ResponseContentDisposition = oss.Ptr(fmt.Sprintf(`attachment; filename="%s"; filename*=UTF-8''%s`, signOptions = append(signOptions, oss.ResponseContentDisposition(fmt.Sprintf(`attachment; filename="%s"; filename*=UTF-8''%s`,
encodedFilename, encodedFilename)) encodedFilename, encodedFilename)))
} }
if args.Speed > 0 { if args.Speed > 0 {
// Byte 转换为 bit // Byte 转换为 bit
@ -427,33 +391,25 @@ func (handler *Driver) Source(ctx context.Context, e fs.Entity, args *driver.Get
if args.Speed > 838860800 { if args.Speed > 838860800 {
args.Speed = 838860800 args.Speed = 838860800
} }
req.TrafficLimit = args.Speed signOptions = append(signOptions, oss.TrafficLimitParam(args.Speed))
} }
return handler.signSourceURL(ctx, e.Source(), args.Expire, req, false) return handler.signSourceURL(ctx, e.Source(), args.Expire, signOptions, false)
} }
func (handler *Driver) signSourceURL(ctx context.Context, path string, expire *time.Time, req *oss.GetObjectRequest, forceSign bool) (string, error) { func (handler *Driver) signSourceURL(ctx context.Context, path string, expire *time.Time, options []oss.Option, forceSign bool) (string, error) {
ttl := time.Duration(24) * time.Hour * 365 * 20 ttl := int64(86400 * 7)
if expire != nil { if expire != nil {
ttl = time.Until(*expire) ttl = int64(time.Until(*expire).Seconds())
}
if req == nil {
req = &oss.GetObjectRequest{}
} }
req.Bucket = &handler.policy.BucketName signedURL, err := handler.bucket.SignURL(path, oss.HTTPGet, ttl, options...)
req.Key = &path
// signedURL, err := handler.client.Presign(path, oss.HTTPGet, ttl, options...)
result, err := handler.client.Presign(ctx, req, oss.PresignExpires(ttl))
if err != nil { if err != nil {
return "", err return "", err
} }
// 将最终生成的签名URL域名换成用户自定义的加速域名(如果有) // 将最终生成的签名URL域名换成用户自定义的加速域名(如果有)
finalURL, err := url.Parse(result.URL) finalURL, err := url.Parse(signedURL)
if err != nil { if err != nil {
return "", err return "", err
} }
@ -502,41 +458,34 @@ func (handler *Driver) Token(ctx context.Context, uploadSession *fs.UploadSessio
} }
// 初始化分片上传 // 初始化分片上传
imur, err := handler.client.InitiateMultipartUpload(ctx, &oss.InitiateMultipartUploadRequest{ options := []oss.Option{
Bucket: &handler.policy.BucketName, oss.WithContext(ctx),
Key: &file.Props.SavePath, oss.Expires(uploadSession.Props.ExpireAt),
ContentType: oss.Ptr(mimeType), oss.ForbidOverWrite(true),
ForbidOverwrite: oss.Ptr(strconv.FormatBool(true)), oss.ContentType(mimeType),
Expires: oss.Ptr(uploadSession.Props.ExpireAt.Format(time.RFC3339)), }
}) imur, err := handler.bucket.InitiateMultipartUpload(file.Props.SavePath, options...)
if err != nil { if err != nil {
return nil, fmt.Errorf("failed to initialize multipart upload: %w", err) return nil, fmt.Errorf("failed to initialize multipart upload: %w", err)
} }
uploadSession.UploadID = *imur.UploadId uploadSession.UploadID = imur.UploadID
// 为每个分片签名上传 URL // 为每个分片签名上传 URL
chunks := chunk.NewChunkGroup(file, handler.chunkSize, &backoff.ConstantBackoff{}, false, handler.l, "") chunks := chunk.NewChunkGroup(file, handler.chunkSize, &backoff.ConstantBackoff{}, false, handler.l, "")
urls := make([]string, chunks.Num()) urls := make([]string, chunks.Num())
ttl := time.Until(uploadSession.Props.ExpireAt) ttl := int64(time.Until(uploadSession.Props.ExpireAt).Seconds())
for chunks.Next() { for chunks.Next() {
err := chunks.Process(func(c *chunk.ChunkGroup, chunk io.Reader) error { err := chunks.Process(func(c *chunk.ChunkGroup, chunk io.Reader) error {
signedURL, err := handler.client.Presign(ctx, &oss.UploadPartRequest{ signedURL, err := handler.bucket.SignURL(file.Props.SavePath, oss.HTTPPut,
Bucket: &handler.policy.BucketName, ttl,
Key: &file.Props.SavePath, oss.AddParam(partNumberParam, strconv.Itoa(c.Index()+1)),
UploadId: imur.UploadId, oss.AddParam(uploadIdParam, imur.UploadID),
PartNumber: int32(c.Index() + 1), oss.ContentType("application/octet-stream"))
Body: chunk,
RequestCommon: oss.RequestCommon{
Headers: map[string]string{
"Content-Type": "application/octet-stream",
},
},
}, oss.PresignExpires(ttl))
if err != nil { if err != nil {
return err return err
} }
urls[c.Index()] = signedURL.URL urls[c.Index()] = signedURL
return nil return nil
}) })
if err != nil { if err != nil {
@ -545,43 +494,29 @@ func (handler *Driver) Token(ctx context.Context, uploadSession *fs.UploadSessio
} }
// 签名完成分片上传的URL // 签名完成分片上传的URL
completeURL, err := handler.client.Presign(ctx, &oss.CompleteMultipartUploadRequest{ completeURL, err := handler.bucket.SignURL(file.Props.SavePath, oss.HTTPPost, ttl,
Bucket: &handler.policy.BucketName, oss.ContentType("application/octet-stream"),
Key: &file.Props.SavePath, oss.AddParam(uploadIdParam, imur.UploadID),
UploadId: imur.UploadId, oss.Expires(time.Now().Add(time.Duration(ttl)*time.Second)),
RequestCommon: oss.RequestCommon{ oss.SetHeader(completeAllHeader, "yes"),
Parameters: map[string]string{ oss.ForbidOverWrite(true),
"callback": callbackPolicyEncoded, oss.AddParam(callbackParam, callbackPolicyEncoded))
},
Headers: map[string]string{
"Content-Type": "application/octet-stream",
"x-oss-complete-all": "yes",
"x-oss-forbid-overwrite": "true",
},
},
}, oss.PresignExpires(ttl))
if err != nil { if err != nil {
return nil, err return nil, err
} }
return &fs.UploadCredential{ return &fs.UploadCredential{
UploadID: *imur.UploadId, UploadID: imur.UploadID,
UploadURLs: urls, UploadURLs: urls,
CompleteURL: completeURL.URL, CompleteURL: completeURL,
SessionID: uploadSession.Props.UploadSessionID, SessionID: uploadSession.Props.UploadSessionID,
ChunkSize: handler.chunkSize, ChunkSize: handler.chunkSize,
Callback: callbackPolicyEncoded,
}, nil }, nil
} }
// 取消上传凭证 // 取消上传凭证
func (handler *Driver) CancelToken(ctx context.Context, uploadSession *fs.UploadSession) error { func (handler *Driver) CancelToken(ctx context.Context, uploadSession *fs.UploadSession) error {
_, err := handler.client.AbortMultipartUpload(ctx, &oss.AbortMultipartUploadRequest{ return handler.bucket.AbortMultipartUpload(oss.InitiateMultipartUploadResult{UploadID: uploadSession.UploadID, Key: uploadSession.Props.SavePath}, oss.WithContext(ctx))
Bucket: &handler.policy.BucketName,
Key: &uploadSession.Props.SavePath,
UploadId: &uploadSession.UploadID,
})
return err
} }
func (handler *Driver) CompleteUpload(ctx context.Context, session *fs.UploadSession) error { func (handler *Driver) CompleteUpload(ctx context.Context, session *fs.UploadSession) error {
@ -625,11 +560,7 @@ func (handler *Driver) LocalPath(ctx context.Context, path string) string {
} }
func (handler *Driver) cancelUpload(imur oss.InitiateMultipartUploadResult) { func (handler *Driver) cancelUpload(imur oss.InitiateMultipartUploadResult) {
if _, err := handler.client.AbortMultipartUpload(context.Background(), &oss.AbortMultipartUploadRequest{ if err := handler.bucket.AbortMultipartUpload(imur); err != nil {
Bucket: &handler.policy.BucketName,
Key: imur.Key,
UploadId: imur.UploadId,
}); err != nil {
handler.l.Warning("failed to abort multipart upload: %s", err) handler.l.Warning("failed to abort multipart upload: %s", err)
} }
} }

Loading…
Cancel
Save