org.mariadb.jdbc.HostAddress Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of analytics-plugin Show documentation
Kill Bill Analytics plugin
There is a newer version: 8.1.2
package org.mariadb.jdbc;

import org.mariadb.jdbc.internal.common.ParameterConstant;
import org.mariadb.jdbc.internal.common.UrlHAMode;

import java.util.ArrayList;
import java.util.List;

public class HostAddress {
    public String host;
    public int port;
    public String type=null;


    /**
     * parse - parse server addresses from the URL fragment
     * @param spec - list of endpoints in one of the forms
     *    1 - host1,....,hostN:port (missing port default to MySQL default 3306
     *    2 - host:port,...,host:port
     * @param haMode High availability mode
     * @return   parsed endpoints
     */
    public static List parse(String spec, UrlHAMode haMode) {
        if (spec == null) throw new IllegalArgumentException("Invalid connection URL, host address must not be empty ");
        String[] tokens = spec.trim().split(",");
        List arr = new ArrayList<>(tokens.length);

        for (int i=0; i < tokens.length; i++) {
            if (tokens[i].startsWith("address=")) {
                arr.add(parseParameterHostAddress(tokens[i]));
            } else {
                arr.add(parseSimpleHostAddress(tokens[i]));
            }
        }

        int  defaultPort = arr.get(arr.size()-1).port;
        if (defaultPort == 0) {
            defaultPort = 3306;
        }

        for (int i = 0; i < arr.size(); i++) {
            if (haMode == UrlHAMode.REPLICATION) {
                if (i==0 && arr.get(i).type == null) arr.get(i).type = ParameterConstant.TYPE_MASTER;
                else if (i!=0 && arr.get(i).type == null) arr.get(i).type = ParameterConstant.TYPE_SLAVE;
            }
            if (arr.get(i).port == 0) {
                arr.get(i).port = defaultPort;
            }
        }
        return arr;
    }
    public HostAddress() {}

    public HostAddress(String host, int port) {
        this.host = host;
        this.port = port;
        this.type = ParameterConstant.TYPE_MASTER;
    }

    public HostAddress(String host, int port, String type) {
        this.host = host;
        this.port = port;
        this.type = type;
    }

    static HostAddress parseSimpleHostAddress(String s) {
        HostAddress result = new HostAddress();
        if (s.startsWith("[")) {
    		/* IPv6 addresses in URLs are enclosed in square brackets */
            int ind = s.indexOf(']');
            result.host = s.substring(1,ind);
            if (ind != (s.length() -1) && s.charAt(ind + 1) == ':') {
                result.port = Integer.parseInt(s.substring(ind+2));
            }
        } else if (s.contains(":")) {
        	  /* Parse host:port */
            String[] hostPort = s.split(":");
            result.host = hostPort[0];
            result.port = Integer.parseInt(hostPort[1]);
        } else {
        	  /* Just host name is given */
            result.host = s;
        }
        return result;
    }
    static HostAddress parseParameterHostAddress(String s) {
        HostAddress result = new HostAddress();
        String[] array = s.split("(?=\\()|(?<=\\))");
        for (int i=1;i< array.length;i++) {
            String[] token = array[i].replace("(","").replace(")","").trim().split("=");
            if (token.length != 2) throw new IllegalArgumentException("Invalid connection URL, expected key=value pairs, found " + array[i]);
            String key = token[0].toLowerCase();
            String value = token[1].toLowerCase();
            if (key.equals("host")) {
                result.host=value.replace("[","").replace("]", "");
            } else if (key.equals("port")) {
                result.port=Integer.parseInt(value);
            } else if (key.equals("type")) {
                if (value.equals(ParameterConstant.TYPE_MASTER) || value.equals(ParameterConstant.TYPE_SLAVE))result.type=value;
            }
        }
        return result;
    }

    public static String toString(List addrs) {
        String s="";
        for(int i=0; i < addrs.size(); i++) {
            if (addrs.get(i).type != null) {
                s+="address=(host="+addrs.get(i).host+")(port="+addrs.get(i).port+")(type="+addrs.get(i).type+")";
            } else {
                boolean isIPv6 = addrs.get(i).host != null && addrs.get(i).host.contains(":");
                String host = (isIPv6) ? ("[" + addrs.get(i).host + "]") : addrs.get(i).host;
                s += host + ":" + addrs.get(i).port;
            }
            if (i < addrs.size() -1) s += ",";
        }
        return s;
    }

    public static String toString(HostAddress[] addrs) {
        String s="";
        for(int i=0; i < addrs.length; i++) {
            if (addrs[i].type != null) {
                s+="address=(host="+addrs[i].host+")(port="+addrs[i].port+")(type="+addrs[i].type+")";
            } else {
                boolean isIPv6 = addrs[i].host != null && addrs[i].host.contains(":");
                String host = (isIPv6)?("[" + addrs[i].host + "]"):addrs[i].host;
                s += host + ":" + addrs[i].port;
            }
            if (i < addrs.length -1) s += ",";
        }
        return s;
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;

        HostAddress that = (HostAddress) o;

        if (port != that.port) return false;
        if (host != null ? !host.equals(that.host) : that.host != null) return false;
        return !(type != null ? !type.equals(that.type) : that.type != null);

    }

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

    @Override
    public String toString() {
        return "HostAddress{" +
                "host='" + host + '\'' +
                ", port=" + port +
                ", type='" + type + '\'' +
                '}';
    }
}