top.jfunc.http.holder.DefaultPhpUrlHolder Maven / Gradle / Ivy

Go to download
package top.jfunc.http.holder;

import top.jfunc.http.util.ParamUtil;
import top.jfunc.http.base.Protocol;

import static top.jfunc.common.utils.StrUtil.*;

/**
 * 适合不知道全路径，知道各个部分的情况下，或者需要知道一个URL中的各个部分，可以不需要考虑诸如一下的一些细节：路径参数处理、查询参数处理、要不要加一个/等繁琐的细节
 * 相反则不太适合用此类，因为{@link DefaultPhpUrlHolder#setUrl(String)}会切割计算各个部分，会有性能损耗
 * @author xiongshiyan at 2019/6/28 , contact me with email [email protected] or phone 15208384257
 */
public class DefaultPhpUrlHolder extends DefaultUrlHolder implements PhpUrlHolder {
    private static final int UNSIGNED_PORT = -1;
    /**
     * 下面三个参数可能都没有
     */
    private Protocol protocol;
    private String host;
    private int port = UNSIGNED_PORT;

    /**
     * 路径，不包括query参数的
     */
    private String path = SLASH;

    /**
     * @return 计算后的url
     */
    @Override
    public String getUrl() {
        //说明不是完整的路径
        if(null == protocol){
            return this.path;
        }

        //有协议说明肯定是完整路径
        if(null == host){
            throw new IllegalArgumentException("host 未指定");
        }

        //如果port未指定
        if(UNSIGNED_PORT == port){
            port = protocol.getDefaultPort();
        }

        //拼装url
        String url = protocol.name().toLowerCase() + COLON_SPLASH + host + COLON + port;
        String p = (path.startsWith(SLASH) ? path : (SLASH + path));
        return url + p;
    }

    @Override
    public PhpUrlHolder setUrl(String destination) {
        // protocol
        int ndx = destination.indexOf(COLON_SPLASH);
        if (ndx != -1) {
            //获取协议，如果协议名不正确，直接抛出异常
            this.protocol = ParamUtil.httpProtocol(destination);
            destination = destination.substring(ndx + 3);
        }
        // host
        ndx = destination.indexOf(SLASH);

        if (ndx == -1) {
            ndx = destination.length();
        }

        if (ndx != 0) {

            String hostToSet = destination.substring(0, ndx);
            destination = destination.substring(ndx);

            // port

            ndx = hostToSet.indexOf(COLON);

            if (ndx != -1) {
                port = Integer.parseInt(hostToSet.substring(ndx + 1));
                hostToSet = hostToSet.substring(0, ndx);
            }

            this.host = hostToSet;
        }

        // path + query
        this.path = destination;

        //处理query参数
        ndx = destination.indexOf(QUESTION_MARK);
        if(-1 != ndx){
            this.path = destination.substring(0, ndx);
            //k1=v1&k2=v2
            destination = destination.substring(ndx+1);
            String[] kvs = destination.split(AND);
            ParamHolder queryParamHolder = queryParamHolder();
            for (String kv : kvs) {
                String[] split = kv.split(EQUALS);
                queryParamHolder.add(split[0] , split[1]);
            }
        }
        return this;
    }

    @Override
    public PhpUrlHolder protocol(Protocol protocol) {
        this.protocol = protocol;
        return this;
    }

    @Override
    public Protocol protocol() {
        return protocol;
    }

    @Override
    public PhpUrlHolder host(String host) {
        this.host = host;
        return this;
    }

    @Override
    public String host() {
        return host;
    }

    @Override
    public PhpUrlHolder port(int port) {
        this.port = port;
        return this;
    }

    @Override
    public int port() {
        return port;
    }

    @Override
    public PhpUrlHolder path(String path , String... paths) {
        this.path = path;
        if(null != paths && paths.length > 0){
            for (String p : paths) {
                this.path = ParamUtil.concatUrlIfNecessary(this.path , p);
            }
        }
        return this;
    }

    @Override
    public String path() {
        return path;
    }
}