SqlSessionFactoryBean.java

/**
 * 
 * Copyright 2014 The Darks ORM Project (Liu lihua)
 * 
 * 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
 * 
 * http://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 darks.orm.spring;

import javax.sql.DataSource;

import org.springframework.beans.BeansException;
import org.springframework.beans.factory.BeanNameAware;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.beans.factory.config.ConfigurableListableBeanFactory;
import org.springframework.beans.factory.support.BeanDefinitionRegistry;
import org.springframework.beans.factory.support.BeanDefinitionRegistryPostProcessor;
import org.springframework.beans.factory.support.BeanNameGenerator;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.core.io.Resource;

import darks.orm.core.config.SpringDataParamConfig;
import darks.orm.core.factory.SqlSessionFactory;
import darks.orm.datasource.factory.DataSourceFactory;
import darks.orm.exceptions.ConfigException;
import darks.orm.log.Logger;
import darks.orm.log.LoggerFactory;

/**
 * Sql session factory config for spring.<br>
 * Examples:
 * 
 * <pre>
 * <bean class="darks.orm.spring.SqlSessionFactoryBean">
 * 		<property name="dataSource" ref="testDataSource" />
 * 		<property name="scanPackages" value="darks.orm.test.mapper,darks.orm.examples.mapper" />
 * 		<property name="configLocation" value="classpath:darks.xml" />
 * 		<property name="dataParamConfig.autoCommit" value="true" />
 * 		<property name="dataParamConfig.resultSetType" value="scroll" />
 * 		<property name="dataParamConfig.sensitive" value="false" />
 * 		<property name="dataParamConfig.concurrency" value="read" />
 * </bean>
 * </pre>
 * 
 * @author Liu Lihua
 * 
 */
public class SqlSessionFactoryBean implements BeanDefinitionRegistryPostProcessor,
		InitializingBean, ApplicationContextAware, BeanNameAware
{
	private static Logger log = LoggerFactory.getLogger(SqlSessionFactoryBean.class);

	private DataSource dataSource;

	private Resource configLocation;

	private String scanPackages;

	private boolean scanSubPackage = false;

	private ApplicationContext context;

	private BeanNameGenerator nameGenerator;

	private String beanName;

	private SpringDataParamConfig dataParamConfig = new SpringDataParamConfig();

	public SqlSessionFactoryBean()
	{

	}

	@Override
	public void setApplicationContext(ApplicationContext context) throws BeansException
	{
		this.context = context;
	}

	@Override
	public void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory)
			throws BeansException
	{

	}

	@Override
	public void setBeanName(String beanName)
	{
		this.beanName = beanName;
	}

	@Override
	public void afterPropertiesSet() throws Exception
	{
		if (scanPackages == null || "".equals(scanPackages.trim()))
		{
			throw new ConfigException("Spring scan orm packages is empty");
		}
		if (configLocation == null)
		{
			throw new ConfigException("Spring darks orm configLocation is empty");
		}
		if (dataSource == null)
		{
			throw new ConfigException("Spring darks orm's 'dataSource' is null");
		}
		buildSqlSessionFactory();
	}

	private void buildSqlSessionFactory() throws Exception
	{
		DataSourceFactory.getInstance().setDataSource(dataSource, dataParamConfig);
		SqlSessionFactory.initialize(configLocation.getInputStream());
	}

	@Override
	public void postProcessBeanDefinitionRegistry(BeanDefinitionRegistry registry)
			throws BeansException
	{
		log.debug("DarksORM register scan package by spring.");
		MapperClassDefinitionScanner scanner = new MapperClassDefinitionScanner(registry);
		scanner.setResourceLoader(context);
		scanner.setBeanNameGenerator(nameGenerator);
		scanner.registerFilters();
		String[] packages = scanPackages.trim().split(",");
		scanner.scan(packages);
	}

	public String getScanPackages()
	{
		return scanPackages;
	}

	public void setScanPackages(String scanPackages)
	{
		this.scanPackages = scanPackages;
	}

	public boolean isScanSubPackage()
	{
		return scanSubPackage;
	}

	public void setScanSubPackage(boolean scanSubPackage)
	{
		this.scanSubPackage = scanSubPackage;
	}

	public void setNameGenerator(BeanNameGenerator nameGenerator)
	{
		this.nameGenerator = nameGenerator;
	}

	public void setDataSource(DataSource dataSource)
	{
		this.dataSource = dataSource;
	}

	public void setConfigLocation(Resource configLocation)
	{
		this.configLocation = configLocation;
	}

	public SpringDataParamConfig getDataParamConfig()
	{
		return dataParamConfig;
	}

	public void setDataParamConfig(SpringDataParamConfig dataParamConfig)
	{
		this.dataParamConfig = dataParamConfig;
	}

}