com.datastax.oss.driver.api.core.config.ProgrammaticDriverConfigLoaderBuilder Java Examples

/**
 * Add customizers that should be applied to the {@link ProgrammaticDriverConfigLoaderBuilder}.
 *
 * @param driverConfigLoaderBuilderCustomizers the customizers to add
 * @return this builder
 */
@SafeVarargs
public final CqlSessionCassandraConnectionBuilder addDriverConfigLoaderBuilderCustomizers(
		Consumer<? super ProgrammaticDriverConfigLoaderBuilder>... driverConfigLoaderBuilderCustomizers) {
	Objects.requireNonNull(driverConfigLoaderBuilderCustomizers,
			"'driverConfigLoaderBuilderCustomizers' must not be null");
	this.driverConfigLoaderBuilderCustomizers.addAll(Arrays.asList(driverConfigLoaderBuilderCustomizers));
	return this;
}
 

public List<Consumer<? super ProgrammaticDriverConfigLoaderBuilder>> getDriverConfigLoaderBuilderCustomizers() {
	return this.driverConfigLoaderBuilderCustomizers;
}
 

private CqlSession createSession(Cassandra cassandra) {
	ProgrammaticDriverConfigLoaderBuilder driverBuilder = DriverConfigLoader.programmaticBuilder()
			.withDuration(DefaultDriverOption.REQUEST_TIMEOUT, Duration.ofSeconds(30))
			.withDuration(DefaultDriverOption.CONNECTION_INIT_QUERY_TIMEOUT, Duration.ofSeconds(3));
	String username = getUsername();
	String password = getPassword();
	if (username != null && password != null) {
		driverBuilder.withString(DefaultDriverOption.AUTH_PROVIDER_USER_NAME, username)
				.withString(DefaultDriverOption.AUTH_PROVIDER_PASSWORD, password)
				.withClass(DefaultDriverOption.AUTH_PROVIDER_CLASS, PlainTextAuthProvider.class);
	}
	if (isSslEnabled()) {
		driverBuilder.withBoolean(DefaultDriverOption.SSL_HOSTNAME_VALIDATION, isHostnameValidation())
				.withClass(DefaultDriverOption.SSL_ENGINE_FACTORY_CLASS, DefaultSslEngineFactory.class);
		List<String> cipherSuites = getCipherSuites();
		if (!cipherSuites.isEmpty()) {
			driverBuilder.withStringList(DefaultDriverOption.SSL_CIPHER_SUITES, new ArrayList<>(cipherSuites));
		}
		Resource truststore = getTruststore();
		if (truststore != null) {
			driverBuilder.withString(DefaultDriverOption.SSL_TRUSTSTORE_PATH, getPath(truststore).toString());
		}
		String truststorePassword = getTruststorePassword();
		if (truststorePassword != null) {
			driverBuilder.withString(DefaultDriverOption.SSL_TRUSTSTORE_PASSWORD, truststorePassword);
		}
		Resource keystore = getKeystore();
		if (keystore != null) {
			driverBuilder.withString(DefaultDriverOption.SSL_KEYSTORE_PATH, getPath(keystore).toString());
		}
		String keystorePassword = getKeystorePassword();
		if (keystorePassword != null) {
			driverBuilder.withString(DefaultDriverOption.SSL_KEYSTORE_PASSWORD, keystorePassword);
		}
	}
	List<Consumer<? super ProgrammaticDriverConfigLoaderBuilder>> driverConfigLoaderBuilderCustomizers =
			getDriverConfigLoaderBuilderCustomizers();
	driverConfigLoaderBuilderCustomizers.forEach(customizer -> customizer.accept(driverBuilder));
	int port = cassandra.getPort();
	int sslPort = cassandra.getSslPort();
	InetSocketAddress contactPoint = new InetSocketAddress(cassandra.getAddress(),
			(isSslEnabled() && sslPort != -1) ? sslPort : port);
	CqlSessionBuilder sessionBuilder = CqlSession.builder().addContactPoint(contactPoint)
			.withConfigLoader(driverBuilder.build());
	String localDataCenter = getLocalDataCenter();
	if (localDataCenter != null) {
		sessionBuilder.withLocalDatacenter(localDataCenter);
	}
	List<TypeCodec<?>> typeCodecs = getTypeCodecs();
	if (!typeCodecs.isEmpty()) {
		sessionBuilder.addTypeCodecs(typeCodecs.toArray(new TypeCodec[0]));
	}
	List<Consumer<? super CqlSessionBuilder>> sessionBuilderCustomizers = getSessionBuilderCustomizers();
	sessionBuilderCustomizers.forEach(customizer -> customizer.accept(sessionBuilder));
	return sessionBuilder.build();
}
 

private CqlSession initialiseSession() throws PermanentBackendException {
    Configuration configuration = getStorageConfig();
    List<InetSocketAddress> contactPoints;
    String[] hostnames = configuration.get(STORAGE_HOSTS);
    int port = configuration.has(STORAGE_PORT) ? configuration.get(STORAGE_PORT) : DEFAULT_PORT;
    try {
        contactPoints = Array.of(hostnames)
                .map(hostName -> hostName.split(":"))
                .map(array -> Tuple.of(array[0], array.length == 2 ? Integer.parseInt(array[1]) : port))
                .map(tuple -> new InetSocketAddress(tuple._1, tuple._2))
                .toJavaList();
    } catch (SecurityException | ArrayIndexOutOfBoundsException | NumberFormatException e) {
        throw new PermanentBackendException("Error initialising cluster contact points", e);
    }

    CqlSessionBuilder builder = CqlSession.builder()
            .addContactPoints(contactPoints)
            .withLocalDatacenter(configuration.get(LOCAL_DATACENTER));

    ProgrammaticDriverConfigLoaderBuilder configLoaderBuilder = DriverConfigLoader.programmaticBuilder();

    configLoaderBuilder.withString(DefaultDriverOption.SESSION_NAME, configuration.get(SESSION_NAME));
    configLoaderBuilder.withDuration(DefaultDriverOption.REQUEST_TIMEOUT, configuration.get(CONNECTION_TIMEOUT));

    if (configuration.get(PROTOCOL_VERSION) != 0) {
        configLoaderBuilder.withInt(DefaultDriverOption.PROTOCOL_VERSION, configuration.get(PROTOCOL_VERSION));
    }

    if (configuration.has(AUTH_USERNAME) && configuration.has(AUTH_PASSWORD)) {
        configLoaderBuilder
                .withClass(DefaultDriverOption.AUTH_PROVIDER_CLASS, PlainTextAuthProvider.class)
                .withString(DefaultDriverOption.AUTH_PROVIDER_USER_NAME, configuration.get(AUTH_USERNAME))
                .withString(DefaultDriverOption.AUTH_PROVIDER_PASSWORD, configuration.get(AUTH_PASSWORD));
    }

    if (configuration.get(SSL_ENABLED)) {
        configLoaderBuilder
                .withClass(DefaultDriverOption.SSL_ENGINE_FACTORY_CLASS, DefaultSslEngineFactory.class)
                .withString(DefaultDriverOption.SSL_TRUSTSTORE_PATH, configuration.get(SSL_TRUSTSTORE_LOCATION))
                .withString(DefaultDriverOption.SSL_TRUSTSTORE_PASSWORD, configuration.get(SSL_TRUSTSTORE_PASSWORD));
    }

    configLoaderBuilder.withInt(DefaultDriverOption.CONNECTION_POOL_LOCAL_SIZE, configuration.get(LOCAL_MAX_CONNECTIONS_PER_HOST));
    configLoaderBuilder.withInt(DefaultDriverOption.CONNECTION_POOL_REMOTE_SIZE, configuration.get(REMOTE_MAX_CONNECTIONS_PER_HOST));
    configLoaderBuilder.withInt(DefaultDriverOption.CONNECTION_MAX_REQUESTS, configuration.get(MAX_REQUESTS_PER_CONNECTION));

    // Keep to 0 for the time being: https://groups.google.com/a/lists.datastax.com/forum/#!topic/java-driver-user/Bc0gQuOVVL0
    // Ideally we want to batch all tables initialisations to happen together when opening a new keyspace
    configLoaderBuilder.withInt(DefaultDriverOption.METADATA_SCHEMA_WINDOW, 0);

    // The following sets the size of Netty ThreadPool executor used by Cassandra driver:
    // https://docs.datastax.com/en/developer/java-driver/4.0/manual/core/async/#threading-model
    configLoaderBuilder.withInt(DefaultDriverOption.NETTY_IO_SIZE, 0); // size of threadpool scales with number of available CPUs when set to 0
    configLoaderBuilder.withInt(DefaultDriverOption.NETTY_ADMIN_SIZE, 0); // size of threadpool scales with number of available CPUs when set to 0


    // Keep the following values to 0 so that when we close the session we don't have to wait for the
    // so called "quiet period", setting this to a different value will slow down Graph.close()
    configLoaderBuilder.withInt(DefaultDriverOption.NETTY_ADMIN_SHUTDOWN_QUIET_PERIOD, 0);
    configLoaderBuilder.withInt(DefaultDriverOption.NETTY_IO_SHUTDOWN_QUIET_PERIOD, 0);

    builder.withConfigLoader(configLoaderBuilder.build());

    return builder.build();
}