io.camunda.connector.inbound.MessageMapper Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of connector-aws-sqs Show documentation
Camunda AWS SQS Connector
There is a newer version: 8.6.2
/*
 * Copyright Camunda Services GmbH and/or licensed to Camunda Services GmbH
 * under one or more contributor license agreements. Licensed under a proprietary license.
 * See the License.txt file for more information. You may not use this file
 * except in compliance with the proprietary license.
 */
package io.camunda.connector.inbound;

import com.amazonaws.services.sqs.model.Message;
import com.amazonaws.services.sqs.model.MessageAttributeValue;
import com.fasterxml.jackson.core.JsonProcessingException;
import io.camunda.connector.aws.ObjectMapperSupplier;
import io.camunda.connector.inbound.model.message.SqsInboundMessage;
import java.util.HashMap;
import java.util.Map;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class MessageMapper {
  private static final Logger LOGGER = LoggerFactory.getLogger(MessageMapper.class);

  public static SqsInboundMessage toSqsInboundMessage(final Message message) {
    SqsInboundMessage sqsInboundMessage = new SqsInboundMessage();
    sqsInboundMessage.setMessageId(message.getMessageId());
    sqsInboundMessage.setReceiptHandle(message.getReceiptHandle());
    sqsInboundMessage.setMD5OfMessageAttributes(message.getMD5OfMessageAttributes());
    sqsInboundMessage.setAttributes(message.getAttributes());

    Map
        sqsInboundMessageAttributes = new HashMap<>();
    for (Map.Entry entry :
        message.getMessageAttributes().entrySet()) {
      io.camunda.connector.inbound.model.message.MessageAttributeValue sqsInboundMessageAttribute =
          toSqsInboundMessageAttribute(entry.getValue());

      sqsInboundMessageAttributes.put(entry.getKey(), sqsInboundMessageAttribute);
    }
    sqsInboundMessage.setMessageAttributes(sqsInboundMessageAttributes);

    sqsInboundMessage.setBody(toObjectIfPossible(message.getBody()));
    sqsInboundMessage.setmD5OfBody(message.getMD5OfBody());

    return sqsInboundMessage;
  }

  private static Object toObjectIfPossible(final String body) {
    try {
      return ObjectMapperSupplier.getMapperInstance().readValue(body, Object.class);
    } catch (JsonProcessingException e) {
      LOGGER.debug("Cannot parse value to JSON object -> using the raw value");
    }
    return body;
  }

  private static io.camunda.connector.inbound.model.message.MessageAttributeValue
      toSqsInboundMessageAttribute(final MessageAttributeValue attributeValue) {
    io.camunda.connector.inbound.model.message.MessageAttributeValue sqsInboundMessageAttribute =
        new io.camunda.connector.inbound.model.message.MessageAttributeValue();

    sqsInboundMessageAttribute.setBinaryValue(attributeValue.getBinaryValue());
    sqsInboundMessageAttribute.setDataType(attributeValue.getDataType());
    sqsInboundMessageAttribute.setStringValue(attributeValue.getStringValue());
    sqsInboundMessageAttribute.setBinaryListValues(attributeValue.getBinaryListValues());
    sqsInboundMessageAttribute.setStringListValues(attributeValue.getStringListValues());

    return sqsInboundMessageAttribute;
  }
}