Java tutorial
/** * Copyright 2013 Amazon.com, Inc. or its affiliates. All Rights Reserved. * * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance with * the License. A copy of the License is located at * * http://aws.amazon.com/apache2.0/ * * or in the "LICENSE" file accompanying this file. This file 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 awslabs.lab31; import java.util.HashMap; import java.util.List; import com.amazonaws.auth.policy.Policy; import com.amazonaws.auth.policy.Principal; import com.amazonaws.auth.policy.Resource; import com.amazonaws.auth.policy.Statement; import com.amazonaws.auth.policy.Statement.Effect; import com.amazonaws.auth.policy.actions.SQSActions; import com.amazonaws.auth.policy.conditions.ConditionFactory; import com.amazonaws.services.sns.AmazonSNSClient; import com.amazonaws.services.sns.model.CreateTopicRequest; import com.amazonaws.services.sns.model.CreateTopicResult; import com.amazonaws.services.sns.model.DeleteTopicRequest; import com.amazonaws.services.sns.model.ListSubscriptionsByTopicRequest; import com.amazonaws.services.sns.model.ListSubscriptionsByTopicResult; import com.amazonaws.services.sns.model.PublishRequest; import com.amazonaws.services.sns.model.SubscribeRequest; import com.amazonaws.services.sns.model.Subscription; import com.amazonaws.services.sns.model.UnsubscribeRequest; import com.amazonaws.services.sqs.AmazonSQSClient; import com.amazonaws.services.sqs.model.CreateQueueRequest; import com.amazonaws.services.sqs.model.CreateQueueResult; import com.amazonaws.services.sqs.model.DeleteMessageRequest; import com.amazonaws.services.sqs.model.DeleteQueueRequest; import com.amazonaws.services.sqs.model.GetQueueAttributesRequest; import com.amazonaws.services.sqs.model.GetQueueAttributesResult; import com.amazonaws.services.sqs.model.Message; import com.amazonaws.services.sqs.model.ReceiveMessageRequest; import com.amazonaws.services.sqs.model.ReceiveMessageResult; import com.amazonaws.services.sqs.model.SendMessageRequest; import com.amazonaws.services.sqs.model.SetQueueAttributesRequest; /** * Project: Lab3.1 */ public abstract class SolutionCode implements ILabCode, IOptionalLabCode { @Override public String createQueue(AmazonSQSClient sqsClient, String queueName) { // TODO: Construct a CreateQueueRequest object using the provided queue name. CreateQueueRequest createQueueRequest = new CreateQueueRequest().withQueueName(queueName); // TODO: Submit the request using the createQueue method of the sqsClient object. CreateQueueResult createQueueResult = sqsClient.createQueue(createQueueRequest); // TODO: Return the queue URL from the request result. return createQueueResult.getQueueUrl(); } @Override public String getQueueArn(AmazonSQSClient sqsClient, String queueUrl) { // TODO: Construct a GetQueueAttributesRequest for the specified queue and for the attribute named "QueueArn". GetQueueAttributesRequest getQueueAttributesRequest = new GetQueueAttributesRequest().withQueueUrl(queueUrl) .withAttributeNames("QueueArn"); // TODO: Submit the request using the getQueueAttributes method of the sqsClient object. GetQueueAttributesResult getQueueAttributesResult = sqsClient.getQueueAttributes(getQueueAttributesRequest); // TODO: Return the QueueArn attribute value. return getQueueAttributesResult.getAttributes().get("QueueArn"); } @Override public String createTopic(AmazonSNSClient snsClient, String topicName) { // TODO: Construct a CreateTopicRequest object using the specified topic name. CreateTopicRequest createTopicRequest = new CreateTopicRequest().withName(topicName); // TODO: Submit the request using the createTopic method of the snsClient object. CreateTopicResult createTopicResult = snsClient.createTopic(createTopicRequest); // TODO: Return the topic ARN from the request result. return createTopicResult.getTopicArn(); } @Override public void createSubscription(AmazonSNSClient snsClient, String queueArn, String topicArn) { // TODO: Construct a SubscribeRequest object using the "sqs" protocol for the specified queue ARN endpoint and // topic ARN. SubscribeRequest subscribeRequest = new SubscribeRequest().withEndpoint(queueArn).withProtocol("sqs") .withTopicArn(topicArn); // TODO: Submit the request using the subscribe method of the snsClient object. snsClient.subscribe(subscribeRequest); } @Override public void publishTopicMessage(AmazonSNSClient snsClient, String topicArn, String subject, String message) { // TODO: Construct a PublishRequest object using the provided subject, message, and topic ARN. PublishRequest publishRequest = new PublishRequest().withMessage(message).withSubject(subject) .withTopicArn(topicArn); // TODO: Submit the request using the publish method of the snsClient object. snsClient.publish(publishRequest); } @Override public void postToQueue(AmazonSQSClient sqsClient, String queueUrl, String messageText) { // TODO: Construct a SendMessageRequest object using the provided queue URL and message. SendMessageRequest sendMessageRequest = new SendMessageRequest().withMessageBody(messageText) .withQueueUrl(queueUrl); // TODO: Submit the request using the sendMessage method of the sqsClient object. sqsClient.sendMessage(sendMessageRequest); } @Override public List<Message> readMessages(AmazonSQSClient sqsClient, String queueUrl) { ReceiveMessageResult receiveMessageResult; // TODO: Construct a ReceiveMessageRequest object using the queue URL provided. Set the maximum number of // messages to 10. ReceiveMessageRequest receiveMessageRequest = new ReceiveMessageRequest().withMaxNumberOfMessages(10) .withQueueUrl(queueUrl); // TODO: Submit the request using the receiveMessage method of the sqsClient object. Store the result in the // receiveMessageResult object (already defined). receiveMessageResult = sqsClient.receiveMessage(receiveMessageRequest); return receiveMessageResult.getMessages(); } @Override public void removeMessage(AmazonSQSClient sqsClient, String queueUrl, String receiptHandle) { // TODO: Construct a DeleteMessageRequest object using the specified queue URL and receipt handle. DeleteMessageRequest deleteMessageRequest = new DeleteMessageRequest().withQueueUrl(queueUrl) .withReceiptHandle(receiptHandle); // TODO: Submit the request using the deleteMessage method of the sqsClient object. sqsClient.deleteMessage(deleteMessageRequest); } @Override public void deleteSubscriptions(AmazonSNSClient snsClient, String topicArn) { // TODO: Construct a ListSubscriptionsByTopicRequest object using the provided topic ARN. ListSubscriptionsByTopicRequest listSubscriptionsByTopicRequest = new ListSubscriptionsByTopicRequest() .withTopicArn(topicArn); // TODO: Submit the request using the listSubscriptionsByTopic method of the snsClient object. ListSubscriptionsByTopicResult listSubscriptionsByTopicResult = snsClient .listSubscriptionsByTopic(listSubscriptionsByTopicRequest); // TODO: Iterate over the subscriptions in the request result. // TODO: For each subscription, construct an UnsubscribeRequest object using the subscription ARN. // TODO: For each unsubscribe request, submit it using the unsubscribe method of the snsClient object. for (Subscription subscription : listSubscriptionsByTopicResult.getSubscriptions()) { UnsubscribeRequest unsubscribeRequest = new UnsubscribeRequest() .withSubscriptionArn(subscription.getSubscriptionArn()); snsClient.unsubscribe(unsubscribeRequest); } } @Override public void deleteTopic(AmazonSNSClient snsClient, String topicArn) { // TODO: Construct a DeleteTopicRequest object using the provided topic ARN. DeleteTopicRequest deleteTopicRequest = new DeleteTopicRequest().withTopicArn(topicArn); // TODO: Submit the request using the deleteTopic method of the snsClient object. snsClient.deleteTopic(deleteTopicRequest); } @Override public void deleteQueue(AmazonSQSClient sqsClient, String queueUrl) { // TODO: Construct a DeleteQueueRequest object using the provided queue URL. DeleteQueueRequest deleteQueueRequest = new DeleteQueueRequest().withQueueUrl(queueUrl); // TODO: Submit the request using the deleteQueue method of the sqsClient object. sqsClient.deleteQueue(deleteQueueRequest); } @Override public void grantNotificationPermission(AmazonSQSClient sqsClient, String queueArn, String queueUrl, String topicArn) { Statement statement = new Statement(Effect.Allow).withActions(SQSActions.SendMessage) .withPrincipals(new Principal("*")).withConditions(ConditionFactory.newSourceArnCondition(topicArn)) .withResources(new Resource(queueArn)); Policy policy = new Policy("SubscriptionPermission").withStatements(statement); HashMap<String, String> attributes = new HashMap<String, String>(); attributes.put("Policy", policy.toJson()); // Create the request to set the queue attributes for policy SetQueueAttributesRequest setQueueAttributesRequest = new SetQueueAttributesRequest().withQueueUrl(queueUrl) .withAttributes(attributes); // Set the queue policy sqsClient.setQueueAttributes(setQueueAttributesRequest); } }