com.dell.cpsd.common.rabbitmq.consumer.handler.DefaultMessageHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of common-rabbitmq Show documentation
The newest version!
/**
 * Copyright © 2017 Dell Inc. or its subsidiaries.  All Rights Reserved.
 * Dell EMC Confidential/Proprietary Information
 */

package com.dell.cpsd.common.rabbitmq.consumer.handler;

import com.dell.cpsd.common.rabbitmq.consumer.error.ErrorContext;
import com.dell.cpsd.common.rabbitmq.consumer.error.ErrorTransformer;
import com.dell.cpsd.common.rabbitmq.message.HasMessageProperties;
import com.dell.cpsd.common.rabbitmq.message.MessagePropertiesContainer;
import com.dell.cpsd.common.rabbitmq.validators.MessageValidationException;
import com.dell.cpsd.common.rabbitmq.validators.MessageValidator;
import com.dell.cpsd.common.rabbitmq.validators.ValidationResult;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.amqp.core.Message;

/**
 * Validates message. Converts exception in case of error.
 * 
 * Copyright © 2017 Dell Inc. or its subsidiaries.  All Rights Reserved.
 * Dell EMC Confidential/Proprietary Information
 * 
 */
public abstract class DefaultMessageHandler>
        implements MessageHandler
{
    private final Logger log = LoggerFactory.getLogger(getClass());

    protected Class                                  messageClass;
    protected MessageValidator                       validator;
    protected String                                    errorRoutingKeyPrefix;
    protected ErrorTransformer> errorTransformer;

    public DefaultMessageHandler(Class messageClass, MessageValidator validator, String errorRoutingKeyPrefix,
            ErrorTransformer> errorTransformer)
    {
        this.messageClass = messageClass;
        this.validator = validator;
        this.errorRoutingKeyPrefix = errorRoutingKeyPrefix;
        this.errorTransformer = errorTransformer;
    }

    protected abstract void executeOperation(M message) throws Exception;

    @Override
    public boolean canHandle(Message message, Object body)
    {
        return messageClass.isInstance(body);
    }

    @Override
    public void handleMessage(M message) throws Exception
    {
        try
        {
            log.info("Received message {} with correlationId [{}]", message.getClass().getSimpleName(), getCorrelationId(message));

            validate(message);
            executeOperation(message);

            log.info("Finished processing message {} with correlationId [{}]", message.getClass().getSimpleName(),
                    getCorrelationId(message));
        }
        catch (Exception e)
        {
            log.error("Failed to process message {} with correlationId [{}]: {}", message.getClass().getSimpleName(),
                    getCorrelationId(message), e.getMessage());
            handleError(e, message);
        }
    }

    private Object getCorrelationId(M message)
    {
        return message.getMessageProperties() == null ? null : message.getMessageProperties().getCorrelationId();
    }

    protected void validate(M message) throws Exception
    {
        if (validator == null)
        {
            return;
        }
        ValidationResult result = validator.validate(message);
        if (!result.isValid())
        {
            throw new MessageValidationException(result);
        }
    }

    protected void handleError(Exception e, M message) throws Exception
    {
        throw convertError(e, message);
    }

    protected Exception convertError(Exception e, M message)
    {
        ErrorContext> context = new ErrorContext<>(message, errorRoutingKeyPrefix);
        return errorTransformer.transform(e, context);
    }
}