在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>这样的结构，来做内部的转发。

 

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

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