spring+hibernate多数据源切换のSessionFactory切换

看到好多多数据源的例子，有切换datasource的，有切换sessionFactory的，我的是mysql和sqlserver两个数据库之间的切换，涉及到方言，所以就采用了后者。因为之前差点成功了，但是由于疏忽犯了个脑残失误就卡在somewhe了 ，今天本来想理清一下思路问大神的，结果理了一下，自己搞定了。不废话了，直接进入正题。

首先说配置方面，正常情况下都是陪一个datasource,一个sessionfactory,有事务还要配置tansationManager，如果是多数据源的话，就会有两个datasource，如果是不同类型的数据库的话，就需要连个Sessionfactory来配置方言。以下是配置内容

	
        
        
        
        
                
        
        
        
        
        
        
        
        
        
        
    
	
		
		
			
				
					org.hibernate.dialect.MySQLDialect
				
				false
				
					org.hibernate.cache.ehcache.EhCacheRegionFactory
				
				false
				false
				80
				80
				
					true
				
				
					org.springframework.orm.hibernate4.SpringSessionContext
				
				
					true
				
				true
				3
				none
			
		
		
	

	
	
        
        
        
        
                
        
        
        
        
        
        
        
        
        
        
    
	
		
		
			
				
					org.hibernate.dialect.SQLServerDialect
				
				false
				
					org.hibernate.cache.ehcache.EhCacheRegionFactory
				
				false
				false
				80
				80
				
					true
				
				
					org.springframework.orm.hibernate4.SpringSessionContext
				
				
					true
				
				true
				3
				none
			
		
		
	
    
    
        
        
             
        
    
    
		
	
 	
    
        
    

好了，xml配置就这么多，然后开始用自己写的类来接管Sessionfactory和 TransationFactory。如上文中看到的

com.ivo.util2.DynamicSessionFactoryImpl 和 com.ivo.util2.DynamicTransactionManager

先写一个继承sessionfactory的接口DynamicSessionFactory

package com.ivo.util2;

import org.hibernate.SessionFactory;

public interface DynamicSessionFactory extends SessionFactory {
    
    public SessionFactory getHibernateSessionFactory();
}

实现这个接口

DynamicSessionFactoryImpl

package com.ivo.util2;

import com.ivo.util.CustomerContextHolder;
import com.ivo.util2.DynamicSessionFactory;

import java.io.Serializable;
import java.sql.Connection;
import java.util.Map;
import java.util.Set;

import javax.naming.NamingException;
import javax.naming.Reference;

import org.hibernate.Cache;
import org.hibernate.HibernateException;
import org.hibernate.Session;
import org.hibernate.SessionBuilder;
import org.hibernate.SessionFactory;
import org.hibernate.StatelessSession;
import org.hibernate.StatelessSessionBuilder;
import org.hibernate.TypeHelper;
import org.hibernate.engine.spi.FilterDefinition;
import org.hibernate.metadata.ClassMetadata;
import org.hibernate.metadata.CollectionMetadata;
import org.hibernate.stat.Statistics;

public class DynamicSessionFactoryImpl implements DynamicSessionFactory {

	/**
	 * 
	 */
	private static final long serialVersionUID = 7328134180626920452L;
	private Map targetSessionFactorys;
	private SessionFactory defaultTargetSessionFactory;

	@Override
	public SessionFactoryOptions getSessionFactoryOptions() {
		return getHibernateSessionFactory().getSessionFactoryOptions();
	}

	@Override
	public SessionBuilder withOptions() {
		return getHibernateSessionFactory().withOptions();
	}

	@Override
	public Session openSession() throws HibernateException {
		return getHibernateSessionFactory().openSession();
	}

	@Override
	public Session getCurrentSession() throws HibernateException {
		return getHibernateSessionFactory().getCurrentSession();
	}

	@Override
	public StatelessSessionBuilder withStatelessOptions() {
		return getHibernateSessionFactory().withStatelessOptions();
	}

	@Override
	public StatelessSession openStatelessSession() {
		return getHibernateSessionFactory().openStatelessSession();
	}

	@Override
	public StatelessSession openStatelessSession(Connection connection) {
		return getHibernateSessionFactory().openStatelessSession(connection);
	}

	@Override
	public ClassMetadata getClassMetadata(
			@SuppressWarnings("rawtypes") Class entityClass) {
		return getHibernateSessionFactory().getClassMetadata(entityClass);
	}

	@Override
	public ClassMetadata getClassMetadata(String entityName) {
		return getHibernateSessionFactory().getClassMetadata(entityName);
	}

	@Override
	public CollectionMetadata getCollectionMetadata(String roleName) {
		return getHibernateSessionFactory().getCollectionMetadata(roleName);
	}

	@Override
	public Map getAllClassMetadata() {
		return getHibernateSessionFactory().getAllClassMetadata();
	}

	@SuppressWarnings("rawtypes")
	@Override
	public Map getAllCollectionMetadata() {
		return getHibernateSessionFactory().getAllCollectionMetadata();
	}

	@Override
	public Statistics getStatistics() {
		return getHibernateSessionFactory().getStatistics();
	}

	@Override
	public void close() throws HibernateException {
		getHibernateSessionFactory().close();
	}

	@Override
	public boolean isClosed() {
		return getHibernateSessionFactory().isClosed();
	}

	@Override
	public Cache getCache() {
		return getHibernateSessionFactory().getCache();
	}

	@SuppressWarnings({ "deprecation", "rawtypes" })
	@Override
	public void evict(Class persistentClass) throws HibernateException {
		getHibernateSessionFactory().evict(persistentClass);
	}

	@SuppressWarnings({ "deprecation", "rawtypes" })
	@Override
	public void evict(Class persistentClass, Serializable id)
			throws HibernateException {
		getHibernateSessionFactory().evict(persistentClass, id);
	}

	@SuppressWarnings("deprecation")
	@Override
	public void evictEntity(String entityName) throws HibernateException {
		getHibernateSessionFactory().evictEntity(entityName);
	}

	@SuppressWarnings("deprecation")
	@Override
	public void evictEntity(String entityName, Serializable id)
			throws HibernateException {
		getHibernateSessionFactory().evictEntity(entityName, id);
	}

	@SuppressWarnings("deprecation")
	@Override
	public void evictCollection(String roleName) throws HibernateException {
		getHibernateSessionFactory().evictCollection(roleName);
	}

	@SuppressWarnings("deprecation")
	@Override
	public void evictCollection(String roleName, Serializable id)
			throws HibernateException {
		getHibernateSessionFactory().evictCollection(roleName, id);
	}

	@SuppressWarnings("deprecation")
	@Override
	public void evictQueries(String cacheRegion) throws HibernateException {
		getHibernateSessionFactory().evictQueries(cacheRegion);
	}

	@SuppressWarnings("deprecation")
	@Override
	public void evictQueries() throws HibernateException {
		getHibernateSessionFactory().evictQueries();
	}

	@SuppressWarnings("rawtypes")
	@Override
	public Set getDefinedFilterNames() {
		return getHibernateSessionFactory().getDefinedFilterNames();
	}

	@Override
	public FilterDefinition getFilterDefinition(String filterName)
			throws HibernateException {
		return getHibernateSessionFactory().getFilterDefinition(filterName);
	}

	@Override
	public boolean containsFetchProfileDefinition(String name) {
		return getHibernateSessionFactory()
				.containsFetchProfileDefinition(name);
	}

	@Override
	public TypeHelper getTypeHelper() {

		return getHibernateSessionFactory().getTypeHelper();
	}

	@Override
	public Reference getReference() throws NamingException {
		return getHibernateSessionFactory().getReference();
	}

	@Override
	public SessionFactory getHibernateSessionFactory() {
		SessionFactory targetSessionFactory = targetSessionFactorys
				.get(CustomerContextHolder.getCustomerType());
		if (targetSessionFactory != null) {
			return targetSessionFactory;
		} else if (defaultTargetSessionFactory != null) {
			return defaultTargetSessionFactory;
		}
		return null;
	}

	public void setTargetSessionFactorys(
			Map targetSessionFactorys) {
		this.targetSessionFactorys = targetSessionFactorys;
	}

	public void setDefaultTargetSessionFactory(
			SessionFactory defaultTargetSessionFactory) {
		this.defaultTargetSessionFactory = defaultTargetSessionFactory;
	}
}

事务管理类DynamicTransactionManager

package com.ivo.util2;

import org.springframework.orm.hibernate4.HibernateTransactionManager;
import javax.sql.DataSource;
import org.hibernate.SessionFactory;
import org.springframework.orm.hibernate4.SessionFactoryUtils;

public class DynamicTransactionManager extends HibernateTransactionManager {

	/**
	 * 
	 */
	private static final long serialVersionUID = 2591641126163504954L;

    @Override
    public DataSource getDataSource() {
        return SessionFactoryUtils.getDataSource(getSessionFactory());
    }
 
    @Override
    public SessionFactory getSessionFactory() {
        DynamicSessionFactory dynamicSessionFactory = (DynamicSessionFactory) super.getSessionFactory();  
        SessionFactory hibernateSessionFactory = dynamicSessionFactory.getHibernateSessionFactory();  
        return hibernateSessionFactory;  
    }
}

最后写一个切换当前线程内容的类

package com.ivo.util2;

public class CustomerContextHolder {

	 public static final String SESSION_FACTORY_MYSQL = "sessionFactoryMS";  
	    public static final String SESSION_FACTORY_SQLSERVER = "sessionFactorySS";  
	      
	    private static final ThreadLocal contextHolder = new ThreadLocal();  
	      
	    public static void setCustomerType(String dbType) {  
	        contextHolder.set(dbType);  
	    }  
	      
	    public static String getCustomerType() {  
	        return contextHolder.get();  
	    }  
	      
	    public static void clearCustomerType() {  
	        contextHolder.remove();  
	    }  
}

OK 大功告成了`(*∩_∩*)′