Hibernate编程实现动态切换SChema访问不同的数据库

在Hibernate里面动态切换SChema实现访问不同的数据库的几种方法供大家考虑：

需求很简单，相同的操作，比如表结构完全相同，程序也完全相同，但需要根据某些条件，分别向不同的schema做操作。
比如，如果当前处理的是A公司，那么向SchemaA 里面保存数据，
如果当前处理的是B公司的，则向SchemaB里面保存数据。

其实就是一套程序，实现后台的动态切换。

我这里提供几种方法，大家自己根据情况考虑，都能实现，注意是实现，不一定适合于正式应用。
方法一：
在Hibernate里面，有一个配置参数，比如下面这个带Schema配置的映射

<?xml version="1.0" encoding="utf-8"?>
<!DOCTYPE hibernate-mapping PUBLIC "-//Hibernate/Hibernate Mapping DTD 3.0//EN"
"http://hibernate.sourceforge.net/hibernate-mapping-3.0.dtd">
<!--
    Mapping file autogenerated by MyEclipse Persistence Tools
-->
<hibernate-mapping>
    <class name="com.csc.poimanager.dao.Poi" table="POI" schema="P_BEIJING">
        <id name="poiId" type="java.lang.Long">
            <column name="POI_ID" precision="10" scale="0" />
            <generator class="increment" />
        </id>
        <property name="cnName" type="java.lang.String">
            <column name="CN_NAME" length="1000" />
        </property>
    </class>
</hibernate-mapping> 

 

其中的schema="P_BEIJING"将 Schema写死了。
我们可以不写这部分，而是使用系统的配置参数
<property name="hibernate.default_schema">POI_BEIJING</property>

这样的话，我们的映射文件就变成了

<?xml version="1.0" encoding="utf-8"?>
<!DOCTYPE hibernate-mapping PUBLIC "-//Hibernate/Hibernate Mapping DTD 3.0//EN"
"http://hibernate.sourceforge.net/hibernate-mapping-3.0.dtd">
<!--
    Mapping file autogenerated by MyEclipse Persistence Tools
-->
<hibernate-mapping>
    <class name="com.csc.poimanager.dao.Poi" table="POI"">
        <id name="poiId" type="java.lang.Long">
            <column name="POI_ID" precision="10" scale="0" />
            <generator class="increment" />
        </id>
        <property name="cnName" type="java.lang.String">
            <column name="CN_NAME" length="1000" />
        </property>
    </class>
</hibernate-mapping>

 

在调用的时候，动态的指定Schema的参数就行了，比如

 public static SessionFactory rebuildSessionFactoryForChangeSchema(String newSchema){
        try {
            Properties p = configuration.getProperties();
            System.out.println("---" + p);
            p.put("hibernate.default_schema", newSchema);
            sessionFactory = configuration.buildSessionFactory();
            System.out.println(" change schema successfully ......... ");
            return sessionFactory;
        } catch (Exception e) {
            System.err
                    .println("%%%% rebuild session factory failed for changing schema %%%%");
            e.printStackTrace();
            return null;
        }
    }

 这个方法能实现切换，但是我们必须每次都返回一个SessionFactory, 否则在并发的时候就会出问题。因为hibernate.default_schema系统只有一个。

 

方法二：
使用多重配置，比如针对天津和北京，我们分别编写对应的映射文件

<?xml version="1.0" encoding="utf-8"?>
<!DOCTYPE hibernate-mapping PUBLIC "-//Hibernate/Hibernate Mapping DTD 3.0//EN"
"http://hibernate.sourceforge.net/hibernate-mapping-3.0.dtd">
<!--
    Mapping file autogenerated by MyEclipse Persistence Tools
-->
<hibernate-mapping>
    <class name="com.csc.poimanager.dao.Poi" table="POI" id="P_BEIJING" schema="P_BEIJING">
        <id name="poiId" type="java.lang.Long">
            <column name="POI_ID" precision="10" scale="0" />
            <generator class="increment" />
        </id>
        <property name="cnName" type="java.lang.String">
            <column name="CN_NAME" length="1000" />
        </property>
    </class>
    <class name="com.csc.poimanager.dao.Poi" id="P_TIANJIN" table="POI" schema="P_TIANJIN">
        <id name="poiId" type="java.lang.Long">
            <column name="POI_ID" precision="10" scale="0" />
            <generator class="increment" />
        </id>
        <property name="cnName" type="java.lang.String">
            <column name="CN_NAME" length="1000" />
        </property>
    </class>
</hibernate-mapping>

 配置文件除了SCHEMA和ID不同外，其它的完全相同，那么如何使用呢？
sessionFactory.getBean("P"+placeName);
这样就可以实现动态调用了。
至于placeName，可以通过前台调用程序传递过来，也可以放到ThreadLocal里面，后者更具有通用性，而且对以前的程序修改量最少，因为只需要写一个getPlaceName()得方法就行了。

方法3：
修改数据库的连接属性，将以前指向SchemaA的改成指向SchemaB,这需要一些高权限的账号做这些事情。
其实，还有一个更简单的方法，那就是自己继承一个DataSource,然后在方法里面，根据ThreadLocal的参数，返回不同的数据连接。比如如果是BEIGJING,则返回 BeiJingDatasource的连接，如果是TIANJIN，则返回TianJinDatasource的连接。
系统在使用时，肯定会从DataSource里面获取连接，我们在内部做了分配，自然前台运行结果也就分配开了。
提示：在你的DataSource里面可以包含Map<String,DataSource>这样的结构，来做内部的转发。


个人总结：
第一种方法只适合于单机，单线程使用。
第二个方法，相对技术难度低一点，单配置起来要写多份，维护比较麻烦
第三个方法，技术难度高，单维护起来简单

我个人当然是用第三个方法了，很有趣。

 

本文来自: IT知道网(http://www.itwis.com) 详细出处参考：http://www.itwis.com/html/java/hibernate/20090824/5678.html