org.trellisldp.jms.JmsPublisher Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of trellis-jms Show documentation
The core components for a Trellis linked data server
There is a newer version: 0.17.2
/*
 * 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 org.trellisldp.jms;

import static java.util.Objects.nonNull;
import static java.util.Objects.requireNonNull;
import static java.util.Optional.of;
import static java.util.ServiceLoader.load;
import static javax.jms.Session.AUTO_ACKNOWLEDGE;
import static org.apache.tamaya.ConfigurationProvider.getConfiguration;
import static org.slf4j.LoggerFactory.getLogger;

import java.util.Iterator;
import java.util.ServiceLoader;

import javax.inject.Inject;
import javax.jms.Connection;
import javax.jms.JMSException;
import javax.jms.Message;
import javax.jms.MessageProducer;
import javax.jms.Session;

import org.apache.activemq.ActiveMQConnectionFactory;
import org.apache.tamaya.Configuration;
import org.slf4j.Logger;
import org.trellisldp.api.ActivityStreamService;
import org.trellisldp.api.Event;
import org.trellisldp.api.EventService;
import org.trellisldp.api.RuntimeTrellisException;

/**
 * A JMS message producer capable of publishing messages to a JMS broker such as ActiveMQ.
 *
 * @author acoburn
 */
public class JmsPublisher implements EventService {

    /** The configuration key controlling the JMS queue name. **/
    public static final String CONFIG_JMS_QUEUE_NAME = "trellis.jms.queue";

    /** The configuration key controlling the JMS broker URL. **/
    public static final String CONFIG_JMS_URL = "trellis.jms.url";

    /** The configuration key controlling the JMS username. **/
    public static final String CONFIG_JMS_USERNAME = "trellis.jms.username";

    /** THe configuration key controlling the JMS password. **/
    public static final String CONFIG_JMS_PASSWORD = "trellis.jms.password";

    private static final Logger LOGGER = getLogger(JmsPublisher.class);
    private static final ActivityStreamService service = of(load(ActivityStreamService.class))
        .map(ServiceLoader::iterator).filter(Iterator::hasNext).map(Iterator::next)
        .orElseThrow(() -> new RuntimeTrellisException("No ActivityStream service available!"));

    private final MessageProducer producer;
    private final Session session;

    /**
     * Create a new JMS Publisher.
     * @throws JMSException when there is a connection error
     */
    @Inject
    public JmsPublisher() throws JMSException {
        this(getConfiguration());
    }

    private JmsPublisher(final Configuration config) throws JMSException {
        this(buildJmsConnection(config).createSession(false, AUTO_ACKNOWLEDGE), config.get(CONFIG_JMS_QUEUE_NAME));
    }

    /**
     * Create a new JMS Publisher.
     * @param conn the connection
     * @throws JMSException when there is a connection error
     */
    public JmsPublisher(final Connection conn) throws JMSException {
        this(conn.createSession(false, AUTO_ACKNOWLEDGE), getConfiguration().get(CONFIG_JMS_QUEUE_NAME));
    }

    /**
     * Create a new JMS Publisher.
     * @param session the JMS session
     * @param queueName the name of the queue
     * @throws JMSException when there is a connection error
     */
    public JmsPublisher(final Session session, final String queueName) throws JMSException {
        requireNonNull(queueName, "JMS Queue name may not be null!");
        this.session = requireNonNull(session, "JMS Session may not be null!");
        this.producer = session.createProducer(session.createQueue(queueName));
    }

    @Override
    public void emit(final Event event) {
        requireNonNull(event, "Cannot emit a null event!");

        service.serialize(event).ifPresent(json -> {
            try {
                final Message message = session.createTextMessage(json);
                message.setStringProperty("Content-Type", "application/ld+json");
                producer.send(message);
            } catch (final JMSException ex) {
                LOGGER.error("Error writing to broker: {}", ex.getMessage());
            }
        });
    }

    private static Connection buildJmsConnection(final Configuration config) throws JMSException {
        final ActiveMQConnectionFactory factory = new ActiveMQConnectionFactory(
                config.get(CONFIG_JMS_URL));
        if (nonNull(config.get(CONFIG_JMS_USERNAME)) && nonNull(config.get(CONFIG_JMS_PASSWORD))) {
            factory.setUserName(config.get(CONFIG_JMS_USERNAME));
            factory.setPassword(config.get(CONFIG_JMS_PASSWORD));
        }
        return factory.createConnection();
    }
}