com.scoperetail.commons.cassandra.config.apache.ApacheCassandraConfig Maven / Gradle / Ivy
The newest version!
package com.scoperetail.commons.cassandra.config.apache;
/*-
* *****
* commons-cassandra-config
* -----
* Copyright (C) 2018 - 2023 Scope Retail Systems Inc.
* -----
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
* THE SOFTWARE.
* =====
*/
import com.datastax.oss.driver.api.core.CqlSessionBuilder;
import com.datastax.oss.driver.api.core.config.DefaultDriverOption;
import com.datastax.oss.driver.api.core.config.DriverConfigLoader;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.data.cassandra.config.AbstractCassandraConfiguration;
import org.springframework.data.cassandra.config.CqlSessionFactoryBean;
import org.springframework.data.cassandra.config.SessionBuilderConfigurer;
import javax.net.ssl.KeyManagerFactory;
import javax.net.ssl.SSLContext;
import java.io.FileInputStream;
import java.io.IOException;
import java.security.GeneralSecurityException;
import java.security.KeyStore;
import java.time.Duration;
@Configuration
@ConditionalOnProperty(value = "db.type", havingValue = "Apache-Cassandra", matchIfMissing = false)
public class ApacheCassandraConfig extends AbstractCassandraConfiguration {
@Value("${spring.data.cassandra.keyspace-name}")
private String keySpace;
@Value("${spring.data.cassandra.contact-points}")
private String contactPoints;
@Value("${spring.data.cassandra.local-datacenter}")
private String localDatacenter;
@Value("${spring.data.cassandra.port}")
private int port;
@Value("${spring.data.cassandra.username}")
private String username;
@Value("${spring.data.cassandra.password}")
private String password;
@Value("${datastax-java-driver.basic.request.timeout:2000}")
private long requestTimeout;
@Value("${datastax-java-driver.advanced.metadata.schema.request-timeout:2000}")
private long schemaRequestTimeout;
@Value("${datastax-java-driver.advanced.connection.init-query-timeout:5000}")
private long initQueryTimeout;
@Value("${datastax-java-driver.advanced.connection.pool.local.size:1}")
private int connectionPoolLocalSize;
@Value("${datastax-java-driver.advanced.connection.pool.remote.size:1}")
private int connectionPoolRemoteSize;
@Value("${datastax-java-driver.advanced.connection.max-requests-per-connection:1024}")
private int maxRequestsPerConnection;
@Value("${db.cassandra.ssl.enabled:false}")
private Boolean isSslEnabled;
@Value("${db.cassandra.truststore.location:}")
private String truststoreLocation;
@Value("${db.cassandra.truststore.password:}")
private String truststorePassword;
@Override
protected String getKeyspaceName() {
return keySpace;
}
@Bean
@Override
public CqlSessionFactoryBean cassandraSession() {
final CqlSessionFactoryBean cassandraSession = new CqlSessionFactoryBean();
cassandraSession.setContactPoints(contactPoints);
cassandraSession.setKeyspaceName(this.getKeyspaceName());
cassandraSession.setLocalDatacenter(localDatacenter);
cassandraSession.setPort(port);
cassandraSession.setUsername(username);
cassandraSession.setPassword(password);
cassandraSession.setSessionBuilderConfigurer(getSessionBuilderConfigurer());
return cassandraSession;
}
@Override
protected SessionBuilderConfigurer getSessionBuilderConfigurer() {
return new SessionBuilderConfigurer() {
@Override
public CqlSessionBuilder configure(final CqlSessionBuilder cqlSessionBuilder) {
cqlSessionBuilder.withConfigLoader(
DriverConfigLoader.programmaticBuilder()
.withDuration(
DefaultDriverOption.REQUEST_TIMEOUT, Duration.ofMillis(requestTimeout))
.withDuration(
DefaultDriverOption.METADATA_SCHEMA_REQUEST_TIMEOUT,
Duration.ofMillis(schemaRequestTimeout))
.withDuration(
DefaultDriverOption.CONNECTION_INIT_QUERY_TIMEOUT,
Duration.ofMillis(initQueryTimeout))
.withInt(DefaultDriverOption.CONNECTION_POOL_LOCAL_SIZE, connectionPoolLocalSize)
.withInt(DefaultDriverOption.CONNECTION_POOL_REMOTE_SIZE, connectionPoolRemoteSize)
// The maximum number of requests that can be executed concurrently on a
// connection.
// This must be between 1 and 32768.
.withInt(DefaultDriverOption.CONNECTION_MAX_REQUESTS, maxRequestsPerConnection)
.build());
if (isSslEnabled) {
cqlSessionBuilder.withSslContext(loadSslContext());
}
return cqlSessionBuilder;
}
};
}
private SSLContext loadSslContext() {
SSLContext sslContext = null;
try {
sslContext = SSLContext.getInstance("TLSv1.2");
KeyManagerFactory kmf =
KeyManagerFactory.getInstance(KeyManagerFactory.getDefaultAlgorithm());
KeyStore ks = KeyStore.getInstance("JKS");
char[] passphrase = truststorePassword.toCharArray();
ks.load(new FileInputStream(truststoreLocation), passphrase);
kmf.init(ks, passphrase);
sslContext.init(kmf.getKeyManagers(), null, null);
} catch (GeneralSecurityException | IOException e) {
throw new RuntimeException(e);
}
return sslContext;
}
}