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 37fe32a056b5b040255afc6894ea40cb18fdce5a Author: Andrea Cosentino <[email protected]> AuthorDate: Wed Sep 23 14:36:09 2020 +0200 CAMEL-15561 - Camel-AWS2-Eventbridge: More producer operations - enableRule --- .../aws2/eventbridge/EventbridgeOperations.java | 3 +- .../aws2/eventbridge/EventbridgeProducer.java | 38 ++++++++++++++++++++++ 2 files changed, 40 insertions(+), 1 deletion(-) diff --git a/components/camel-aws2-eventbridge/src/main/java/org/apache/camel/component/aws2/eventbridge/EventbridgeOperations.java b/components/camel-aws2-eventbridge/src/main/java/org/apache/camel/component/aws2/eventbridge/EventbridgeOperations.java index 3939216..8b18e02 100644 --- a/components/camel-aws2-eventbridge/src/main/java/org/apache/camel/component/aws2/eventbridge/EventbridgeOperations.java +++ b/components/camel-aws2-eventbridge/src/main/java/org/apache/camel/component/aws2/eventbridge/EventbridgeOperations.java @@ -21,5 +21,6 @@ public enum EventbridgeOperations { putRule, putTargets, removeTargets, - deleteRule + deleteRule, + enableRule } diff --git a/components/camel-aws2-eventbridge/src/main/java/org/apache/camel/component/aws2/eventbridge/EventbridgeProducer.java b/components/camel-aws2-eventbridge/src/main/java/org/apache/camel/component/aws2/eventbridge/EventbridgeProducer.java index dcf1c30..f8ae11a 100644 --- a/components/camel-aws2-eventbridge/src/main/java/org/apache/camel/component/aws2/eventbridge/EventbridgeProducer.java +++ b/components/camel-aws2-eventbridge/src/main/java/org/apache/camel/component/aws2/eventbridge/EventbridgeProducer.java @@ -36,6 +36,8 @@ import software.amazon.awssdk.awscore.exception.AwsServiceException; import software.amazon.awssdk.services.eventbridge.EventBridgeClient; import software.amazon.awssdk.services.eventbridge.model.DeleteRuleRequest; import software.amazon.awssdk.services.eventbridge.model.DeleteRuleResponse; +import software.amazon.awssdk.services.eventbridge.model.EnableRuleRequest; +import software.amazon.awssdk.services.eventbridge.model.EnableRuleResponse; import software.amazon.awssdk.services.eventbridge.model.PutRuleRequest; import software.amazon.awssdk.services.eventbridge.model.PutRuleResponse; import software.amazon.awssdk.services.eventbridge.model.PutTargetsRequest; @@ -73,6 +75,9 @@ public class EventbridgeProducer extends DefaultProducer { case deleteRule: deleteRule(getEndpoint().getEventbridgeClient(), exchange); break; + case enableRule: + enableRule(getEndpoint().getEventbridgeClient(), exchange); + break; default: throw new IllegalArgumentException("Unsupported operation"); } @@ -251,6 +256,39 @@ public class EventbridgeProducer extends DefaultProducer { message.setBody(result); } } + + private void enableRule(EventBridgeClient eventbridgeClient, Exchange exchange) throws InvalidPayloadException { + if (getConfiguration().isPojoRequest()) { + Object payload = exchange.getIn().getMandatoryBody(); + if (payload instanceof EnableRuleRequest) { + EnableRuleResponse result; + try { + result = eventbridgeClient.enableRule((EnableRuleRequest) payload); + } catch (AwsServiceException ase) { + LOG.trace("Enable Rule command returned the error code {}", ase.awsErrorDetails().errorCode()); + throw ase; + } + Message message = getMessageForResponse(exchange); + message.setBody(result); + } + } else { + EnableRuleRequest.Builder builder = EnableRuleRequest.builder(); + if (ObjectHelper.isNotEmpty(exchange.getIn().getHeader(EventbridgeConstants.RULE_NAME))) { + String ruleName = exchange.getIn().getHeader(EventbridgeConstants.RULE_NAME, String.class); + builder.name(ruleName); + } + builder.eventBusName(getConfiguration().getEventbusName()); + EnableRuleResponse result; + try { + result = eventbridgeClient.enableRule(builder.build()); + } catch (AwsServiceException ase) { + LOG.trace("Enable Rule command returned the error code {}", ase.awsErrorDetails().errorCode()); + throw ase; + } + Message message = getMessageForResponse(exchange); + message.setBody(result); + } + } @Override public EventbridgeEndpoint getEndpoint() {
