multipart_copy.go 4.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223
  1. package sample
  2. import (
  3. "fmt"
  4. "sync"
  5. "github.com/aliyun/aliyun-oss-go-sdk/oss"
  6. )
  7. // MultipartCopySample Multipart Copy Sample
  8. func MultipartCopySample() {
  9. var objectSrc = "my-object-src"
  10. var objectDesc = "my-object-desc"
  11. // 创建Bucket
  12. bucket, err := GetTestBucket(bucketName)
  13. if err != nil {
  14. HandleError(err)
  15. }
  16. err = bucket.PutObjectFromFile(objectSrc, localFile)
  17. if err != nil {
  18. HandleError(err)
  19. }
  20. // 场景1:大文件分片拷贝,按照文件片大小分片
  21. chunks, err := oss.SplitFileByPartNum(localFile, 3)
  22. if err != nil {
  23. HandleError(err)
  24. }
  25. imur, err := bucket.InitiateMultipartUpload(objectDesc)
  26. if err != nil {
  27. HandleError(err)
  28. }
  29. parts := []oss.UploadPart{}
  30. for _, chunk := range chunks {
  31. part, err := bucket.UploadPartCopy(imur, objectSrc, chunk.Offset, chunk.Size,
  32. chunk.Number)
  33. if err != nil {
  34. HandleError(err)
  35. }
  36. parts = append(parts, part)
  37. }
  38. _, err = bucket.CompleteMultipartUpload(imur, parts)
  39. if err != nil {
  40. HandleError(err)
  41. }
  42. err = bucket.DeleteObject(objectDesc)
  43. if err != nil {
  44. HandleError(err)
  45. }
  46. // 场景2:大文件分片拷贝,按照指定文件片数
  47. chunks, err = oss.SplitFileByPartSize(localFile, 1024*100)
  48. if err != nil {
  49. HandleError(err)
  50. }
  51. imur, err = bucket.InitiateMultipartUpload(objectDesc)
  52. if err != nil {
  53. HandleError(err)
  54. }
  55. parts = []oss.UploadPart{}
  56. for _, chunk := range chunks {
  57. part, err := bucket.UploadPartCopy(imur, objectSrc, chunk.Offset, chunk.Size,
  58. chunk.Number)
  59. if err != nil {
  60. HandleError(err)
  61. }
  62. parts = append(parts, part)
  63. }
  64. _, err = bucket.CompleteMultipartUpload(imur, parts)
  65. if err != nil {
  66. HandleError(err)
  67. }
  68. err = bucket.DeleteObject(objectDesc)
  69. if err != nil {
  70. HandleError(err)
  71. }
  72. // 场景3:大文件分片拷贝,初始化时指定对象属性
  73. chunks, err = oss.SplitFileByPartNum(localFile, 3)
  74. if err != nil {
  75. HandleError(err)
  76. }
  77. imur, err = bucket.InitiateMultipartUpload(objectDesc, oss.Meta("myprop", "mypropval"))
  78. if err != nil {
  79. HandleError(err)
  80. }
  81. parts = []oss.UploadPart{}
  82. for _, chunk := range chunks {
  83. part, err := bucket.UploadPartCopy(imur, objectSrc, chunk.Offset, chunk.Size,
  84. chunk.Number)
  85. if err != nil {
  86. HandleError(err)
  87. }
  88. parts = append(parts, part)
  89. }
  90. _, err = bucket.CompleteMultipartUpload(imur, parts)
  91. if err != nil {
  92. HandleError(err)
  93. }
  94. err = bucket.DeleteObject(objectDesc)
  95. if err != nil {
  96. HandleError(err)
  97. }
  98. // 场景4:大文件分片拷贝,每个分片可以有线程/进程/机器独立完成,下面示例是每个线程拷贝一个分片
  99. partNum := 4
  100. chunks, err = oss.SplitFileByPartNum(localFile, partNum)
  101. if err != nil {
  102. HandleError(err)
  103. }
  104. imur, err = bucket.InitiateMultipartUpload(objectDesc)
  105. if err != nil {
  106. HandleError(err)
  107. }
  108. // 并发拷贝分片上传
  109. var waitgroup sync.WaitGroup
  110. var ps = make([]oss.UploadPart, partNum)
  111. for _, chunk := range chunks {
  112. waitgroup.Add(1)
  113. go func(chunk oss.FileChunk) {
  114. part, err := bucket.UploadPartCopy(imur, objectSrc, chunk.Offset, chunk.Size,
  115. chunk.Number)
  116. if err != nil {
  117. HandleError(err)
  118. }
  119. ps[chunk.Number-1] = part
  120. waitgroup.Done()
  121. }(chunk)
  122. }
  123. // 等待拷贝完成
  124. waitgroup.Wait()
  125. // 通知完成
  126. _, err = bucket.CompleteMultipartUpload(imur, ps)
  127. if err != nil {
  128. HandleError(err)
  129. }
  130. err = bucket.DeleteObject(objectDesc)
  131. if err != nil {
  132. HandleError(err)
  133. }
  134. // 场景5:大文件分片拷贝,对拷贝有约束条件,满足时候拷贝,不满足时报错
  135. chunks, err = oss.SplitFileByPartNum(localFile, 3)
  136. if err != nil {
  137. HandleError(err)
  138. }
  139. imur, err = bucket.InitiateMultipartUpload(objectDesc)
  140. if err != nil {
  141. HandleError(err)
  142. }
  143. parts = []oss.UploadPart{}
  144. for _, chunk := range chunks {
  145. constraint := oss.CopySourceIfMatch("InvalidETag")
  146. _, err := bucket.UploadPartCopy(imur, objectSrc, chunk.Offset, chunk.Size,
  147. chunk.Number, constraint)
  148. fmt.Println(err)
  149. }
  150. err = bucket.AbortMultipartUpload(imur)
  151. if err != nil {
  152. HandleError(err)
  153. }
  154. err = bucket.DeleteObject(objectDesc)
  155. if err != nil {
  156. HandleError(err)
  157. }
  158. // 场景6:大文件分片拷贝一部分后,中止上传,上传的数据将丢弃,UploadId也将无效
  159. chunks, err = oss.SplitFileByPartNum(localFile, 3)
  160. if err != nil {
  161. HandleError(err)
  162. }
  163. imur, err = bucket.InitiateMultipartUpload(objectDesc)
  164. if err != nil {
  165. HandleError(err)
  166. }
  167. parts = []oss.UploadPart{}
  168. for _, chunk := range chunks {
  169. part, err := bucket.UploadPartCopy(imur, objectSrc, chunk.Offset, chunk.Size,
  170. chunk.Number)
  171. if err != nil {
  172. HandleError(err)
  173. }
  174. parts = append(parts, part)
  175. }
  176. err = bucket.AbortMultipartUpload(imur)
  177. if err != nil {
  178. HandleError(err)
  179. }
  180. // 删除object和bucket
  181. err = DeleteTestBucketAndObject(bucketName)
  182. if err != nil {
  183. HandleError(err)
  184. }
  185. fmt.Println("MultipartCopySample completed")
  186. }