io.quarkus.kafka.client.runtime.KafkaBindingConverter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of quarkus-kafka-client Show documentation

Connect to Apache Kafka with its native API

There is a newer version: 3.17.5

package io.quarkus.kafka.client.runtime;

import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Optional;

import io.quarkus.kubernetes.service.binding.runtime.ServiceBinding;
import io.quarkus.kubernetes.service.binding.runtime.ServiceBindingConfigSource;
import io.quarkus.kubernetes.service.binding.runtime.ServiceBindingConverter;

public class KafkaBindingConverter implements ServiceBindingConverter {

    @Override
    public Optional convert(List serviceBindings) {
        Optional matchingByType = ServiceBinding.singleMatchingByType("kafka", serviceBindings);
        if (!matchingByType.isPresent()) {
            return Optional.empty();
        }

        Map properties = new HashMap<>();
        ServiceBinding binding = matchingByType.get();

        String bootstrapServers = binding.getProperties().get("bootstrapServers");
        if (bootstrapServers == null) {
            bootstrapServers = binding.getProperties().get("bootstrap-servers");
        }
        if (bootstrapServers != null) {
            properties.put("kafka.bootstrap.servers", bootstrapServers);
        }

        String securityProtocol = binding.getProperties().get("securityProtocol");
        if (securityProtocol != null) {
            properties.put("kafka.security.protocol", securityProtocol);
        }

        String saslMechanism = binding.getProperties().get("saslMechanism");
        if (saslMechanism != null) {
            properties.put("kafka.sasl.mechanism", saslMechanism);
        }
        String user = binding.getProperties().get("user");
        String password = binding.getProperties().get("password");
        if ((user != null) && (password != null)) {
            if ("PLAIN".equals(saslMechanism)) {
                properties.put("kafka.sasl.jaas.config",
                        String.format(
                                "org.apache.kafka.common.security.plain.PlainLoginModule required username='%s' password='%s';",
                                user, password));
            }

            if ("SCRAM-SHA-512".equals(saslMechanism) || "SCRAM-SHA-256".equals(saslMechanism)) {
                properties.put("kafka.sasl.jaas.config",
                        String.format(
                                "org.apache.kafka.common.security.scram.ScramLoginModule required username='%s' password='%s';",
                                user, password));
            }
        }

        return Optional.of(new ServiceBindingConfigSource("kafka-k8s-service-binding-source", properties));
    }
}