com.dell.cpsd.common.rabbitmq.registration.RegistrationAwareClassMapper 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.registration;

import com.dell.cpsd.common.logging.ILogger;
import com.dell.cpsd.common.rabbitmq.MessageAnnotationProcessor;
import com.dell.cpsd.common.rabbitmq.log.RabbitMQLoggingManager;
import com.dell.cpsd.common.rabbitmq.registration.notifier.model.MessageDirectionType;
import com.dell.cpsd.common.rabbitmq.registration.notifier.model.MessageExchangeDto;
import com.dell.cpsd.common.rabbitmq.registration.notifier.model.MessageQueueDto;
import com.dell.cpsd.common.rabbitmq.registration.notifier.model.MessageRegistrationDto;
import com.fasterxml.jackson.databind.JsonMappingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.SerializationFeature;
import com.fasterxml.jackson.databind.jsonschema.JsonSchema;
import org.springframework.amqp.support.converter.DefaultClassMapper;

import java.io.IOException;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 
 * Copyright © 2017 Dell Inc. or its subsidiaries.  All Rights Reserved.
 * Dell EMC Confidential/Proprietary Information
 * 
 *
 * @since SINCE-TBD
 */
public class RegistrationAwareClassMapper extends DefaultClassMapper implements MessageRegistrationAware
{
    private static final ILogger LOGGER = RabbitMQLoggingManager.getLogger(RegistrationAwareClassMapper.class);

    private String  serviceName;
    private boolean autoRegister;

    private Map classRegistrations = new HashMap<>();

    public RegistrationAwareClassMapper(String serviceName, boolean autoRegister)
    {
        this.serviceName = serviceName;
        this.autoRegister = autoRegister;
    }

    public MessageRegistrationBuilder builder(Class expectedClass)
    {
        return new MessageRegistrationBuilder(this, serviceName, expectedClass);
    }

    public void add(MessageRegistrationDto messageRegistrationDto)
    {
        this.classRegistrations.put(messageRegistrationDto.getMessageType(), messageRegistrationDto);
    }

    public void apply()
    {
        final Map> classMappings = new HashMap<>();

        classRegistrations.forEach((k, v) ->
        {
            classMappings.put(k, v.getMessageClass());
        });

        super.setIdClassMapping(classMappings);
    }

    @Override
    public Collection getRegistrations()
    {
        return classRegistrations.values();
    }

    @Override
    public boolean isAutoRegister()
    {
        return autoRegister;
    }

    public class MessageRegistrationBuilder
    {
        private RegistrationAwareClassMapper classMapper;
        private String                       serviceName;
        private Class                     expectedClass;
        private List exchanges = new ArrayList<>();
        private List    queues    = new ArrayList<>();

        public MessageRegistrationBuilder(RegistrationAwareClassMapper classMapper, String serviceName, Class expectedClass)
        {
            this.classMapper = classMapper;
            this.serviceName = serviceName;
            this.expectedClass = expectedClass;
        }

        public MessageRegistrationBuilder toExchange(String name)
        {
            return toExchange(name, MessageDirectionType.PRODUCE);
        }

        public MessageRegistrationBuilder toExchange(String name, MessageDirectionType direction)
        {
            exchanges.add(new MessageExchangeDto(name, direction));
            return this;
        }

        public MessageRegistrationBuilder fromQueue(String name)
        {
            queues.add(new MessageQueueDto(name));
            return this;
        }

        public MessageRegistrationDto register()
        {
            final List result = new ArrayList<>();

            final MessageAnnotationProcessor messageAnnotationProcessor = new MessageAnnotationProcessor();
            messageAnnotationProcessor.process((messageType, messageClass) ->
            {
                JsonSchema schema = createSchema(expectedClass);
                result.add(new MessageRegistrationDto(serviceName, messageClass, messageType, "1.0", schema, exchanges, queues));
            }, expectedClass);

            MessageRegistrationDto dto = result.iterator().next();
            classMapper.add(dto);

            return dto;
        }

        private JsonSchema createSchema(Class clazz)
        {
            ObjectMapper mapper = new ObjectMapper();
            try
            {
                mapper.configure(SerializationFeature.WRITE_ENUMS_USING_TO_STRING, true);
                return mapper.generateJsonSchema(clazz);
            }
            catch (JsonMappingException e)
            {
                LOGGER.error(e.getMessage(), e);
            }
            catch (IOException e)
            {
                LOGGER.error(e.getMessage(), e);
            }

            return null;
        }
    }
}