terrastore.communication.NodeConfiguration.java Source code

Introduction

Here is the source code for terrastore.communication.NodeConfiguration.java

Source

/**
 * Copyright 2009 - 2011 Sergio Bossa (sergio.bossa@gmail.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 terrastore.communication;

import java.io.IOException;
import java.io.Serializable;
import java.net.InetAddress;
import java.net.NetworkInterface;
import java.util.Enumeration;
import java.util.Set;
import org.apache.commons.lang.builder.EqualsBuilder;
import org.apache.commons.lang.builder.HashCodeBuilder;
import org.msgpack.MessagePackable;
import org.msgpack.MessageTypeException;
import org.msgpack.MessageUnpackable;
import org.msgpack.Packer;
import org.msgpack.Unpacker;
import terrastore.util.collect.Sets;
import terrastore.util.io.MsgPackUtils;

/**
 * @author Sergio Bossa
 */
public class NodeConfiguration implements MessagePackable, MessageUnpackable, Serializable {

    private static final long serialVersionUID = 12345678901L;
    //
    private String name;
    private String nodeBindHost;
    private String nodePublishHosts;
    private int nodePort;
    private String httpHost;
    private int httpPort;

    public NodeConfiguration(String name, String nodeHost, int nodePort, String httpHost, int httpPort) {
        try {
            this.name = name;
            this.nodeBindHost = nodeHost;
            this.nodePublishHosts = InetAddress.getByName(nodeHost).isAnyLocalAddress() ? getPublishAddresses()
                    : nodeHost;
            this.nodePort = nodePort;
            this.httpHost = httpHost;
            this.httpPort = httpPort;
        } catch (Exception ex) {
            throw new IllegalStateException(ex.getMessage(), ex);
        }
    }

    public NodeConfiguration() {
    }

    public String getName() {
        return name;
    }

    public String getHttpHost() {
        return httpHost;
    }

    public int getHttpPort() {
        return httpPort;
    }

    public String getNodeBindHost() {
        return nodeBindHost;
    }

    public Set<String> getNodePublishHosts() {
        String[] hosts = nodePublishHosts.split(",");
        return Sets.hash(hosts);
    }

    public int getNodePort() {
        return nodePort;
    }

    @Override
    public void messagePack(Packer packer) throws IOException {
        MsgPackUtils.packString(packer, name);
        MsgPackUtils.packString(packer, nodeBindHost);
        MsgPackUtils.packString(packer, nodePublishHosts);
        MsgPackUtils.packInt(packer, nodePort);
        MsgPackUtils.packString(packer, httpHost);
        MsgPackUtils.packInt(packer, httpPort);
    }

    @Override
    public void messageUnpack(Unpacker unpacker) throws IOException, MessageTypeException {
        name = MsgPackUtils.unpackString(unpacker);
        nodeBindHost = MsgPackUtils.unpackString(unpacker);
        nodePublishHosts = MsgPackUtils.unpackString(unpacker);
        nodePort = MsgPackUtils.unpackInt(unpacker);
        httpHost = MsgPackUtils.unpackString(unpacker);
        httpPort = MsgPackUtils.unpackInt(unpacker);
    }

    @Override
    public boolean equals(Object obj) {
        if (obj instanceof NodeConfiguration) {
            NodeConfiguration other = (NodeConfiguration) obj;
            return new EqualsBuilder().append(this.name, other.name).isEquals();
        } else {
            return false;
        }
    }

    @Override
    public int hashCode() {
        return new HashCodeBuilder().append(this.name).toHashCode();
    }

    private String getPublishAddresses() throws Exception {
        Enumeration<NetworkInterface> nics = NetworkInterface.getNetworkInterfaces();
        StringBuilder result = new StringBuilder();
        while (nics != null && nics.hasMoreElements()) {
            NetworkInterface nic = nics.nextElement();
            if (nic.isUp() && !nic.isLoopback() && !nic.isVirtual() && nic.getInetAddresses().hasMoreElements()) {
                Enumeration<InetAddress> addresses = nic.getInetAddresses();
                while (addresses.hasMoreElements()) {
                    if (result.length() > 0) {
                        result.append(",");
                    }
                    result.append(addresses.nextElement().getHostAddress());
                }
            }
        }
        return result.toString();
    }

}