ShardedSqlSessionFactoryBean.java

/*
 * @(#)ShardedSqlSessionFactoryBean.java 2012-8-1 下午10:00:00
 *
 * Copyright (c) 2011-2012 Makersoft.org all rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 *
 */
package org.makersoft.shards.spring;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Properties;

import javax.sql.DataSource;

import org.apache.ibatis.plugin.Interceptor;
import org.apache.ibatis.session.SqlSessionFactory;
import org.apache.ibatis.type.TypeHandler;
import org.makersoft.shards.ShardedConfiguration;
import org.makersoft.shards.cfg.ShardConfiguration;
import org.makersoft.shards.cfg.impl.ShardConfigurationImpl;
import org.makersoft.shards.id.IdGenerator;
import org.makersoft.shards.session.ShardedSqlSessionFactory;
import org.makersoft.shards.strategy.ShardStrategyFactory;
import org.makersoft.shards.utils.Assert;
import org.makersoft.shards.utils.StringUtil;
import org.mybatis.spring.SqlSessionFactoryBean;
import org.springframework.beans.factory.FactoryBean;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.core.io.Resource;
import org.springframework.util.CollectionUtils;

/**
 * 
 */
public class ShardedSqlSessionFactoryBean implements
		FactoryBean<ShardedSqlSessionFactory>, InitializingBean {

	private Resource configLocation;

	private Resource[] mapperLocations;

	private Map<Integer, DataSource> dataSources;
	
	private Properties configurationProperties;

	private String environment = ShardedSqlSessionFactory.class.getSimpleName();

//	private boolean failFast;
	
	private Interceptor[] plugins;

	private TypeHandler<?>[] typeHandlers;

	private String typeHandlersPackage;

	private Class<?>[] typeAliases;

	private String typeAliasesPackage;
	
	private ShardedSqlSessionFactory shardedSqlSessionFactory;
	
	private ShardStrategyFactory shardStrategyFactory;
	
	private IdGenerator idGenerator; 
	
	//直接配置ShardConfiguration
	private List<ShardConfigurationImpl> shardConfigurations;
	
	@Override
	public void afterPropertiesSet() throws Exception {
//		Assert.notNull(dataSources, "data sources can not be null.");
		
		List<ShardConfiguration> shardConfigs = new ArrayList<ShardConfiguration>();
		
		if(CollectionUtils.isEmpty(shardConfigurations)){
			for(Map.Entry<Integer, DataSource> entry : dataSources.entrySet()){
				int shardId = entry.getKey();	//虚拟分区ID
				DataSource dataSource = entry.getValue();	//虚拟分区所属数据源
				
				SqlSessionFactoryBean factoryBean = new SqlSessionFactoryBean();
				factoryBean.setConfigLocation(this.configLocation);
				factoryBean.setMapperLocations(this.mapperLocations);
				factoryBean.setDataSource(dataSource);
				factoryBean.setEnvironment(this.environment);
				factoryBean.setConfigurationProperties(this.configurationProperties);
				factoryBean.setPlugins(this.plugins);
				factoryBean.setTypeHandlers(this.typeHandlers);
				factoryBean.setTypeHandlersPackage(this.typeHandlersPackage);
				factoryBean.setTypeAliases(this.typeAliases);
				factoryBean.setTypeAliasesPackage(this.typeAliasesPackage);
				
				SqlSessionFactory sessionFacotry = factoryBean.getObject();
				
				shardConfigs.add(new ShardConfigurationImpl(shardId, dataSource, sessionFacotry));
			}
		}else {
			for(ShardConfigurationImpl shardConfiguration : shardConfigurations){
				
				Assert.notNull(shardConfiguration.getShardId(), "shard id can not be null.");
				Assert.notNull(shardConfiguration.getShardDataSource(), "data source can not be null.");

				if(shardConfiguration.getConfigLocation() == null) {
					shardConfiguration.setConfigLocation(this.configLocation);
				}

				if(shardConfiguration.getMapperLocations() == null || shardConfiguration.getMapperLocations().length == 0) {
					shardConfiguration.setMapperLocations(this.mapperLocations);
				}

				SqlSessionFactoryBean factoryBean = new SqlSessionFactoryBean();
				factoryBean.setConfigLocation(shardConfiguration.getConfigLocation());
				factoryBean.setMapperLocations(shardConfiguration.getMapperLocations());

				factoryBean.setDataSource(shardConfiguration.getShardDataSource());
				factoryBean.setEnvironment(this.environment);
				factoryBean.setConfigurationProperties(this.configurationProperties);
				factoryBean.setPlugins(this.plugins);
				factoryBean.setTypeHandlers(this.typeHandlers);

				if(StringUtil.isEmptyOrWhitespace(shardConfiguration.getTypeHandlersPackage())) {
					shardConfiguration.setTypeHandlersPackage(this.typeHandlersPackage);
				}

				factoryBean.setTypeHandlersPackage(shardConfiguration.getTypeHandlersPackage());
				factoryBean.setTypeAliases(this.typeAliases);

				if(StringUtil.isEmptyOrWhitespace(shardConfiguration.getTypeAliasesPackage())) {
					shardConfiguration.setTypeAliasesPackage(this.typeAliasesPackage);
				}

				factoryBean.setTypeAliasesPackage(shardConfiguration.getTypeAliasesPackage());

				SqlSessionFactory sessionFacotry = factoryBean.getObject();
				shardConfiguration.setSqlSessionFactory(sessionFacotry);

				shardConfigs.add(shardConfiguration);
			}
			
		}
		
		ShardedConfiguration configuration = new ShardedConfiguration(shardConfigs, this.shardStrategyFactory, idGenerator);
		shardedSqlSessionFactory = configuration.buildShardedSessionFactory();
	}

	@Override
	public ShardedSqlSessionFactory getObject() throws Exception {
		if (this.shardedSqlSessionFactory == null) {
			afterPropertiesSet();
		}

		return this.shardedSqlSessionFactory;
	}

	@Override
	public Class<?> getObjectType() {
		return this.shardedSqlSessionFactory == null ? ShardedSqlSessionFactory.class
				: this.shardedSqlSessionFactory.getClass();
	}

	@Override
	public boolean isSingleton() {
		return true;
	}
	
	//Setter from here
	public void setDataSources(final Map<Integer,DataSource> dataSources) {
		this.dataSources = dataSources;
	}

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

	public void setMapperLocations(Resource[] mapperLocations) {
		this.mapperLocations = mapperLocations;
	}

	public void setConfigurationProperties(Properties configurationProperties) {
		this.configurationProperties = configurationProperties;
	}


	public void setEnvironment(String environment) {
		this.environment = environment;
	}

	public void setPlugins(Interceptor[] plugins) {
		this.plugins = plugins;
	}

	public void setTypeHandlers(TypeHandler<?>[] typeHandlers) {
		this.typeHandlers = typeHandlers;
	}

	public void setTypeHandlersPackage(String typeHandlersPackage) {
		this.typeHandlersPackage = typeHandlersPackage;
	}

	public void setTypeAliases(Class<?>[] typeAliases) {
		this.typeAliases = typeAliases;
	}

	public void setTypeAliasesPackage(String typeAliasesPackage) {
		this.typeAliasesPackage = typeAliasesPackage;
	}

	public void setShardStrategyFactory(ShardStrategyFactory shardStrategyFactory) {
		this.shardStrategyFactory = shardStrategyFactory;
	}

	public void setIdGenerator(IdGenerator idGenerator) {
		this.idGenerator = idGenerator;
	}

	public void setShardConfigurations(List<ShardConfigurationImpl> shardConfigurations) {
		this.shardConfigurations = shardConfigurations;
	}

}