SqlMapConfiguration

package com.ibatis.sqlmap.engine.config;

import com.ibatis.common.beans.*;
import com.ibatis.common.resources.*;
import com.ibatis.sqlmap.client.*;
import com.ibatis.sqlmap.client.extensions.*;
import com.ibatis.sqlmap.engine.accessplan.*;
import com.ibatis.sqlmap.engine.cache.*;
import com.ibatis.sqlmap.engine.cache.fifo.*;
import com.ibatis.sqlmap.engine.cache.lru.*;
import com.ibatis.sqlmap.engine.cache.memory.*;
import com.ibatis.sqlmap.engine.datasource.*;
import com.ibatis.sqlmap.engine.impl.*;
import com.ibatis.sqlmap.engine.mapping.result.*;
import com.ibatis.sqlmap.engine.mapping.statement.*;
import com.ibatis.sqlmap.engine.scope.*;
import com.ibatis.sqlmap.engine.transaction.*;
import com.ibatis.sqlmap.engine.transaction.external.*;
import com.ibatis.sqlmap.engine.transaction.jdbc.*;
import com.ibatis.sqlmap.engine.transaction.jta.*;
import com.ibatis.sqlmap.engine.type.*;

import java.util.*;

public class SqlMapConfiguration {
private static final Probe PROBE = ProbeFactory.getProbe();
private ErrorContext errorContext;
private SqlMapExecutorDelegate delegate;
private TypeHandlerFactory typeHandlerFactory;
private SqlMapClientImpl client;
private Integer defaultStatementTimeout;

public SqlMapConfiguration() {
errorContext = new ErrorContext();
delegate = new SqlMapExecutorDelegate();
typeHandlerFactory = delegate.getTypeHandlerFactory();
client = new SqlMapClientImpl(delegate);
registerDefaultTypeAliases();
}

public TypeHandlerFactory getTypeHandlerFactory() {
return typeHandlerFactory;
}

public ErrorContext getErrorContext() {
return errorContext;
}

public SqlMapClientImpl getClient() {
return client;
}

public SqlMapExecutorDelegate getDelegate() {
return delegate;
}

public void setClassInfoCacheEnabled (boolean classInfoCacheEnabled) {
errorContext.setActivity("setting class info cache enabled/disabled");
ClassInfo.setCacheEnabled(classInfoCacheEnabled);
}

public void setLazyLoadingEnabled (boolean lazyLoadingEnabled) {
errorContext.setActivity("setting lazy loading enabled/disabled");
client.getDelegate().setLazyLoadingEnabled(lazyLoadingEnabled);
}

public void setStatementCachingEnabled (boolean statementCachingEnabled) {
errorContext.setActivity("setting statement caching enabled/disabled");
client.getDelegate().setStatementCacheEnabled(statementCachingEnabled);
}

public void setCacheModelsEnabled (boolean cacheModelsEnabled) {
errorContext.setActivity("setting cache models enabled/disabled");
client.getDelegate().setCacheModelsEnabled(cacheModelsEnabled);
}

public void setEnhancementEnabled (boolean enhancementEnabled) {
errorContext.setActivity("setting enhancement enabled/disabled");
try {
enhancementEnabled = enhancementEnabled && Resources.classForName("net.sf.cglib.proxy.InvocationHandler") != null;
} catch (ClassNotFoundException e) {
enhancementEnabled = false;
}
client.getDelegate().setEnhancementEnabled(enhancementEnabled);
AccessPlanFactory.setBytecodeEnhancementEnabled(enhancementEnabled);
}

public void setUseColumnLabel(boolean useColumnLabel) {
client.getDelegate().setUseColumnLabel(useColumnLabel);
}

public void setForceMultipleResultSetSupport(boolean forceMultipleResultSetSupport) {
client.getDelegate().setForceMultipleResultSetSupport(forceMultipleResultSetSupport);
}

public void setDefaultStatementTimeout(Integer defaultTimeout) {
errorContext.setActivity("setting default timeout");
if (defaultTimeout != null) {
try {
defaultStatementTimeout = defaultTimeout;
} catch (NumberFormatException e) {
throw new SqlMapException("Specified defaultStatementTimeout is not a valid integer");
}
}
}

public void setTransactionManager(TransactionManager txManager) {
delegate.setTxManager(txManager);
}

public void setResultObjectFactory(ResultObjectFactory rof) {
delegate.setResultObjectFactory(rof);
}

public void newTypeHandler(Class javaType, String jdbcType, Object callback) {
try {
errorContext.setActivity("building a building custom type handler");
TypeHandlerFactory typeHandlerFactory = client.getDelegate().getTypeHandlerFactory();
TypeHandler typeHandler;
if (callback instanceof TypeHandlerCallback) {
typeHandler = new CustomTypeHandler((TypeHandlerCallback) callback);
} else if (callback instanceof TypeHandler) {
typeHandler = (TypeHandler) callback;
} else {
throw new RuntimeException("The object '" + callback + "' is not a valid implementation of TypeHandler or TypeHandlerCallback");
}
errorContext.setMoreInfo("Check the javaType attribute '" + javaType + "' (must be a classname) or the jdbcType '" + jdbcType + "' (must be a JDBC type name).");
if (jdbcType != null && jdbcType.length() > 0) {
typeHandlerFactory.register(javaType, jdbcType, typeHandler);
} else {
typeHandlerFactory.register(javaType, typeHandler);
}
} catch (Exception e) {
throw new SqlMapException("Error registering occurred. Cause: " + e, e);
}
errorContext.setMoreInfo(null);
errorContext.setObjectId(null);
}

public CacheModelConfig newCacheModelConfig(String id, CacheController controller, boolean readOnly, boolean serialize) {
return new CacheModelConfig(this, id, controller, readOnly, serialize);
}

public ParameterMapConfig newParameterMapConfig(String id, Class parameterClass) {
return new ParameterMapConfig(this, id, parameterClass);
}

public ResultMapConfig newResultMapConfig(String id, Class resultClass, String groupBy, String extended, String xmlName) {
return new ResultMapConfig(this, id, resultClass, groupBy, extended, xmlName);
}

public MappedStatementConfig newMappedStatementConfig(String id, MappedStatement statement, SqlSource processor,
String parameterMapName, Class parameterClass,
String resultMapName, String[] additionalResultMapNames,
Class resultClass, Class[] additionalResultClasses,
String resultSetType, Integer fetchSize,
boolean allowRemapping, Integer timeout, String cacheModelName,
String xmlResultName) {
return new MappedStatementConfig(this, id, statement, processor, parameterMapName, parameterClass, resultMapName,
additionalResultMapNames, resultClass, additionalResultClasses, cacheModelName, resultSetType, fetchSize,
allowRemapping, timeout, defaultStatementTimeout, xmlResultName);
}

public void finalizeSqlMapConfig() {
wireUpCacheModels();
bindResultMapDiscriminators();
}

TypeHandler resolveTypeHandler(TypeHandlerFactory typeHandlerFactory, Class clazz, String propertyName, Class javaType, String jdbcType) {
return resolveTypeHandler(typeHandlerFactory, clazz, propertyName, javaType, jdbcType, false);
}

TypeHandler resolveTypeHandler(TypeHandlerFactory typeHandlerFactory, Class clazz, String propertyName, Class javaType, String jdbcType, boolean useSetterToResolve) {
TypeHandler handler;
if (clazz == null) {
// Unknown
handler = typeHandlerFactory.getUnkownTypeHandler();
} else if (DomTypeMarker.class.isAssignableFrom(clazz)) {
// DOM
handler = typeHandlerFactory.getTypeHandler(String.class, jdbcType);
} else if (java.util.Map.class.isAssignableFrom(clazz)) {
// Map
if (javaType == null) {
handler = typeHandlerFactory.getUnkownTypeHandler(); //BUG 1012591 - typeHandlerFactory.getTypeHandler(java.lang.Object.class, jdbcType);
} else {
handler = typeHandlerFactory.getTypeHandler(javaType, jdbcType);
}
} else if (typeHandlerFactory.getTypeHandler(clazz, jdbcType) != null) {
// Primitive
handler = typeHandlerFactory.getTypeHandler(clazz, jdbcType);
} else {
// JavaBean
if (javaType == null) {
if (useSetterToResolve) {
Class type = PROBE.getPropertyTypeForSetter(clazz, propertyName);
handler = typeHandlerFactory.getTypeHandler(type, jdbcType);
} else {
Class type = PROBE.getPropertyTypeForGetter(clazz, propertyName);
handler = typeHandlerFactory.getTypeHandler(type, jdbcType);
}
} else {
handler = typeHandlerFactory.getTypeHandler(javaType, jdbcType);
}
}
return handler;
}

private void registerDefaultTypeAliases() {
// TRANSACTION ALIASES
typeHandlerFactory.putTypeAlias("JDBC", JdbcTransactionConfig.class.getName());
typeHandlerFactory.putTypeAlias("JTA", JtaTransactionConfig.class.getName());
typeHandlerFactory.putTypeAlias("EXTERNAL", ExternalTransactionConfig.class.getName());

// DATA SOURCE ALIASES
typeHandlerFactory.putTypeAlias("SIMPLE", SimpleDataSourceFactory.class.getName());
typeHandlerFactory.putTypeAlias("DBCP", DbcpDataSourceFactory.class.getName());
typeHandlerFactory.putTypeAlias("JNDI", JndiDataSourceFactory.class.getName());

// CACHE ALIASES
typeHandlerFactory.putTypeAlias("FIFO", FifoCacheController.class.getName());
typeHandlerFactory.putTypeAlias("LRU", LruCacheController.class.getName());
typeHandlerFactory.putTypeAlias("MEMORY", MemoryCacheController.class.getName());
// use a string for OSCache to avoid unnecessary loading of properties upon init
typeHandlerFactory.putTypeAlias("OSCACHE", "com.ibatis.sqlmap.engine.cache.oscache.OSCacheController");

// TYPE ALIASEs
typeHandlerFactory.putTypeAlias("dom", DomTypeMarker.class.getName());
typeHandlerFactory.putTypeAlias("domCollection", DomCollectionTypeMarker.class.getName());
typeHandlerFactory.putTypeAlias("xml", XmlTypeMarker.class.getName());
typeHandlerFactory.putTypeAlias("xmlCollection", XmlCollectionTypeMarker.class.getName());
}

private void wireUpCacheModels() {
// Wire Up Cache Models
Iterator cacheNames = client.getDelegate().getCacheModelNames();
while (cacheNames.hasNext()) {
String cacheName = (String) cacheNames.next();
CacheModel cacheModel = client.getDelegate().getCacheModel(cacheName);
Iterator statementNames = cacheModel.getFlushTriggerStatementNames();
while (statementNames.hasNext()) {
String statementName = (String) statementNames.next();
MappedStatement statement = client.getDelegate().getMappedStatement(statementName);
if (statement != null) {
statement.addExecuteListener(cacheModel);
} else {
throw new RuntimeException("Could not find statement named '" + statementName + "' for use as a flush trigger for the cache model named '" + cacheName + "'.");
}
}
}
}

private void bindResultMapDiscriminators() {
// Bind discriminators
Iterator names = delegate.getResultMapNames();
while (names.hasNext()) {
String name = (String) names.next();
ResultMap rm = delegate.getResultMap(name);
Discriminator disc = rm.getDiscriminator();
if (disc != null) {
disc.bindSubMaps();
}
}
}

}