cu.uci.uengine.amqp.SubmitsListener.java Source code

Introduction

Here is the source code for cu.uci.uengine.amqp.SubmitsListener.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package cu.uci.uengine.amqp;

import com.rabbitmq.client.Channel;
import cu.uci.uengine.Engine;
import cu.uci.uengine.adapters.SubmissionDTOToSubmissionAdapter;
import cu.uci.uengine.adapters.SubmissionToVerdictDTOAdapter;
import cu.uci.uengine.config.Config;
import cu.uci.uengine.model.Submission;
import cu.uci.uengine.model.dto.SubmissionDTO;
import cu.uci.uengine.model.dto.VerdictDTO;
import java.io.IOException;
import java.util.Arrays;
import javax.annotation.Resource;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.springframework.amqp.AmqpException;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.core.MessagePostProcessor;
import org.springframework.amqp.rabbit.core.ChannelAwareMessageListener;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.amqp.support.converter.JsonMessageConverter;
import org.springframework.amqp.support.converter.MessageConversionException;
import org.springframework.beans.factory.config.BeanDefinition;
import org.springframework.context.annotation.AnnotationConfigApplicationContext;
import org.springframework.context.annotation.Scope;
import org.springframework.stereotype.Component;

/**
 *
 * @author lan
 */
@Component
@Scope(BeanDefinition.SCOPE_PROTOTYPE)
public class SubmitsListener implements ChannelAwareMessageListener {

    static Log log = LogFactory.getLog(SubmitsListener.class.getName());
    public static boolean isDebugging;

    @Resource
    private Engine engine;

    @Resource
    private RabbitTemplate submitTemplate;

    @Resource
    private JsonMessageConverter jsonMessageConverter;

    @Override
    public void onMessage(Message message, Channel channel) throws Exception {

        message.getMessageProperties().setHeader("__TypeId__", "cu.uci.uengine.model.dto.SubmissionDTO");

        try {
            SubmissionDTO submissionDTO = (SubmissionDTO) jsonMessageConverter.fromMessage(message);

            Submission submit = new SubmissionDTOToSubmissionAdapter(submissionDTO);

            log.info(String.format("Working on submission: %s", submit.toString()));

            Submission result = engine.call(submit);

            log.info(String.format("Sending sid %s: %s. %s", result.getId(), result.getVerdict(),
                    result.getErrorMessage() == null ? "" : result.getErrorMessage()));

            VerdictDTO verdict = new SubmissionToVerdictDTOAdapter(result);

            submitTemplate.convertAndSend(verdict, new MessagePostProcessor() {
                @Override
                public Message postProcessMessage(Message message) throws AmqpException {
                    message.getMessageProperties().setHeader("__TypeId__", "UEngineVerdict");
                    return message;
                }
            });

            channel.basicAck(message.getMessageProperties().getDeliveryTag(), false);

        } catch (MessageConversionException messageConversionException) {
            channel.basicNack(message.getMessageProperties().getDeliveryTag(), false, false);
            log.error(String.format("Wrong submission format: %s", messageConversionException.getMessage()));
        } catch (Exception ex) {
            channel.basicNack(message.getMessageProperties().getDeliveryTag(), false, false);
            log.error(String.format("Error preocessing message with correlationId: %s",
                    message.getMessageProperties().getCorrelationId()));
        }
    }

    public static final void main(String[] args) throws IOException {
        AnnotationConfigApplicationContext appCtx = new AnnotationConfigApplicationContext(Config.class);

        //TODO: @Lan refactorizar esto, sobre todo quitar la variable isDebuguin esttica y hacer un mecanismo ms prctico.
        if (args.length > 0) {
            isDebugging = Arrays.asList(args).contains("debug");
            if (isDebugging) {
                log.info("DEBUGGING IS ENABLED!!");
            }
        }
    }

}