292 lines
13 KiB
Java
292 lines
13 KiB
Java
package com.moparisthebest.jdbc;
|
|
|
|
import java.sql.Connection;
|
|
import java.sql.PreparedStatement;
|
|
import java.sql.ResultSet;
|
|
import java.sql.SQLException;
|
|
import java.util.*;
|
|
|
|
import static com.moparisthebest.jdbc.TryClose.tryClose;
|
|
|
|
/**
|
|
* This class caches the PreparedStatement's it creates for the strings you send in, then closes them when the close() method is called.
|
|
* Since PreparedStatement is not thread-safe, this class cannot be either. Be sure to call it from only a single thread
|
|
* or synchronize around it.
|
|
*/
|
|
public class CachingQueryMapper extends QueryMapper {
|
|
|
|
protected final Map<String, PreparedStatement> cache;
|
|
|
|
protected CachingQueryMapper(Connection conn, String jndiName, ResultSetMapper cm, final int maxEntries) {
|
|
super(conn, jndiName, cm);
|
|
if (maxEntries > 0) { // we want a limited cache
|
|
final float loadFactor = 0.75f; // default for HashMaps
|
|
// if we set the initialCapacity this way, nothing should ever need re-sized
|
|
final int initialCapacity = ((int) Math.ceil(maxEntries / loadFactor)) + 1;
|
|
cache = new LinkedHashMap<String, PreparedStatement>(initialCapacity, loadFactor, true) {
|
|
@Override
|
|
protected boolean removeEldestEntry(Map.Entry<String, PreparedStatement> eldest) {
|
|
final boolean remove = size() > maxEntries;
|
|
if(remove){
|
|
//System.out.printf("closing PreparedStatement '%s' with key '%s'\n", eldest.getValue(), eldest.getKey());
|
|
tryClose(eldest.getValue());
|
|
}
|
|
return remove;
|
|
}
|
|
};
|
|
} else
|
|
cache = new HashMap<String, PreparedStatement>();
|
|
}
|
|
|
|
public CachingQueryMapper(Connection conn, ResultSetMapper cm, final int maxEntries) {
|
|
this(conn, null, cm, maxEntries);
|
|
}
|
|
|
|
public CachingQueryMapper(Connection conn, final int maxEntries) {
|
|
this(conn, null, null, maxEntries);
|
|
}
|
|
|
|
public CachingQueryMapper(String jndiName, ResultSetMapper cm, final int maxEntries) {
|
|
this(null, jndiName, cm, maxEntries);
|
|
}
|
|
|
|
public CachingQueryMapper(String jndiName, final int maxEntries) {
|
|
this(null, jndiName, null, maxEntries);
|
|
}
|
|
|
|
protected CachingQueryMapper(Connection conn, String jndiName, ResultSetMapper cm) {
|
|
this(conn, jndiName, cm, 20); // default size of 20
|
|
}
|
|
|
|
public CachingQueryMapper(Connection conn, ResultSetMapper cm) {
|
|
this(conn, null, cm);
|
|
}
|
|
|
|
public CachingQueryMapper(Connection conn) {
|
|
this(conn, null, null);
|
|
}
|
|
|
|
public CachingQueryMapper(String jndiName, ResultSetMapper cm) {
|
|
this(null, jndiName, cm);
|
|
}
|
|
|
|
public CachingQueryMapper(String jndiName) {
|
|
this(null, jndiName, null);
|
|
}
|
|
|
|
protected PreparedStatement getPreparedStatement(String sql) throws SQLException {
|
|
return getPreparedStatement(sql, ResultSet.TYPE_FORWARD_ONLY,ResultSet.CONCUR_READ_ONLY);
|
|
}
|
|
|
|
protected PreparedStatement getPreparedStatement(String sql, int resultSetType, int resultSetConcurrency) throws SQLException {
|
|
PreparedStatement ps = cache.get(sql);
|
|
if (ps == null) {
|
|
//System.out.println("cache miss");
|
|
ps = conn.prepareStatement(sql,resultSetType,resultSetConcurrency);
|
|
cache.put(sql, ps);
|
|
}
|
|
//else System.out.println("cache hit");
|
|
return ps;
|
|
}
|
|
|
|
public void clearCache(boolean close) {
|
|
//System.out.println("cache size: "+cache.size());
|
|
for (PreparedStatement ps : cache.values())
|
|
tryClose(ps);
|
|
if (close)
|
|
super.close();
|
|
else
|
|
cache.clear();
|
|
}
|
|
|
|
public void clearCache() {
|
|
this.clearCache(false);
|
|
}
|
|
|
|
@Override
|
|
public void close() {
|
|
this.clearCache(true);
|
|
}
|
|
|
|
@Override
|
|
public int executeUpdate(String sql, Object... bindObjects) throws SQLException {
|
|
return super.executeUpdate(getPreparedStatement(sql), bindObjects);
|
|
}
|
|
|
|
@Override
|
|
public boolean executeUpdateSuccess(String sql, Object... bindObjects) throws SQLException {
|
|
return super.executeUpdateSuccess(getPreparedStatement(sql), bindObjects);
|
|
}
|
|
|
|
// these grab ResultSets from the database
|
|
|
|
@Override
|
|
public ResultSet toResultSet(String sql, Object... bindObjects) throws SQLException {
|
|
return super.toResultSet(getPreparedStatement(sql), bindObjects);
|
|
}
|
|
|
|
@Override
|
|
public ResultSet toResultSet(String sql, Integer rsType, Integer rsConcurrency, Object... bindObjects) throws SQLException {
|
|
return super.toResultSet(getPreparedStatement(sql,rsType,rsConcurrency), bindObjects);
|
|
}
|
|
|
|
// DO NOT EDIT BELOW THIS LINE, OR CHANGE THIS COMMENT, CODE AUTOMATICALLY GENERATED BY genQueryMapper.sh
|
|
|
|
@Override
|
|
public <T> T toObject(String sql, Class<T> componentType, final Object... bindObjects) throws SQLException {
|
|
return super.toObject(getPreparedStatement(sql), componentType, bindObjects);
|
|
}
|
|
|
|
@Override
|
|
public <T extends Map<String, V>, V> Map<String, V> toSingleMap(String sql, Class<T> componentType, Class<V> mapValType, final Object... bindObjects) throws SQLException {
|
|
return super.toSingleMap(getPreparedStatement(sql), componentType, mapValType, bindObjects);
|
|
}
|
|
|
|
@Override
|
|
public <V> Map<String, V> toSingleMap(String sql, Class<V> mapValType, final Object... bindObjects) throws SQLException {
|
|
return super.toSingleMap(getPreparedStatement(sql), mapValType, bindObjects);
|
|
}
|
|
|
|
@Override
|
|
public <T extends Collection<E>, E> T toCollection(String sql, final Class<T> collectionType, Class<E> componentType, final Object... bindObjects) throws SQLException {
|
|
return super.toCollection(getPreparedStatement(sql), collectionType, componentType, bindObjects);
|
|
}
|
|
|
|
@Override
|
|
public <T extends Collection<E>, E> T toCollection(String sql, T list, Class<E> componentType, final Object... bindObjects) throws SQLException {
|
|
return super.toCollection(getPreparedStatement(sql), list, componentType, bindObjects);
|
|
}
|
|
|
|
@Override
|
|
public <T extends Map<K, E>, K, E> T toMap(String sql, T map, Class<K> mapKeyType, Class<E> componentType, final Object... bindObjects) throws SQLException {
|
|
return super.toMap(getPreparedStatement(sql), map, mapKeyType, componentType, bindObjects);
|
|
}
|
|
|
|
@Override
|
|
public <T extends Map<K, E>, K, E extends Collection<C>, C> T toMapCollection(String sql, final Class<T> returnType, Class<K> mapKeyType, Class<E> collectionType, Class<C> componentType, final Object... bindObjects) throws SQLException {
|
|
return super.toMapCollection(getPreparedStatement(sql), returnType, mapKeyType, collectionType, componentType, bindObjects);
|
|
}
|
|
|
|
@Override
|
|
public <T extends Map<K, E>, K, E extends Collection<C>, C> T toMapCollection(String sql, T map, Class<K> mapKeyType, Class<E> collectionType, Class<C> componentType, final Object... bindObjects) throws SQLException {
|
|
return super.toMapCollection(getPreparedStatement(sql), map, mapKeyType, collectionType, componentType, bindObjects);
|
|
}
|
|
|
|
@Override
|
|
public <T> ListIterator<T> toListIterator(String sql, final Class<T> type, final Object... bindObjects) throws SQLException {
|
|
return super.toListIterator(getPreparedStatement(sql), type, bindObjects);
|
|
}
|
|
|
|
@Override
|
|
public <T> Iterator<T> toIterator(String sql, final Class<T> type, final Object... bindObjects) throws SQLException {
|
|
return super.toIterator(getPreparedStatement(sql), type, bindObjects);
|
|
}
|
|
|
|
@Override
|
|
public <T> T[] toArray(String sql, final Class<T> type, final Object... bindObjects) throws SQLException {
|
|
return super.toArray(getPreparedStatement(sql), type, bindObjects);
|
|
}
|
|
|
|
@Override
|
|
public <E> List<E> toList(String sql, Class<E> componentType, final Object... bindObjects) throws SQLException {
|
|
return super.toList(getPreparedStatement(sql), componentType, bindObjects);
|
|
}
|
|
|
|
@Override
|
|
public <K, E> Map<K, E> toMap(String sql, Class<K> mapKeyType, Class<E> componentType, final Object... bindObjects) throws SQLException {
|
|
return super.toMap(getPreparedStatement(sql), mapKeyType, componentType, bindObjects);
|
|
}
|
|
|
|
@Override
|
|
public <K, E extends List<C>, C> Map<K, E> toMapList(String sql, Class<K> mapKeyType, Class<C> componentType, final Object... bindObjects) throws SQLException {
|
|
return super.toMapList(getPreparedStatement(sql), mapKeyType, componentType, bindObjects);
|
|
}
|
|
|
|
@Override
|
|
public <T extends Collection<E>, E extends Map<String, V>, V> T toCollectionMap(String sql, final Class<T> collectionType, Class<E> componentType, Class<V> mapValType, final Object... bindObjects) throws SQLException {
|
|
return super.toCollectionMap(getPreparedStatement(sql), collectionType, componentType, mapValType, bindObjects);
|
|
}
|
|
|
|
@Override
|
|
public <T extends Collection<E>, E extends Map<String, V>, V> T toCollectionMap(String sql, T list, Class<E> componentType, Class<V> mapValType, final Object... bindObjects) throws SQLException {
|
|
return super.toCollectionMap(getPreparedStatement(sql), list, componentType, mapValType, bindObjects);
|
|
}
|
|
|
|
@Override
|
|
public <T extends Map<K, E>, K, E extends Map<String, V>, V> T toMapMap(String sql, final Class<T> returnType, Class<K> mapKeyType, Class<E> componentType, Class<V> mapValType, final Object... bindObjects) throws SQLException {
|
|
return super.toMapMap(getPreparedStatement(sql), returnType, mapKeyType, componentType, mapValType, bindObjects);
|
|
}
|
|
|
|
@Override
|
|
public <T extends Map<K, E>, K, E extends Map<String, V>, V> T toMapMap(String sql, T map, Class<K> mapKeyType, Class<E> componentType, Class<V> mapValType, final Object... bindObjects) throws SQLException {
|
|
return super.toMapMap(getPreparedStatement(sql), map, mapKeyType, componentType, mapValType, bindObjects);
|
|
}
|
|
|
|
@Override
|
|
public <T extends Map<K, C>, K, C extends Collection<E>, E extends Map<String, V>, V> T toMapCollectionMap(String sql, final Class<T> returnType, Class<K> mapKeyType, Class<C> collectionType, Class<E> componentType, Class<V> mapValType, final Object... bindObjects) throws SQLException {
|
|
return super.toMapCollectionMap(getPreparedStatement(sql), returnType, mapKeyType, collectionType, componentType, mapValType, bindObjects);
|
|
}
|
|
|
|
@Override
|
|
public <T extends Map<K, C>, K, C extends Collection<E>, E extends Map<String, V>, V> T toMapCollectionMap(String sql, T map, Class<K> mapKeyType, Class<C> collectionType, Class<E> componentType, Class<V> mapValType, final Object... bindObjects) throws SQLException {
|
|
return super.toMapCollectionMap(getPreparedStatement(sql), map, mapKeyType, collectionType, componentType, mapValType, bindObjects);
|
|
}
|
|
|
|
@Override
|
|
public <T extends Map<String, V>, V> ListIterator<Map<String, V>> toListIteratorMap(String sql, final Class<T> type, Class<V> mapValType, final Object... bindObjects) throws SQLException {
|
|
return super.toListIteratorMap(getPreparedStatement(sql), type, mapValType, bindObjects);
|
|
}
|
|
|
|
@Override
|
|
public <T extends Map<String, V>, V> Iterator<Map<String, V>> toIteratorMap(String sql, final Class<T> type, Class<V> mapValType, final Object... bindObjects) throws SQLException {
|
|
return super.toIteratorMap(getPreparedStatement(sql), type, mapValType, bindObjects);
|
|
}
|
|
|
|
@Override
|
|
public <T extends Map<String, V>, V> Map<String, V>[] toArrayMap(String sql, final Class<T> type, Class<V> mapValType, final Object... bindObjects) throws SQLException {
|
|
return super.toArrayMap(getPreparedStatement(sql), type, mapValType, bindObjects);
|
|
}
|
|
|
|
@Override
|
|
public <E extends Map<String, V>, V> List<Map<String, V>> toListMap(String sql, Class<E> componentType, Class<V> mapValType, final Object... bindObjects) throws SQLException {
|
|
return super.toListMap(getPreparedStatement(sql), componentType, mapValType, bindObjects);
|
|
}
|
|
|
|
@Override
|
|
public <K, E extends Map<String, V>, V> Map<K, Map<String, V>> toMapMap(String sql, Class<K> mapKeyType, Class<E> componentType, Class<V> mapValType, final Object... bindObjects) throws SQLException {
|
|
return super.toMapMap(getPreparedStatement(sql), mapKeyType, componentType, mapValType, bindObjects);
|
|
}
|
|
|
|
@Override
|
|
public <K, E extends Map<String, V>, V> Map<K, List<Map<String, V>>> toMapListMap(String sql, Class<K> mapKeyType, Class<E> componentType, Class<V> mapValType, final Object... bindObjects) throws SQLException {
|
|
return super.toMapListMap(getPreparedStatement(sql), mapKeyType, componentType, mapValType, bindObjects);
|
|
}
|
|
|
|
@Override
|
|
public <V> ListIterator<Map<String, V>> toListIteratorMap(String sql, Class<V> mapValType, final Object... bindObjects) throws SQLException {
|
|
return super.toListIteratorMap(getPreparedStatement(sql), mapValType, bindObjects);
|
|
}
|
|
|
|
@Override
|
|
public <V> Iterator<Map<String, V>> toIteratorMap(String sql, Class<V> mapValType, final Object... bindObjects) throws SQLException {
|
|
return super.toIteratorMap(getPreparedStatement(sql), mapValType, bindObjects);
|
|
}
|
|
|
|
@Override
|
|
public <V> List<Map<String, V>> toListMap(String sql, Class<V> mapValType, final Object... bindObjects) throws SQLException {
|
|
return super.toListMap(getPreparedStatement(sql), mapValType, bindObjects);
|
|
}
|
|
|
|
@Override
|
|
public <K, V> Map<K, Map<String, V>> toMapMap(String sql, Class<K> mapKeyType, Class<V> mapValType, final Object... bindObjects) throws SQLException {
|
|
return super.toMapMap(getPreparedStatement(sql), mapKeyType, mapValType, bindObjects);
|
|
}
|
|
|
|
@Override
|
|
public <K, V> Map<K, List<Map<String, V>>> toMapListMap(String sql, Class<K> mapKeyType, Class<V> mapValType, final Object... bindObjects) throws SQLException {
|
|
return super.toMapListMap(getPreparedStatement(sql), mapKeyType, mapValType, bindObjects);
|
|
}
|
|
|
|
}
|
|
|