io.dropwizard.cassandra.CassandraBundle Maven / Gradle / Ivy

Go to download
package io.dropwizard.cassandra;

import brave.Tracing;
import com.datastax.oss.driver.api.core.CqlSession;
import io.dropwizard.core.Configuration;
import io.dropwizard.core.ConfiguredBundle;
import io.dropwizard.core.setup.Bootstrap;
import io.dropwizard.core.setup.Environment;

import javax.annotation.Nullable;

import static java.util.Objects.requireNonNull;

public abstract class CassandraBundle implements ConfiguredBundle, CassandraConfiguration {
    @Nullable
    private CqlSession session;

    @Override
    public void initialize(final Bootstrap bootstrap) {
        // do nothing
    }

    @Override
    public void run(final T configuration, final Environment environment) throws Exception {
        final CassandraFactory cassandraFactory = requireNonNull(getCassandraFactory(configuration));

        final Tracing tracing = Tracing.current();

        this.session = cassandraFactory.build(environment.metrics(), environment.lifecycle(), environment.healthChecks(), tracing);
    }

    public abstract CassandraFactory getCassandraFactory(T configuration);

    public CqlSession getSession() {
        return requireNonNull(session);
    }
}