This is an automated email from the ASF dual-hosted git repository. acosentino pushed a commit to branch main in repository https://gitbox.apache.org/repos/asf/camel.git
commit 5b0cdad8ba5982d3d8e49cbdc269cc32744da2f6 Author: Andrea Cosentino <[email protected]> AuthorDate: Fri Jun 16 10:21:13 2023 +0200 CAMEL-19159 - Camel-AWS: Support Profile Credential provider as configuration - AWS Kinesis Signed-off-by: Andrea Cosentino <[email protected]> --- .../aws2/kinesis/client/KinesisClientFactory.java | 10 ++- .../impl/KinesisClientIAMProfileOptimizedImpl.java | 94 ++++++++++++++++++++++ 2 files changed, 102 insertions(+), 2 deletions(-) diff --git a/components/camel-aws/camel-aws2-kinesis/src/main/java/org/apache/camel/component/aws2/kinesis/client/KinesisClientFactory.java b/components/camel-aws/camel-aws2-kinesis/src/main/java/org/apache/camel/component/aws2/kinesis/client/KinesisClientFactory.java index c8f3228ea78..c6b8abb072d 100644 --- a/components/camel-aws/camel-aws2-kinesis/src/main/java/org/apache/camel/component/aws2/kinesis/client/KinesisClientFactory.java +++ b/components/camel-aws/camel-aws2-kinesis/src/main/java/org/apache/camel/component/aws2/kinesis/client/KinesisClientFactory.java @@ -18,6 +18,7 @@ package org.apache.camel.component.aws2.kinesis.client; import org.apache.camel.component.aws2.kinesis.Kinesis2Configuration; import org.apache.camel.component.aws2.kinesis.client.impl.KinesisClientIAMOptimizedImpl; +import org.apache.camel.component.aws2.kinesis.client.impl.KinesisClientIAMProfileOptimizedImpl; import org.apache.camel.component.aws2.kinesis.client.impl.KinesisClientStandardImpl; /** @@ -35,7 +36,12 @@ public final class KinesisClientFactory { * @return KinesisClient */ public static KinesisInternalClient getKinesisClient(Kinesis2Configuration configuration) { - return configuration.isUseDefaultCredentialsProvider() - ? new KinesisClientIAMOptimizedImpl(configuration) : new KinesisClientStandardImpl(configuration); + if (Boolean.TRUE.equals(configuration.isUseDefaultCredentialsProvider())) { + return new KinesisClientIAMOptimizedImpl(configuration); + } else if (Boolean.TRUE.equals(configuration.isUseProfileCredentialsProvider())) { + return new KinesisClientIAMProfileOptimizedImpl(configuration); + } else { + return new KinesisClientStandardImpl(configuration); + } } } diff --git a/components/camel-aws/camel-aws2-kinesis/src/main/java/org/apache/camel/component/aws2/kinesis/client/impl/KinesisClientIAMProfileOptimizedImpl.java b/components/camel-aws/camel-aws2-kinesis/src/main/java/org/apache/camel/component/aws2/kinesis/client/impl/KinesisClientIAMProfileOptimizedImpl.java new file mode 100644 index 00000000000..167720af8ef --- /dev/null +++ b/components/camel-aws/camel-aws2-kinesis/src/main/java/org/apache/camel/component/aws2/kinesis/client/impl/KinesisClientIAMProfileOptimizedImpl.java @@ -0,0 +1,94 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.apache.camel.component.aws2.kinesis.client.impl; + +import org.apache.camel.component.aws2.kinesis.Kinesis2Configuration; +import org.apache.camel.component.aws2.kinesis.client.KinesisInternalClient; +import org.apache.camel.util.ObjectHelper; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; +import software.amazon.awssdk.auth.credentials.ProfileCredentialsProvider; +import software.amazon.awssdk.http.SdkHttpClient; +import software.amazon.awssdk.http.SdkHttpConfigurationOption; +import software.amazon.awssdk.http.apache.ApacheHttpClient; +import software.amazon.awssdk.http.apache.ProxyConfiguration; +import software.amazon.awssdk.regions.Region; +import software.amazon.awssdk.services.kinesis.KinesisClient; +import software.amazon.awssdk.services.kinesis.KinesisClientBuilder; +import software.amazon.awssdk.utils.AttributeMap; + +import java.net.URI; + +/** + * Manage an AWS Kinesis client for all users to use (enabling temporary creds). This implementation is for remote + * instances to manage the credentials on their own (eliminating credential rotations) + */ +public class KinesisClientIAMProfileOptimizedImpl implements KinesisInternalClient { + private static final Logger LOG = LoggerFactory.getLogger(KinesisClientIAMProfileOptimizedImpl.class); + private Kinesis2Configuration configuration; + + /** + * Constructor that uses the config file. + */ + public KinesisClientIAMProfileOptimizedImpl(Kinesis2Configuration configuration) { + LOG.trace("Creating an AWS Kinesis client for an ec2 instance with IAM temporary credentials (normal for ec2s)."); + this.configuration = configuration; + } + + /** + * Getting the Kinesis client that is used. + * + * @return Amazon Kinesis Client. + */ + @Override + public KinesisClient getKinesisClient() { + KinesisClient client = null; + KinesisClientBuilder clientBuilder = KinesisClient.builder(); + ProxyConfiguration.Builder proxyConfig = null; + ApacheHttpClient.Builder httpClientBuilder = null; + + if (ObjectHelper.isNotEmpty(configuration.getProxyHost()) && ObjectHelper.isNotEmpty(configuration.getProxyPort())) { + proxyConfig = ProxyConfiguration.builder(); + URI proxyEndpoint = URI.create(configuration.getProxyProtocol() + "://" + configuration.getProxyHost() + ":" + + configuration.getProxyPort()); + proxyConfig.endpoint(proxyEndpoint); + httpClientBuilder = ApacheHttpClient.builder().proxyConfiguration(proxyConfig.build()); + clientBuilder = clientBuilder.httpClientBuilder(httpClientBuilder); + } + if (configuration.getProfileCredentialsName() != null) { + clientBuilder = clientBuilder + .credentialsProvider(ProfileCredentialsProvider.create(configuration.getProfileCredentialsName())); + } + if (ObjectHelper.isNotEmpty(configuration.getRegion())) { + clientBuilder = clientBuilder.region(Region.of(configuration.getRegion())); + } + if (configuration.isOverrideEndpoint()) { + clientBuilder.endpointOverride(URI.create(configuration.getUriEndpointOverride())); + } + if (configuration.isTrustAllCertificates()) { + SdkHttpClient ahc = ApacheHttpClient.builder().buildWithDefaults(AttributeMap + .builder() + .put( + SdkHttpConfigurationOption.TRUST_ALL_CERTIFICATES, + Boolean.TRUE) + .build()); + clientBuilder.httpClient(ahc); + } + client = clientBuilder.build(); + return client; + } +}
