io.latent.storm.rabbitmq.config.ConfigAvailableHosts Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of storm-rabbitmq Show documentation

A library of tools for interacting with RabbitMQ from Storm.

There is a newer version: 1.0.0

package io.latent.storm.rabbitmq.config;

import java.io.Serializable;
import java.util.Map;
import java.util.Map.Entry;
import java.util.TreeMap;

import com.rabbitmq.client.Address;

/**
 * Simple configuration class to allow users to specify a set hosts to connect
 * to for high availability purposes.
 *
 */
public class ConfigAvailableHosts implements Serializable {

    private static final long serialVersionUID = -7444594758428554141L;

    private static final String HOST_DELIMINITER = "|";

    /**
     * {@link TreeMap} of host configurations (order may matter...).
     */
    private final Map hostsMap = new TreeMap();

    /**
     * 
     * @return The map of hostname to port we'll use to connect to
     */
    public Map getHostsMap() {
        return hostsMap;
    }

    /**
     * 
     * @return Whether or not there are any configured high availability hosts
     *         in this config
     */
    public boolean isEmpty() {
        return hostsMap.isEmpty();
    }

    /**
     * 
     * @return The {@link Map} of RabbitMQ hosts, converted to the necessary
     *         {@link Address} array
     */
    public Address[] toAddresses() {
        final Address[] addresses = new Address[hostsMap.size()];
        int i = 0;
        for (final Entry entry : hostsMap.entrySet()) {
            if (entry.getKey() != null) {
                addresses[i++] = entry.getValue() == null ? new Address(entry.getKey()) : new Address(entry.getKey(), entry.getValue());
            }
        }
        return addresses;
    }

    @Override
    public String toString() {
        final StringBuilder builder = new StringBuilder();
        boolean first = true;
        for (final String host : hostsMap.keySet()) {
            final Integer port = hostsMap.get(host);
            if (!first) {
                builder.append(HOST_DELIMINITER);
            } else {
                first = false;
            }
            builder.append(host);
            builder.append(port != null ? ":" + port : "");
        }
        return builder.toString();
    }

    public static ConfigAvailableHosts fromString(final String serialzed) {
        final ConfigAvailableHosts value = new ConfigAvailableHosts();
        final String[] hosts = serialzed.split("[" + HOST_DELIMINITER + "]");
        for (final String host : hosts) {
            if (!host.isEmpty()) {
                String[] brokenUp = host.split(":");
                value.getHostsMap().put(brokenUp[0], brokenUp.length == 2 ? Integer.parseInt(brokenUp[1]) : null);
            }
        }
        return value;
    }
}