com.nesscomputing.jms.ActiveMQConnectionFactoryProvider.java Source code

Introduction

Here is the source code for com.nesscomputing.jms.ActiveMQConnectionFactoryProvider.java

Source

/**
 * Copyright (C) 2012 Ness Computing, Inc.
 *
 * 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.nesscomputing.jms;

import static java.lang.String.format;

import java.lang.annotation.Annotation;
import java.net.URI;
import java.util.Collections;
import java.util.Set;

import javax.annotation.Nonnull;
import javax.jms.ConnectionFactory;

import com.google.common.base.Preconditions;
import com.google.common.collect.Sets;
import com.google.inject.Binding;
import com.google.inject.Inject;
import com.google.inject.Injector;
import com.google.inject.Key;
import com.google.inject.Provider;
import com.google.inject.ProvisionException;
import com.google.inject.TypeLiteral;

import org.apache.activemq.ActiveMQConnectionFactory;
import org.apache.commons.lang3.StringUtils;

/**
 * Provides a Connection factory to retrieve JMS connections from ActiveMQ.
 */
final class ActiveMQConnectionFactoryProvider implements Provider<ConnectionFactory> {
    private final JmsConfig jmsConfig;
    private final Annotation annotation;
    private Set<JmsUriInterceptor> interceptors = Collections.emptySet();
    private final String connectionName;

    ActiveMQConnectionFactoryProvider(@Nonnull final JmsConfig jmsConfig, @Nonnull final String connectionName,
            @Nonnull final Annotation annotation) {
        this.connectionName = connectionName;
        Preconditions.checkArgument(annotation != null, "no binding annotation");

        this.jmsConfig = jmsConfig;
        this.annotation = annotation;
    }

    @Inject
    public void injectInjector(Injector injector) {
        final TypeLiteral<Set<JmsUriInterceptor>> type = new TypeLiteral<Set<JmsUriInterceptor>>() {
        };

        Binding<Set<JmsUriInterceptor>> binding = injector.getExistingBinding(Key.get(type, annotation));

        if (binding != null) {
            interceptors = Sets.union(interceptors, binding.getProvider().get());
        }

        binding = injector.getExistingBinding(Key.get(type));

        if (binding != null) {
            interceptors = Sets.union(interceptors, binding.getProvider().get());
        }
    }

    @Override
    public ConnectionFactory get() {
        String jmsUriStr = jmsConfig.getJmsConnectionUrlFormat();

        if (StringUtils.isBlank(jmsUriStr)) {
            throw new ProvisionException(
                    format("JMS URI for %s can not be blank!", annotation == null ? "<default>" : annotation));
        }

        for (final JmsUriInterceptor interceptor : interceptors) {
            jmsUriStr = interceptor.transform(connectionName, jmsUriStr);
        }

        final URI jmsUri = URI.create(jmsUriStr);

        return new ActiveMQConnectionFactory(jmsUri);
    }
}