浏览代码

fix download/upload endless loop when file size == 0

yubin.byb 9 年之前
父节点
当前提交
eea21c7e41
共有 2 个文件被更改,包括 4 次插入4 次删除
  1. 2 2
      oss/download.go
  2. 2 2
      oss/upload.go

+ 2 - 2
oss/download.go

@@ -172,7 +172,7 @@ func (bucket Bucket) downloadFile(objectKey, filePath string, partSize int64, op
 	// 等待分片下载完成
 	completed := 0
 	ps := make([]downloadPart, len(parts))
-	for {
+	for completed < len(parts) {
 		select {
 		case part := <-results:
 			completed++
@@ -371,7 +371,7 @@ func (bucket Bucket) downloadFileWithCp(objectKey, filePath string, partSize int
 
 	// 等待分片下载完成
 	completed := 0
-	for {
+	for completed < len(parts) {
 		select {
 		case part := <-results:
 			completed++

+ 2 - 2
oss/upload.go

@@ -154,7 +154,7 @@ func (bucket Bucket) uploadFile(objectKey, filePath string, partSize int64, opti
 	// 等待分配分片上传完成
 	completed := 0
 	parts := make([]UploadPart, len(chunks))
-	for {
+	for completed < len(chunks) {
 		select {
 		case part := <-results:
 			completed++
@@ -408,7 +408,7 @@ func (bucket Bucket) uploadFileWithCp(objectKey, filePath string, partSize int64
 
 	// 等待分配分片上传完成
 	completed := 0
-	for {
+	for completed < len(chunks) {
 		select {
 		case part := <-results:
 			completed++