Skip to content

Commit 1674bab

Browse files
Removing AWSRequestOverride config for KMS GenerateRandom operation
1 parent ecc04a9 commit 1674bab

File tree

1 file changed

+3
-7
lines changed
  • athena-federation-sdk/src/main/java/com/amazonaws/athena/connector/lambda/security

1 file changed

+3
-7
lines changed

athena-federation-sdk/src/main/java/com/amazonaws/athena/connector/lambda/security/KmsKeyFactory.java

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -79,13 +79,9 @@ public EncryptionKey create(AwsRequestOverrideConfiguration awsRequestOverrideCo
7979
throw new AthenaConnectorException(e.getMessage(), ErrorDetails.builder().errorCode(FederationSourceErrorCode.ENTITY_NOT_FOUND_EXCEPTION.toString()).build());
8080
}
8181

82-
GenerateRandomRequest.Builder randomBuilder = GenerateRandomRequest.builder()
83-
.numberOfBytes(AesGcmBlockCrypto.NONCE_BYTES);
84-
if (awsRequestOverrideConfiguration != null) {
85-
logger.info("Using AWS KMS Request Override Configuration for RandomKeyGeneration:");
86-
randomBuilder.overrideConfiguration(awsRequestOverrideConfiguration);
87-
}
88-
GenerateRandomResponse randomResponse = kmsClient.generateRandom(randomBuilder.build());
82+
GenerateRandomRequest randomRequest = GenerateRandomRequest.builder()
83+
.numberOfBytes(AesGcmBlockCrypto.NONCE_BYTES).build();
84+
GenerateRandomResponse randomResponse = kmsClient.generateRandom(randomRequest);
8985

9086
return new EncryptionKey(dataKeyResponse.plaintext().asByteArray(), randomResponse.plaintext().asByteArray());
9187
}

0 commit comments

Comments
 (0)