|
3 | 3 | import software.amazon.awssdk.protocols.jsoncore.JsonWriter; |
4 | 4 | import software.amazon.awssdk.services.s3.model.CreateMultipartUploadRequest; |
5 | 5 | import software.amazon.awssdk.services.s3.model.PutObjectRequest; |
| 6 | +import software.amazon.awssdk.services.s3.model.S3Request; |
6 | 7 | import software.amazon.encryption.s3.S3EncryptionClientException; |
7 | 8 | import software.amazon.encryption.s3.materials.EncryptedDataKey; |
8 | 9 | import software.amazon.encryption.s3.materials.EncryptionMaterials; |
@@ -36,12 +37,19 @@ public PutObjectRequest encodeMetadata(EncryptionMaterials materials, byte[] iv, |
36 | 37 | } |
37 | 38 |
|
38 | 39 | public CreateMultipartUploadRequest encodeMetadata(EncryptionMaterials materials, byte[] iv, CreateMultipartUploadRequest createMultipartUploadRequest) { |
39 | | - Map<String, String> newMetadata = addMetadataToMap(createMultipartUploadRequest.metadata(), materials, iv); |
40 | | - return createMultipartUploadRequest.toBuilder() |
41 | | - .metadata(newMetadata) |
42 | | - .build(); |
| 40 | + if(_instructionFileConfig.isInstructionFilePutEnabled()) { |
| 41 | + final String metadataString = metadataToString(materials, iv); |
| 42 | + PutObjectRequest putObjectRequest = ConvertSDKRequests.convertRequest(createMultipartUploadRequest); |
| 43 | + _instructionFileConfig.putInstructionFile(putObjectRequest, metadataString); |
| 44 | + // the original request object is returned as-is |
| 45 | + return createMultipartUploadRequest; |
| 46 | + } else { |
| 47 | + Map<String, String> newMetadata = addMetadataToMap(createMultipartUploadRequest.metadata(), materials, iv); |
| 48 | + return createMultipartUploadRequest.toBuilder() |
| 49 | + .metadata(newMetadata) |
| 50 | + .build(); |
| 51 | + } |
43 | 52 | } |
44 | | - |
45 | 53 | private String metadataToString(EncryptionMaterials materials, byte[] iv) { |
46 | 54 | // this is just the metadata map serialized as JSON |
47 | 55 | // so first get the Map |
|
0 commit comments