SqlMapConfiguration.java

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");
			}
		}
	}

	//taotao add
	//public void setDatabaseProductName(String databaseProductName) {
	//	client.getDelegate().setDatabaseProductName(databaseProductName);
	//}

	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();
			}
		}
	}

}