com.zhizus.forest.thrift.client.ServerInfo Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of thrift-client Show documentation
The newest version!
package com.zhizus.forest.thrift.client;

import com.google.common.base.Splitter;
import com.google.common.collect.Lists;
import com.google.common.collect.MapMaker;
import org.apache.commons.lang3.StringUtils;

import java.util.List;
import java.util.concurrent.ConcurrentMap;
import java.util.concurrent.atomic.AtomicInteger;

public final class ServerInfo {

    private static ConcurrentMap allInfos = new MapMaker().weakValues().makeMap();

    private static Splitter splitter = Splitter.on(':');

    private final String host;

    private final int port;

    private final int chance;


    private AtomicInteger activeCount = new AtomicInteger(0);

    private ServerInfo(String serverconfig) {
        List split = splitter.splitToList(serverconfig);
        this.host = split.get(0);
        this.port = Integer.parseInt(split.get(1));
        if (split.size() > 2)
            this.chance = Integer.parseInt(split.get(2));
        else
            this.chance = 1;
    }

    public int activeCountGet() {
        return activeCount.get();
    }

    public int incrementAndGet() {
        return activeCount.incrementAndGet();
    }

    public ServerInfo(String host, int port) {

        this.host = host;
        this.port = port;
        this.chance = 1;
    }

    public static final List ofs(String[] serverconfigs) {
        List l = Lists.newArrayListWithCapacity(serverconfigs.length);
        for (String string : serverconfigs) {
            l.add(of(string));
        }
        return l;
    }

    public static final List ofs(String serverconfigs) {
        return ofs(StringUtils.split(serverconfigs, ','));
    }

    public static final ServerInfo of(String serverconfig) {
        if (allInfos.containsKey(serverconfig))
            return allInfos.get(serverconfig);
        ServerInfo tsi = new ServerInfo(serverconfig);
        ServerInfo rt = allInfos.putIfAbsent(serverconfig, tsi);
        return rt == null ? tsi : rt;
    }

    public String getHost() {
        return host;
    }

    public int getPort() {
        return port;
    }

    public int getChance() {
        return chance;
    }

    @Override
    public String toString() {
        return "ServerInfo [host=" + host + ", port=" + port + ",chance=" + chance + "]";
    }

    @Override
    public int hashCode() {
        final int prime = 31;
        int result = 1;
        result = prime * result + ((host == null) ? 0 : host.hashCode());
        result = prime * result + port;
        result = prime * result + chance;
        return result;
    }

    @Override
    public boolean equals(Object obj) {
        if (this == obj)
            return true;
        if (obj == null)
            return false;
        if (getClass() != obj.getClass())
            return false;
        ServerInfo other = (ServerInfo) obj;
        if (chance != other.chance)
            return false;
        if (host == null) {
            if (other.host != null)
                return false;
        } else if (!host.equals(other.host))
            return false;
        if (port != other.port)
            return false;
        return true;
    }

}