[camel] 02/10: CAMEL-14868 - Camel-AWS2-*: Where possible, give the possiblity to the end user to pass an AWS Request pojo as body, aws2-mq create broker

[acosentino]

This is an automated email from the ASF dual-hosted git repository.

acosentino pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/camel.git

commit 141d8fd77e0ae98fa3452369c8897af006ff4b31
Author: Andrea Cosentino <anco...@gmail.com>
AuthorDate: Wed Apr 15 12:38:51 2020 +0200

    CAMEL-14868 - Camel-AWS2-*: Where possible, give the possiblity to the end 
user to pass an AWS Request pojo as body, aws2-mq create broker
---
 .../org/apache/camel/component/aws2/mq/MQ2Producer.java | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

diff --git 
a/components/camel-aws2-mq/src/main/java/org/apache/camel/component/aws2/mq/MQ2Producer.java
 
b/components/camel-aws2-mq/src/main/java/org/apache/camel/component/aws2/mq/MQ2Producer.java
index d9a769d..0cd5a89 100644
--- 
a/components/camel-aws2-mq/src/main/java/org/apache/camel/component/aws2/mq/MQ2Producer.java
+++ 
b/components/camel-aws2-mq/src/main/java/org/apache/camel/component/aws2/mq/MQ2Producer.java
@@ -146,7 +146,7 @@ public class MQ2Producer extends DefaultProducer {
     }
 
     @SuppressWarnings("unchecked")
-    private void createBroker(MqClient mqClient, Exchange exchange) {
+    private void createBroker(MqClient mqClient, Exchange exchange) throws 
InvalidPayloadException {
         String brokerName;
         String brokerEngine;
         String brokerEngineVersion;
@@ -154,6 +154,20 @@ public class MQ2Producer extends DefaultProducer {
         String instanceType;
         Boolean publiclyAccessible;
         List<User> users;
+        if (getConfiguration().isPojoRequest()) {
+            Object payload = exchange.getIn().getMandatoryBody();
+            if (payload instanceof CreateBrokerRequest) {
+                CreateBrokerResponse result;
+                try {
+                    result = mqClient.createBroker((CreateBrokerRequest) 
payload);
+                } catch (AwsServiceException ase) {
+                    LOG.trace("Create Broker command returned the error code 
{}", ase.awsErrorDetails().errorCode());
+                    throw ase;
+                }
+                Message message = getMessageForResponse(exchange);
+                message.setBody(result);
+            }
+        } else {
         CreateBrokerRequest.Builder builder = CreateBrokerRequest.builder();
         if 
(ObjectHelper.isNotEmpty(exchange.getIn().getHeader(MQ2Constants.BROKER_NAME))) 
{
             brokerName = exchange.getIn().getHeader(MQ2Constants.BROKER_NAME, 
String.class);
@@ -206,6 +220,7 @@ public class MQ2Producer extends DefaultProducer {
         }
         Message message = getMessageForResponse(exchange);
         message.setBody(result);
+        }
     }
 
     private void deleteBroker(MqClient mqClient, Exchange exchange) {