|
25 | 25 | import com.google.cloud.storage.BlobId; |
26 | 26 | import com.google.cloud.storage.BlobInfo; |
27 | 27 | import com.google.cloud.storage.BlobWriteSessionConfigs; |
28 | | -import com.google.cloud.storage.GrpcStorageOptions; |
29 | | -import com.google.cloud.storage.HttpStorageOptions; |
30 | 28 | import com.google.cloud.storage.ParallelCompositeUploadBlobWriteSessionConfig; |
31 | 29 | import com.google.cloud.storage.ParallelCompositeUploadBlobWriteSessionConfig.ExecutorSupplier; |
32 | 30 | import com.google.cloud.storage.Storage; |
@@ -82,23 +80,11 @@ final class TransferManagerImpl implements TransferManager { |
82 | 80 | .build(); |
83 | 81 | } |
84 | 82 | // Create the blobWriteSessionConfig for ParallelCompositeUpload |
85 | | - ParallelCompositeUploadBlobWriteSessionConfig pcuConfig = |
86 | | - BlobWriteSessionConfigs.parallelCompositeUpload() |
87 | | - .withExecutorSupplier(ExecutorSupplier.useExecutor(executor)); |
88 | | - if (storageOptions instanceof GrpcStorageOptions |
89 | | - && transferManagerConfig.isAllowParallelCompositeUpload()) { |
90 | | - storageOptions = |
91 | | - ((GrpcStorageOptions) storageOptions) |
92 | | - .toBuilder() |
93 | | - .setBlobWriteSessionConfig(pcuConfig) |
94 | | - .build(); |
95 | | - } else if (storageOptions instanceof HttpStorageOptions |
96 | | - && transferManagerConfig.isAllowParallelCompositeUpload()) { |
97 | | - storageOptions = |
98 | | - ((HttpStorageOptions) storageOptions) |
99 | | - .toBuilder() |
100 | | - .setBlobWriteSessionConfig(pcuConfig) |
101 | | - .build(); |
| 83 | + if (transferManagerConfig.isAllowParallelCompositeUpload()) { |
| 84 | + ParallelCompositeUploadBlobWriteSessionConfig pcuConfig = |
| 85 | + BlobWriteSessionConfigs.parallelCompositeUpload() |
| 86 | + .withExecutorSupplier(ExecutorSupplier.useExecutor(executor)); |
| 87 | + storageOptions = storageOptions.toBuilder().setBlobWriteSessionConfig(pcuConfig).build(); |
102 | 88 | } |
103 | 89 | pcuQueue = new ArrayDeque<>(); |
104 | 90 | this.storage = storageOptions.getService(); |
|
0 commit comments