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 ecb137a7c66f21ee860c9c54ca2fd7bdc4400e2c Author: Andrea Cosentino <[email protected]> AuthorDate: Mon Jun 19 11:35:33 2023 +0200 CAMEL-19159 - Camel-AWS: Support Profile Credential provider as configuration - AWS MSK Signed-off-by: Andrea Cosentino <[email protected]> --- .../aws2/msk/client/MSK2ClientFactory.java | 10 ++- .../impl/MSK2ClientProfileOptimizedImpl.java | 93 ++++++++++++++++++++++ 2 files changed, 101 insertions(+), 2 deletions(-) diff --git a/components/camel-aws/camel-aws2-msk/src/main/java/org/apache/camel/component/aws2/msk/client/MSK2ClientFactory.java b/components/camel-aws/camel-aws2-msk/src/main/java/org/apache/camel/component/aws2/msk/client/MSK2ClientFactory.java index 0e2d4b89bfd..31ca559eeec 100644 --- a/components/camel-aws/camel-aws2-msk/src/main/java/org/apache/camel/component/aws2/msk/client/MSK2ClientFactory.java +++ b/components/camel-aws/camel-aws2-msk/src/main/java/org/apache/camel/component/aws2/msk/client/MSK2ClientFactory.java @@ -18,6 +18,7 @@ package org.apache.camel.component.aws2.msk.client; import org.apache.camel.component.aws2.msk.MSK2Configuration; import org.apache.camel.component.aws2.msk.client.impl.MSK2ClientOptimizedImpl; +import org.apache.camel.component.aws2.msk.client.impl.MSK2ClientProfileOptimizedImpl; import org.apache.camel.component.aws2.msk.client.impl.MSK2ClientStandardImpl; /** @@ -35,7 +36,12 @@ public final class MSK2ClientFactory { * @return MqClient */ public static MSK2InternalClient getKafkaClient(MSK2Configuration configuration) { - return Boolean.TRUE.equals(configuration.isUseDefaultCredentialsProvider()) - ? new MSK2ClientOptimizedImpl(configuration) : new MSK2ClientStandardImpl(configuration); + if (Boolean.TRUE.equals(configuration.isUseDefaultCredentialsProvider())) { + return new MSK2ClientOptimizedImpl(configuration); + } else if (Boolean.TRUE.equals(configuration.isUseProfileCredentialsProvider())) { + return new MSK2ClientProfileOptimizedImpl(configuration); + } else { + return new MSK2ClientStandardImpl(configuration); + } } } diff --git a/components/camel-aws/camel-aws2-msk/src/main/java/org/apache/camel/component/aws2/msk/client/impl/MSK2ClientProfileOptimizedImpl.java b/components/camel-aws/camel-aws2-msk/src/main/java/org/apache/camel/component/aws2/msk/client/impl/MSK2ClientProfileOptimizedImpl.java new file mode 100644 index 00000000000..5533528b522 --- /dev/null +++ b/components/camel-aws/camel-aws2-msk/src/main/java/org/apache/camel/component/aws2/msk/client/impl/MSK2ClientProfileOptimizedImpl.java @@ -0,0 +1,93 @@ +/* + * 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.msk.client.impl; + +import org.apache.camel.component.aws2.msk.MSK2Configuration; +import org.apache.camel.component.aws2.msk.client.MSK2InternalClient; +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.kafka.KafkaClient; +import software.amazon.awssdk.services.kafka.KafkaClientBuilder; +import software.amazon.awssdk.utils.AttributeMap; + +import java.net.URI; + +/** + * Manage an AWS MSK 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 MSK2ClientProfileOptimizedImpl implements MSK2InternalClient { + private static final Logger LOG = LoggerFactory.getLogger(MSK2ClientProfileOptimizedImpl.class); + private MSK2Configuration configuration; + + /** + * Constructor that uses the config file. + */ + public MSK2ClientProfileOptimizedImpl(MSK2Configuration configuration) { + LOG.trace("Creating an AWS MSK client for an ec2 instance with IAM temporary credentials (normal for ec2s)."); + this.configuration = configuration; + } + + /** + * Getting the Kafka aws client that is used. + * + * @return Kafka Client. + */ + @Override + public KafkaClient getKafkaClient() { + KafkaClient client = null; + KafkaClientBuilder clientBuilder = KafkaClient.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.httpClientBuilder(httpClientBuilder) + .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; + } +}
