com.instaclustr.cassandra.service.kubernetes.KubernetesCassandraConfigReader Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of commons Show documentation

Common classes and utilities integrated with various projects

There is a newer version: 2.0.0

package com.instaclustr.cassandra.service.kubernetes;

import static com.instaclustr.cassandra.service.kubernetes.CassandraKubernetesHelper.CASSANDRA_CONFIG_NAME;
import static java.lang.String.format;

import java.io.StringReader;
import java.util.Optional;

import com.google.inject.Inject;
import com.google.inject.Provider;
import com.instaclustr.cassandra.service.CassandraConfigReader;
import com.instaclustr.kubernetes.ConfigMapReader;
import com.instaclustr.kubernetes.SecretReader;
import io.kubernetes.client.apis.CoreV1Api;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public final class KubernetesCassandraConfigReader implements CassandraConfigReader {

    private static final Logger logger = LoggerFactory.getLogger(KubernetesCassandraConfigReader.class);

    private static final String EMPTY_CONFIG = "datastax-java-driver {}";

    private final Provider coreV1ApiProvider;

    @Inject
    public KubernetesCassandraConfigReader(final Provider coreV1ApiProvider) {
        this.coreV1ApiProvider = coreV1ApiProvider;
    }

    @Override
    public StringReader getSecretReader() {

        final String sidecarSecretName = CassandraKubernetesHelper.getCassandraSidecarSecretName();

        try {
            Optional bytes = new SecretReader(coreV1ApiProvider).read(sidecarSecretName, CASSANDRA_CONFIG_NAME);

            if (!bytes.isPresent()) {
                throw new NullPointerException();
            }

            final String secret = new String(bytes.get());

            logger.debug(format("Read Secret %s on key %s:\n%s", sidecarSecretName, CASSANDRA_CONFIG_NAME, secret));

            return new StringReader(secret);
        } catch (final Exception ex) {
            logger.warn(format("There is not secret %s with key %s, returning empty config.", sidecarSecretName, CASSANDRA_CONFIG_NAME));
            return new StringReader(EMPTY_CONFIG);
        }
    }

    @Override
    public StringReader getConfigMapReader() {

        final String sidecarConfigMapName = CassandraKubernetesHelper.getCassandraSidecarConfigMapName();

        try {
            Optional data = new ConfigMapReader(coreV1ApiProvider).read(sidecarConfigMapName, CASSANDRA_CONFIG_NAME);

            if (!data.isPresent()) {
                throw new NullPointerException();
            }

            logger.debug(format("Read ConfigMap %s on key %s:\n%s", sidecarConfigMapName, CASSANDRA_CONFIG_NAME, data.get()));

            return new StringReader(data.get());
        } catch (final Exception ex) {
            logger.warn(format("There is not config map %s with key %s, returning empty config.", sidecarConfigMapName, CASSANDRA_CONFIG_NAME));
            return new StringReader(EMPTY_CONFIG);
        }
    }
}