com.trivago.mail.pigeon.web.data.process.QueueNewsletter.java Source code

Introduction

Here is the source code for com.trivago.mail.pigeon.web.data.process.QueueNewsletter.java

Source

/**
 * Copyright (C) 2011-2012 trivago GmbH <mario.mueller@trivago.com>, <christian.krause@trivago.com>
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.trivago.mail.pigeon.web.data.process;

import com.rabbitmq.client.AMQP;
import com.rabbitmq.client.Channel;
import com.rabbitmq.client.Connection;
import com.trivago.mail.pigeon.bean.*;
import com.trivago.mail.pigeon.json.MailTransport;
import com.trivago.mail.pigeon.queue.ConnectionPool;
import org.apache.log4j.Logger;
import org.neo4j.graphdb.Relationship;
import org.svenson.JSON;

import java.io.IOException;

public class QueueNewsletter {
    private static final String channelName = "messages";

    private static final Logger log = Logger.getLogger(QueueNewsletter.class);

    private TemplateProcessor templateProcessor;

    public void queueNewsletter(Mail mail, Sender sender, RecipientGroup group) {
        templateProcessor = new TemplateProcessor();

        final Iterable<Relationship> recipients = group.getRecipients();
        log.info("Pushing new newsletter (ID: " + mail.getId() + ") into queue.");

        Campaign campaign = mail.getCampaign();
        for (Relationship recipient : recipients) {
            Recipient recipientBean = new Recipient(recipient.getEndNode());
            queueNewsletter(mail, sender, recipientBean, campaign);
            mail.addRecipient(recipientBean);
        }
    }

    private void queueNewsletter(Mail mail, Sender sender, Recipient recipient, Campaign campaign) {
        Connection conn = ConnectionPool.getConnection();
        Channel channel = null;
        MailTransport transport = templateProcessor.processMail(mail, recipient, sender, campaign);

        if (transport == null) {
            log.warn(
                    "Template processor returned null instead of a mail transport object. This is probably a bug!");
            return;
        }

        if (transport.shouldAbortSending() && !transport.shouldEnforceSending()) {
            log.info("Skipped mail to " + transport.getTo() + " because transport aborted sending.");
            return;
        }

        String json = JSON.defaultJSON().forValue(transport);

        try {
            channel = conn.createChannel();
            channel.exchangeDeclare("mailpidgeon", "direct", true);
            channel.queueDeclare(channelName, true, false, false, null);
            channel.queueBind(channelName, "mailpidgeon", "mailpidgeon");

            byte[] messageBodyBytes = json.getBytes();
            channel.basicPublish("mailpidgeon", "mailpidgeon", null, messageBodyBytes);

        } catch (IOException e) {
            log.error(e);
        } finally {
            if (channel != null) {
                try {
                    channel.close();
                } catch (IOException e) {
                    log.error("Could not close channel", e);
                }
            }
        }
    }

    public int getProgress(long newsletterId) {
        Connection conn = ConnectionPool.getConnection();
        Channel channel = null;
        try {
            channel = conn.createChannel();
            AMQP.Queue.DeclareOk declareOk = channel.queueDeclarePassive(channelName);
            return declareOk.getMessageCount();
        } catch (Exception e) {
            log.error("Error while fetching progress", e);
        } finally {
            assert channel != null;
            try {
                channel.close();
            } catch (IOException e) {
                log.error("Could not close channel", e);
            }
        }
        return 0;
    }
}