在Hibernate里面动态切换SChema实现访问不同的数据库的几种方法

在Hibernate里面动态切换SChema实现访问不同的数据库的几种方法

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

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

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

[xhtml]  view plain copy

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

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

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

[xhtml]  view plain copy

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

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

[java]  view plain copy

1. public static SessionFactory rebuildSessionFactoryForChangeSchema(String newSchema){  
2.        try {  
3.            Properties p = configuration.getProperties();  
4.            System.out.println("---" + p);  
5.            p.put("hibernate.default_schema", newSchema);  
6.            sessionFactory = configuration.buildSessionFactory();  
7.            System.out.println(" change schema successfully ......... ");  
8.            return sessionFactory;  
9.        } catch (Exception e) {  
10.            System.err  
11.                    .println("%%%% rebuild session factory failed for changing schema %%%%");  
12.            e.printStackTrace();  
13.            return null;  
14.        }  
15.    }  

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


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

[xhtml]  view plain copy

1. <?xml version="1.0" encoding="utf-8"?>  
2. <!DOCTYPE hibernate-mapping PUBLIC "-//Hibernate/Hibernate Mapping DTD 3.0//EN"  
3. "http://hibernate.sourceforge.net/hibernate-mapping-3.0.dtd">  
4. <!-- 
5.     Mapping file autogenerated by MyEclipse Persistence Tools 
6. -->  
7. <hibernate-mapping>  
8.     <class name="com.csc.poimanager.dao.Poi" table="POI" id="P_BEIJING" schema="P_BEIJING">  
9.         <id name="poiId" type="java.lang.Long">  
10.             <column name="POI_ID" precision="10" scale="0" />  
11.             <generator class="increment" />  
12.         </id>  
13.         <property name="cnName" type="java.lang.String">  
14.             <column name="CN_NAME" length="1000" />  
15.         </property>  
16.     </class>  
17.     <class name="com.csc.poimanager.dao.Poi" id="P_TIANJIN" table="POI" schema="P_TIANJIN">  
18.         <id name="poiId" type="java.lang.Long">  
19.             <column name="POI_ID" precision="10" scale="0" />  
20.             <generator class="increment" />  
21.         </id>  
22.         <property name="cnName" type="java.lang.String">  
23.             <column name="CN_NAME" length="1000" />  
24.         </property>  
25.     </class>  
26. </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>这样的结构，来做内部的转发。

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

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