io.streamnative.pulsar.handlers.kop.EndPoint Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of pulsar-protocol-handler-kafka Show documentation
Kafka on Pulsar implemented using Pulsar Protocol Handler
There is a newer version: 4.0.0.4
Show newest version
/**
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package io.streamnative.pulsar.handlers.kop;


import com.google.common.annotations.VisibleForTesting;
import java.net.InetSocketAddress;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
import lombok.Getter;
import org.apache.commons.lang3.StringUtils;
import org.apache.kafka.common.security.auth.SecurityProtocol;

/**
 * The endpoint that KoP binds.
 */
public class EndPoint {

    public static final String END_POINT_SEPARATOR = ",";
    private static final String PROTOCOL_MAP_SEPARATOR = ",";
    private static final String PROTOCOL_SEPARATOR = ":";

    @Getter
    private final String originalListener; // the original listener string, like "PLAINTEXT://localhost:9092"
    @Getter
    private final String listenerName;
    @Getter
    private final SecurityProtocol securityProtocol;
    @Getter
    private final String hostname;
    @Getter
    private final int port;
    @Getter
    private final boolean validInProtocolMap;
    @Getter
    private final boolean tlsEnabled;

    public EndPoint(final String listener, final Map protocolMap) {
        this.originalListener = listener;
        AdvertisedListener advertisedListener = AdvertisedListener.create(listener);
        this.listenerName = advertisedListener.getListenerName();
        this.hostname = advertisedListener.getHostname();
        this.port = advertisedListener.getPort();

        if (protocolMap == null || protocolMap.isEmpty()) {
            validInProtocolMap = false;
            this.securityProtocol = SecurityProtocol.forName(listenerName);
        } else {
            validInProtocolMap = true;
            this.securityProtocol = protocolMap.get(this.listenerName);
            if (this.securityProtocol == null) {
                throw new IllegalStateException(this.listenerName + " is not set in kafkaProtocolMap");
            }
        }
        this.tlsEnabled = (securityProtocol == SecurityProtocol.SSL) || (securityProtocol == SecurityProtocol.SASL_SSL);
    }

    public InetSocketAddress getInetAddress() {
        return new InetSocketAddress(hostname, port);
    }

    // listeners must be enable to be split into at least 1 token
    private static String[] getListenerArray(final String listeners) {
        if (StringUtils.isEmpty(listeners)) {
            throw new IllegalStateException("listeners is empty");
        }
        final String[] listenerArray = listeners.split(END_POINT_SEPARATOR);
        if (listenerArray.length == 0) {
            throw new IllegalStateException(listeners + " is split into 0 tokens by " + END_POINT_SEPARATOR);
        }
        return listenerArray;
    }

    @VisibleForTesting
    public static Map parseListeners(final String listeners) {
        return parseListeners(listeners, "");
    }

    private static Map parseListeners(final String listeners,
                                                        final Map protocolMap) {
        final Map endPointMap = new HashMap<>();
        for (String listener : getListenerArray(listeners)) {
            final EndPoint endPoint = new EndPoint(listener, protocolMap);
            if (endPointMap.containsKey(endPoint.listenerName)) {
                throw new IllegalStateException(
                        listeners + " has multiple listeners whose listenerName is " + endPoint.listenerName);
            } else {
                endPointMap.put(endPoint.listenerName, endPoint);
            }
        }
        return endPointMap;
    }

    @VisibleForTesting
    public static Map parseListeners(final String listeners, final String protocolMapString) {
        return parseListeners(listeners, parseProtocolMap(protocolMapString));
    }

    public static String findListener(final String listeners, final String name) {
        if (name == null) {
            return null;
        }
        for (String listener : getListenerArray(listeners)) {
            if (listener.contains(":") && listener.substring(0, listener.indexOf(":")).equals(name)) {
                return listener;
            }
        }
        throw new IllegalStateException("listener \"" + name + "\" doesn't exist in " + listeners);
    }

    public static String findFirstListener(String listeners) {
        return getListenerArray(listeners)[0];
    }

    public static EndPoint getPlainTextEndPoint(final String listeners) {
        for (String listener : listeners.split(END_POINT_SEPARATOR)) {
            if (listener.startsWith(SecurityProtocol.PLAINTEXT.name())
                    || listener.startsWith(SecurityProtocol.SASL_PLAINTEXT.name())) {
                return new EndPoint(listener, null);
            }
        }
        throw new IllegalStateException(listeners + " has no plain text endpoint");
    }

    public static EndPoint getSslEndPoint(final String listeners) {
        for (String listener : listeners.split(END_POINT_SEPARATOR)) {
            if (listener.startsWith(SecurityProtocol.SSL.name())
                    || listener.startsWith(SecurityProtocol.SASL_SSL.name())) {
                return new EndPoint(listener, null);
            }
        }
        throw new IllegalStateException(listeners + " has no ssl endpoint");
    }

    public static Map parseProtocolMap(final String kafkaProtocolMap) {
        if (StringUtils.isEmpty(kafkaProtocolMap)) {
            return Collections.emptyMap();
        }

        final Map protocolMap = new HashMap<>();

        for (String protocolSet : kafkaProtocolMap.split(PROTOCOL_MAP_SEPARATOR)) {
            String[] protocol = protocolSet.split(PROTOCOL_SEPARATOR);
            if (protocol.length != 2) {
                throw new IllegalStateException(
                        "wrong format for kafkaProtocolMap " + kafkaProtocolMap);
            }
            if (protocolMap.containsKey(protocol[0])) {
                throw new IllegalStateException(
                        kafkaProtocolMap + " has multiple listeners whose listenerName is " + protocol[0]);
            }
            protocolMap.put(protocol[0], SecurityProtocol.forName(protocol[1]));
        }
        return protocolMap;
    }

}