net.oschina.thrift.client.support.pool.FixedTSConfigProvider Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of thrift-client-support Show documentation
thrift-client-support is an sdk for easy use thrift connect to the thrift server.
There is a newer version: 0.0.5
Show newest version
package net.oschina.thrift.client.support.pool;


import net.oschina.thrift.client.support.config.TSConfig;
import net.oschina.thrift.client.support.util.Utils;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.List;

/**
 * 固定Thrift 服务配置提供者， 提供初始化的时候固定配置
 *
 * @author Arvin
 * @time 2017/3/2 11:05
 */
public class FixedTSConfigProvider extends AbstractTSConfigProvider {

    public FixedTSConfigProvider(Collection configCol) {
        super(parseTsConfig(configCol));
    }

    /**
     * 格式:
     * hlocalhost:p9090:t5000:w2,hlocalhost:p9090:t5000:w2
     * 如果有多个可以使用英文逗号分割
     *
     * @param configString 配置
     */
    public FixedTSConfigProvider(String configString) {
        super(parseTsConfig(Arrays.asList(configString.split(" *, *"))));
    }

    /**
     *      * 通过集合进行配置每一项都是以下格式的字符串：
     * host:port:timeout:weight
     * 参数说明：
     * host:    必填，前缀h，服务主机地址
     * port:    必填，前缀p，端口
     * timeout: 选填，前缀t，默认是5000，该值有效范围是大于0的数字，如果小于1，则记为5000
     * weight:  选填，前缀w，默认是1，该值有效范围是大于0的数字，如果小于1，则记为1
     *
     * 实例： hlocalhost:p9090:t5000:w2
     * 
     *
     * @param configCol 配置列表
     */
    private static List parseTsConfig(Collection configCol) {
        List configList = new ArrayList();
        if (configCol == null || configCol.isEmpty()) {
            throw new IllegalArgumentException("没有指定Thrift服务器列表配置！");
        }
        for (String configItemStr : configCol) {
            // 通过字符串解析一个
            TSConfig config = parseTsConfig(configItemStr);
            if (config != null) {
                // weight有多少个就房多少个到list中
                for (int i = 0; i < config.getWeight(); ++i) {
                    configList.add(config);
                }
            }
        }
        return configList;
    }

    /**
     * 解析配置
     *
     * @param configItemStr 配置字段
     * @return
     */
    private static TSConfig parseTsConfig(String configItemStr) {
        String[] arr = configItemStr.split(" *: *");
        TSConfig config = new TSConfig();
        for (String item : arr) {
            if (Utils.isBlank(item)) {
                continue;
            }
            if (item.matches("(?i)h.*")) { // host
                config.setHost(item.substring(1));
            }
            if (item.matches("(?i)p.*")) { // post
                config.setPort(Integer.parseInt(item.substring(1)));
            }
            if (item.matches("(?i)t.*")) { // timeout
                int timeout = Integer.parseInt(item.substring(1));
                if (timeout > 0) {
                    config.setTimeout(timeout);
                }
            }
            if (item.matches("(?i)w.*")) { // weight
                int weight = Integer.parseInt(item.substring(1));
                if (weight > 0) {
                    config.setWeight(weight);
                }
            }

        }
        return config;
    }


    public static void main(String[] args) {
        parseTsConfig("h127.0.0.1:p9090:t5000:w1");
        parseTsConfig("h127.0.0.1:p9090:t-1:w1");
        parseTsConfig(":hlocalhost:t500:w2:p9090");
    }
}