org.springframework.data.cassandra.core.cql.support.MapPreparedStatementCache Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of spring-data-cassandra Show documentation
Show all versions of spring-data-cassandra Show documentation
Cassandra support for Spring Data
/*
* Copyright 2017-2019 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* https://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.springframework.data.cassandra.core.cql.support;
import lombok.EqualsAndHashCode;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.function.Supplier;
import org.springframework.util.Assert;
import com.datastax.driver.core.Cluster;
import com.datastax.driver.core.PreparedStatement;
import com.datastax.driver.core.RegularStatement;
import com.datastax.driver.core.Session;
/**
* {@link PreparedStatementCache} backed by a {@link Map} cache. Defaults to simple {@link ConcurrentHashMap} caching.
*
* Statements are cached with a key consisting of {@link Cluster}, {@code keyspace} and the {@code cql} text. Statement
* options (idempotency, timeouts) apply from the statement that was initially prepared.
*
* @author Mark Paluch
* @since 2.0
*/
public class MapPreparedStatementCache implements PreparedStatementCache {
private final Map cache;
/**
* Create a new {@link MapPreparedStatementCache}.
*
* @param cache must not be {@literal null}.
*/
private MapPreparedStatementCache(Map cache) {
Assert.notNull(cache, "Cache must not be null");
this.cache = cache;
}
/**
* Create a {@link MapPreparedStatementCache} using {@link ConcurrentHashMap}.
*
* @return the new {@link MapPreparedStatementCache} backed by {@link ConcurrentHashMap}.
*/
public static MapPreparedStatementCache create() {
return of(new ConcurrentHashMap<>());
}
/**
* Create a {@link MapPreparedStatementCache} using the given {@link Map}.
*
* @return the new {@link MapPreparedStatementCache} backed the given {@link Map}.
*/
public static MapPreparedStatementCache of(Map cache) {
return new MapPreparedStatementCache(cache);
}
/**
* @return the underlying {@link Map cache}.
*/
protected Map getCache() {
return this.cache;
}
/* (non-Javadoc)
* @see org.springframework.data.cassandra.core.cql.support.PrepatedStatementCache#getPreparedStatement(com.datastax.driver.core.RegularStatement, com.datastax.driver.core.Session, java.util.function.Supplier)
*/
@Override
public PreparedStatement getPreparedStatement(Session session, RegularStatement statement,
Supplier preparer) {
CacheKey cacheKey = new CacheKey(session, statement.toString());
return getCache().computeIfAbsent(cacheKey, key -> preparer.get());
}
/**
* {@link CacheKey} for {@link PreparedStatement} caching.
*/
@EqualsAndHashCode
protected static class CacheKey {
final Cluster cluster;
final String keyspace;
final String cql;
CacheKey(Session session, String cql) {
this.cluster = session.getCluster();
this.keyspace = session.getLoggedKeyspace();
this.cql = cql;
}
}
}