| /* |
| * 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.aws.ddb; |
| |
| import com.amazonaws.AmazonServiceException; |
| import com.amazonaws.ClientConfiguration; |
| import com.amazonaws.auth.AWSCredentials; |
| import com.amazonaws.auth.AWSCredentialsProvider; |
| import com.amazonaws.auth.AWSStaticCredentialsProvider; |
| import com.amazonaws.auth.BasicAWSCredentials; |
| import com.amazonaws.regions.Regions; |
| import com.amazonaws.services.dynamodbv2.AmazonDynamoDB; |
| import com.amazonaws.services.dynamodbv2.AmazonDynamoDBClientBuilder; |
| import com.amazonaws.services.dynamodbv2.model.CreateTableRequest; |
| import com.amazonaws.services.dynamodbv2.model.DescribeTableRequest; |
| import com.amazonaws.services.dynamodbv2.model.KeySchemaElement; |
| import com.amazonaws.services.dynamodbv2.model.ProvisionedThroughput; |
| import com.amazonaws.services.dynamodbv2.model.ResourceNotFoundException; |
| import com.amazonaws.services.dynamodbv2.model.TableDescription; |
| import com.amazonaws.services.dynamodbv2.model.TableStatus; |
| |
| import org.apache.camel.Component; |
| import org.apache.camel.Consumer; |
| import org.apache.camel.Processor; |
| import org.apache.camel.Producer; |
| import org.apache.camel.spi.UriEndpoint; |
| import org.apache.camel.spi.UriParam; |
| import org.apache.camel.support.ScheduledPollEndpoint; |
| import org.apache.camel.util.ObjectHelper; |
| |
| /** |
| * The aws-ddb component is used for storing and retrieving data from Amazon's DynamoDB service. |
| */ |
| @UriEndpoint(firstVersion = "2.10.0", scheme = "aws-ddb", title = "AWS DynamoDB", syntax = "aws-ddb:tableName", producerOnly = true, label = "cloud,database,nosql") |
| public class DdbEndpoint extends ScheduledPollEndpoint { |
| |
| @UriParam |
| private DdbConfiguration configuration; |
| |
| private AmazonDynamoDB ddbClient; |
| |
| public DdbEndpoint(String uri, Component component, DdbConfiguration configuration) { |
| super(uri, component); |
| this.configuration = configuration; |
| } |
| |
| @Override |
| public Consumer createConsumer(Processor processor) throws Exception { |
| throw new UnsupportedOperationException("You cannot receive messages from this endpoint"); |
| } |
| |
| @Override |
| public Producer createProducer() throws Exception { |
| return new DdbProducer(this); |
| } |
| |
| @Override |
| public void doStart() throws Exception { |
| super.doStart(); |
| |
| ddbClient = configuration.getAmazonDDBClient() != null ? configuration.getAmazonDDBClient() |
| : createDdbClient(); |
| |
| String tableName = getConfiguration().getTableName(); |
| log.trace("Querying whether table [{}] already exists...", tableName); |
| |
| try { |
| DescribeTableRequest request = new DescribeTableRequest().withTableName(tableName); |
| TableDescription tableDescription = ddbClient.describeTable(request).getTable(); |
| if (!isTableActive(tableDescription)) { |
| waitForTableToBecomeAvailable(tableName); |
| } |
| |
| log.trace("Table [{}] already exists", tableName); |
| return; |
| } catch (ResourceNotFoundException e) { |
| log.trace("Table [{}] doesn't exist yet", tableName); |
| log.trace("Creating table [{}]...", tableName); |
| TableDescription tableDescription = createTable(tableName); |
| if (!isTableActive(tableDescription)) { |
| waitForTableToBecomeAvailable(tableName); |
| } |
| |
| log.trace("Table [{}] created", tableName); |
| } |
| } |
| |
| @Override |
| public void doStop() throws Exception { |
| if (ObjectHelper.isEmpty(configuration.getAmazonDDBClient())) { |
| if (ddbClient != null) { |
| ddbClient.shutdown(); |
| } |
| } |
| super.doStop(); |
| } |
| |
| private TableDescription createTable(String tableName) { |
| CreateTableRequest createTableRequest = new CreateTableRequest().withTableName(tableName) |
| .withKeySchema( |
| new KeySchemaElement().withAttributeName( |
| configuration.getKeyAttributeName()) |
| .withKeyType(configuration.getKeyAttributeType())) |
| .withProvisionedThroughput( |
| new ProvisionedThroughput().withReadCapacityUnits(configuration.getReadCapacity()) |
| .withWriteCapacityUnits(configuration.getWriteCapacity())); |
| return getDdbClient().createTable(createTableRequest).getTableDescription(); |
| } |
| |
| public DdbConfiguration getConfiguration() { |
| return configuration; |
| } |
| |
| public AmazonDynamoDB getDdbClient() { |
| return ddbClient; |
| } |
| |
| AmazonDynamoDB createDdbClient() { |
| AmazonDynamoDB client = null; |
| ClientConfiguration clientConfiguration = null; |
| AmazonDynamoDBClientBuilder clientBuilder = null; |
| boolean isClientConfigFound = false; |
| if (ObjectHelper.isNotEmpty(configuration.getProxyHost()) && ObjectHelper.isNotEmpty(configuration.getProxyPort())) { |
| clientConfiguration = new ClientConfiguration(); |
| clientConfiguration.setProxyProtocol(configuration.getProxyProtocol()); |
| clientConfiguration.setProxyHost(configuration.getProxyHost()); |
| clientConfiguration.setProxyPort(configuration.getProxyPort()); |
| isClientConfigFound = true; |
| } |
| if (configuration.getAccessKey() != null && configuration.getSecretKey() != null) { |
| AWSCredentials credentials = new BasicAWSCredentials(configuration.getAccessKey(), configuration.getSecretKey()); |
| AWSCredentialsProvider credentialsProvider = new AWSStaticCredentialsProvider(credentials); |
| if (isClientConfigFound) { |
| clientBuilder = AmazonDynamoDBClientBuilder.standard().withClientConfiguration(clientConfiguration).withCredentials(credentialsProvider); |
| } else { |
| clientBuilder = AmazonDynamoDBClientBuilder.standard().withCredentials(credentialsProvider); |
| } |
| } else { |
| if (isClientConfigFound) { |
| clientBuilder = AmazonDynamoDBClientBuilder.standard(); |
| } else { |
| clientBuilder = AmazonDynamoDBClientBuilder.standard().withClientConfiguration(clientConfiguration); |
| } |
| } |
| if (ObjectHelper.isNotEmpty(configuration.getRegion())) { |
| clientBuilder = clientBuilder.withRegion(Regions.valueOf(configuration.getRegion())); |
| } |
| client = clientBuilder.build(); |
| return client; |
| } |
| |
| private void waitForTableToBecomeAvailable(String tableName) { |
| log.trace("Waiting for [{}] to become ACTIVE...", tableName); |
| |
| long waitTime = 5 * 60 * 1000; |
| while (waitTime > 0) { |
| try { |
| Thread.sleep(1000 * 5); |
| waitTime -= 5000; |
| } catch (Exception e) { |
| } |
| try { |
| DescribeTableRequest request = new DescribeTableRequest().withTableName(tableName); |
| TableDescription tableDescription = getDdbClient().describeTable(request).getTable(); |
| if (isTableActive(tableDescription)) { |
| log.trace("Table [{}] became active", tableName); |
| return; |
| } |
| log.trace("Table [{}] not active yet", tableName); |
| } catch (AmazonServiceException ase) { |
| if (!ase.getErrorCode().equalsIgnoreCase("ResourceNotFoundException")) { |
| throw ase; |
| } |
| } |
| } |
| |
| throw new RuntimeException("Table " + tableName + " never went active"); |
| } |
| |
| private boolean isTableActive(TableDescription tableDescription) { |
| return tableDescription.getTableStatus().equals(TableStatus.ACTIVE.toString()); |
| } |
| } |