ConnectInfo.java

/**
 * Dianping.com Inc.
 * Copyright (c) 2003-2013 All Rights Reserved.
 */
package com.dianping.pigeon.remoting.invoker.domain;

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;

import com.dianping.pigeon.util.NetUtils;
import org.apache.commons.lang.builder.ReflectionToStringBuilder;
import org.apache.commons.lang.builder.ToStringStyle;
import org.springframework.util.CollectionUtils;

public class ConnectInfo {

    private String host;
    private int port;
    private ConcurrentMap<String, Integer> serviceNames;

    public ConnectInfo(String serviceName, String host, int port, int weight) {
        this(new ConcurrentHashMap<String, Integer>(), host, port);
        this.serviceNames.put(serviceName, weight);
    }

    private ConnectInfo(ConcurrentMap<String, Integer> serviceNames, String host, int port) {
        this.serviceNames = serviceNames;
        this.host = host;
        this.port = port;
    }

    public void addServiceNames(Map<String, Integer> serviceNames) {
        if (!CollectionUtils.isEmpty(serviceNames)) {
            this.serviceNames.putAll(serviceNames);
        }
    }

    public Map<String, Integer> getServiceNames() {
        return serviceNames;
    }

    /**
     * @return the connect
     */
    public String getConnect() {
        return NetUtils.toAddress(host, port);
    }

    /**
     * @return the host
     */
    public String getHost() {
        return host;
    }

    /**
     * @return the port
     */
    public int getPort() {
        return port;
    }

    public void setHost(String host) {
        this.host = host;
    }

    public void setPort(int port) {
        this.port = port;
    }

    @Override
    public String toString() {
        return ReflectionToStringBuilder.toString(this, ToStringStyle.SHORT_PREFIX_STYLE);
    }

}