Skip to content

Commit 17c220d

Browse files
author
jojoliang
committed
update latest 0.7.23 stable
1 parent 982ae0a commit 17c220d

File tree

4 files changed

+32
-3
lines changed

4 files changed

+32
-3
lines changed

ci.go

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -206,14 +206,21 @@ func (s *CIService) GetVideoAuditingJob(ctx context.Context, jobid string) (*Get
206206
}
207207

208208
// ci put https://cloud.tencent.com/document/product/460/18147
209-
func (s *CIService) Put(ctx context.Context, name string, r io.Reader, opt *ObjectPutOptions) (*ImageProcessResult, *Response, error) {
209+
func (s *CIService) Put(ctx context.Context, name string, r io.Reader, uopt *ObjectPutOptions) (*ImageProcessResult, *Response, error) {
210210
if err := CheckReaderLen(r); err != nil {
211211
return nil, nil, err
212212
}
213+
opt := cloneObjectPutOptions(uopt)
213214
totalBytes, err := GetReaderLen(r)
214215
if err != nil && opt != nil && opt.Listener != nil {
215216
return nil, nil, err
216217
}
218+
if err == nil {
219+
// 与 go http 保持一致, 非bytes.Buffer/bytes.Reader/strings.Reader由用户指定ContentLength, 或使用 Chunk 上传
220+
if opt != nil && opt.ContentLength == 0 && IsLenReader(r) {
221+
opt.ContentLength = totalBytes
222+
}
223+
}
217224
reader := TeeReader(r, nil, totalBytes, nil)
218225
if s.client.Conf.EnableCRC {
219226
reader.writer = crc64.New(crc64.MakeTable(crc64.ECMA))

helper.go

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -159,6 +159,20 @@ func GetReaderLen(reader io.Reader) (length int64, err error) {
159159
return
160160
}
161161

162+
func IsLenReader(reader io.Reader) bool {
163+
switch reader.(type) {
164+
case *bytes.Buffer:
165+
return true
166+
case *bytes.Reader:
167+
return true
168+
case *strings.Reader:
169+
return true
170+
default:
171+
return false
172+
}
173+
return false
174+
}
175+
162176
func CheckReaderLen(reader io.Reader) error {
163177
nlen, err := GetReaderLen(reader)
164178
if err != nil || nlen < singleUploadMaxLength {

object.go

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -181,14 +181,21 @@ type ObjectPutOptions struct {
181181
// Put Object请求可以将一个文件(Oject)上传至指定Bucket。
182182
//
183183
// https://www.qcloud.com/document/product/436/7749
184-
func (s *ObjectService) Put(ctx context.Context, name string, r io.Reader, opt *ObjectPutOptions) (*Response, error) {
184+
func (s *ObjectService) Put(ctx context.Context, name string, r io.Reader, uopt *ObjectPutOptions) (*Response, error) {
185185
if err := CheckReaderLen(r); err != nil {
186186
return nil, err
187187
}
188+
opt := cloneObjectPutOptions(uopt)
188189
totalBytes, err := GetReaderLen(r)
189190
if err != nil && opt != nil && opt.Listener != nil {
190191
return nil, err
191192
}
193+
if err == nil {
194+
// 与 go http 保持一致, 非bytes.Buffer/bytes.Reader/strings.Reader由用户指定ContentLength, 或使用 Chunk 上传
195+
if opt != nil && opt.ContentLength == 0 && IsLenReader(r) {
196+
opt.ContentLength = totalBytes
197+
}
198+
}
192199
reader := TeeReader(r, nil, totalBytes, nil)
193200
if s.client.Conf.EnableCRC {
194201
reader.writer = crc64.New(crc64.MakeTable(crc64.ECMA))

object_part.go

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,8 @@ func (s *ObjectService) UploadPart(ctx context.Context, name, uploadID string, p
8181
}
8282
// 分块上传不支持 Chunk 上传
8383
if err == nil {
84-
if opt != nil && opt.ContentLength == 0 {
84+
// 与 go http 保持一致, 非bytes.Buffer/bytes.Reader/strings.Reader需用户指定ContentLength
85+
if opt != nil && opt.ContentLength == 0 && IsLenReader(r) {
8586
opt.ContentLength = totalBytes
8687
}
8788
}

0 commit comments

Comments
 (0)