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

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

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

<!-- mysql -->
	<bean id="dataSourceMS" class="com.mchange.v2.c3p0.ComboPooledDataSource" destroy-method="close">
        <property name="driverClass" value="${datasource.driver}"/>
        <property name="jdbcUrl" value="${datasource.url}"/>
        <property name="user" value="${datasource.username}"/>
        <property name="password" value="${datasource.password}"/>
                
        <property name="acquireIncrement" value="${c3p0.acquireIncrement}"/>
        <property name="initialPoolSize" value="${c3p0.initialPoolSize}"/>
        <property name="minPoolSize" value="${c3p0.minPoolSize}"/>
        <property name="maxPoolSize" value="${c3p0.maxPoolSize}"/>
        <property name="maxIdleTime" value="${c3p0.maxIdleTime}"/>
        <property name="idleConnectionTestPeriod" value="${c3p0.idleConnectionTestPeriod}"/>
        <property name="maxStatements" value="${c3p0.maxStatements}"/>
        <property name="numHelperThreads" value="${c3p0.numHelperThreads}"/>
        <!-- <property name="preferredTestQuery" value="${c3p0.preferredTestQuery}"/> -->
        <property name="testConnectionOnCheckout" value="${c3p0.testConnectionOnCheckout}"/>
    </bean>
	<bean id="sessionFactoryMS"	class="org.springframework.orm.hibernate4.LocalSessionFactoryBean">
		<property name="dataSource" ref="dataSourceMS" />
		<property name="hibernateProperties">
			<props>
				<prop key="hibernate.dialect">
					org.hibernate.dialect.MySQLDialect
				</prop>
				<prop key="hibernate.show_sql">false</prop>
				<prop key="hibernate.cache.region.factory_class">
					org.hibernate.cache.ehcache.EhCacheRegionFactory
				</prop>
				<prop key="hibernate.format_sql">false</prop>
				<prop key="hibernate.use_sql_comments">false</prop>
				<prop key="hibernate.jdbc.fetch_size">80</prop>
				<prop key="hibernate.jdbc.batch_size">80</prop>
				<prop key="hibernate.jdbc.use_streams_for_binary">
					true
				</prop>
				<prop key="hibernate.current_session_context_class">
					org.springframework.orm.hibernate4.SpringSessionContext
				</prop>
				<prop key="hibernate.cache.use_second_level_cache">
					true
				</prop>
				<prop key="hibernate.cache.use_query_cache">true</prop>
				<prop key="hibernate.max_fetch_depth">3</prop>
				<prop key="hibernate.hbm2ddl.auto">none</prop>
			</props>
		</property>
		<property name="packagesToScan" value="com.ivo.*" />
	</bean>

	<!-- sqlServer -->
	<bean id="dataSourceSS" class="com.mchange.v2.c3p0.ComboPooledDataSource" destroy-method="close">
        <property name="driverClass" value="${datasource2.driver}"/>
        <property name="jdbcUrl" value="${datasource2.url}"/>
        <property name="user" value="${datasource2.username}"/>
        <property name="password" value="${datasource2.password}"/>
                
        <property name="acquireIncrement" value="${c3p0.acquireIncrement}"/>
        <property name="initialPoolSize" value="${c3p0.initialPoolSize}"/>
        <property name="minPoolSize" value="${c3p0.minPoolSize}"/>
        <property name="maxPoolSize" value="${c3p0.maxPoolSize}"/>
        <property name="maxIdleTime" value="${c3p0.maxIdleTime}"/>
        <property name="idleConnectionTestPeriod" value="${c3p0.idleConnectionTestPeriod}"/>
        <property name="maxStatements" value="${c3p0.maxStatements}"/>
        <property name="numHelperThreads" value="${c3p0.numHelperThreads}"/>
        <!-- <property name="preferredTestQuery" value="${c3p0.preferredTestQuery}"/> -->
        <property name="testConnectionOnCheckout" value="${c3p0.testConnectionOnCheckout}"/>
    </bean>
	<bean id="sessionFactorySS"	class="org.springframework.orm.hibernate4.LocalSessionFactoryBean">
		<property name="dataSource" ref="dataSourceSS" />
		<property name="hibernateProperties">
			<props>
				<prop key="hibernate.dialect">
					org.hibernate.dialect.SQLServerDialect
				</prop>
				<prop key="hibernate.show_sql">false</prop>
				<prop key="hibernate.cache.region.factory_class">
					org.hibernate.cache.ehcache.EhCacheRegionFactory
				</prop>
				<prop key="hibernate.format_sql">false</prop>
				<prop key="hibernate.use_sql_comments">false</prop>
				<prop key="hibernate.jdbc.fetch_size">80</prop>
				<prop key="hibernate.jdbc.batch_size">80</prop>
				<prop key="hibernate.jdbc.use_streams_for_binary">
					true
				</prop>
				<prop key="hibernate.current_session_context_class">
					org.springframework.orm.hibernate4.SpringSessionContext
				</prop>
				<prop key="hibernate.cache.use_second_level_cache">
					true
				</prop>
				<prop key="hibernate.cache.use_query_cache">true</prop>
				<prop key="hibernate.max_fetch_depth">3</prop>
				<prop key="hibernate.hbm2ddl.auto">none</prop>
			</props>
		</property>
		<property name="packagesToScan" value="com.ivo.*" />
	</bean>
    <!-- 动态SessionFactory -->
    <bean id="sessionFactory" class="com.ivo.util2.DynamicSessionFactoryImpl">
        <property name="defaultTargetSessionFactory" ref="sessionFactoryMS"/>
        <property name="targetSessionFactorys">
            <map>     
                <entry value-ref="sessionFactoryMS" key="sessionFactoryMS"/>
                <entry value-ref="sessionFactorySS" key="sessionFactorySS"/>
            </map> 
        </property>
    </bean>
    <bean class="com.ivo.core.HibernateInvoker">
		<property name="sessionFactory" ref="sessionFactory" />
	</bean>
 	<!-- 自定义动态切换SessionFactory事务管理器,注入sessionFactory  -->
    <bean id="transactionManager" class="com.ivo.util2.DynamicTransactionManager">
        <property name="sessionFactory" ref="sessionFactory" />
    </bean>

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

com.ivo.util2.DynamicSessionFactoryImpl 和 <span style="font-family: Arial, Helvetica, sans-serif;">com.ivo.util2.DynamicTransactionManager</span>

先写一个继承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<Object, SessionFactory> 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<String, ClassMetadata> 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<Object, SessionFactory> 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<String> contextHolder = new ThreadLocal<String>();  
	      
	    public static void setCustomerType(String dbType) {  
	        contextHolder.set(dbType);  
	    }  
	      
	    public static String getCustomerType() {  
	        return contextHolder.get();  
	    }  
	      
	    public static void clearCustomerType() {  
	        contextHolder.remove();  
	    }  
}

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


  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值