Fix some bug in thread upload

tencentyun · Jan 11, 2018 · 1f99e53 · 1f99e53
1 parent 033e196
commit 1f99e53
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/qcloud_cos/cos_client.py b/qcloud_cos/cos_client.py
@@ -1279,6 +1279,9 @@ def upload_file(self, Bucket, Key, LocalFilePath, PartSize=10, MAXThread=5, **kw
                     offset += part_size
 
             pool.wait_completion()
+            result = pool.get_result()
+            if not result['success_all']:
+                raise CosClientError('some upload_part fail after max_retry')
             lst = sorted(lst, key=lambda x: x['PartNumber'])  # 按PartNumber升序排列
 
             # 完成分片上传
@@ -1385,6 +1388,10 @@ def copy(self, Bucket, Key, CopySource, CopyStatus='Copy', PartSize=10, MAXThrea
                 offset += part_size
 
         pool.wait_completion()
+        result = pool.get_result()
+        if not result['success_all']:
+            raise CosClientError('some upload_part_copy fail after max_retry')
+
         lst = sorted(lst, key=lambda x: x['PartNumber'])  # 按PartNumber升序排列
         # 完成分片上传
         try:
@@ -1450,6 +1457,9 @@ def upload_file_from_buffer(self, Bucket, Key, Body, MaxBufferSize=100, PartSize
             data = Body.read(part_size)
 
         pool.wait_completion()
+        result = pool.get_result()
+        if not result['success_all']:
+            raise CosClientError('some upload_part fail after max_retry')
         lst = sorted(lst, key=lambda x: x['PartNumber'])  # 按PartNumber升序排列
 
         # 完成分片上传