Hibernate根据方言dialect动态连接多数据库

Hibernate根据方言dialect动态连接多数据库

由于最近需要在web项目中动态链接不同地址的数据库，且要链接的子数据库的信息在总数据库（即web项目的主数据库）的表中，所以动手写了一个类。

web项目试用了hibernate，动态生成的子数据库链接打算也用hibernate，虽然动态生成的sessionfactory类，以及Configuration配置没有子数据库的对象关系映射，但是使用 native SQL 也方便。

代码如下：

1. public class TempSessionFactory {
2. /**
3. * Location of hibernate.cfg.xml file.
4. * Location should be on the classpath as Hibernate uses
5. * #resourceAsStream style lookup for its configuration file.
6. * The default classpath location of the hibernate config file is
7. * in the default package. Use #setConfigFile() to update
8. * the location of the configuration file for the current session.
9. */
10. //private static String CONFIG_FILE_LOCATION = "/hibernate.cfg.xml";
11. private final ThreadLocal<Session> threadLocal = new ThreadLocal<Session>();
12. private Configuration configuration = new Configuration();
13. private org.hibernate.SessionFactory sessionFactory;
14. //private static String configFile = CONFIG_FILE_LOCATION;
15. /*  static {
16. try {
17. configuration.configure(configFile);
18. sessionFactory = configuration.buildSessionFactory();
19. } catch (Exception e) {
20. System.err
21. .println("%%%% Error Creating SessionFactory %%%%");
22. e.printStackTrace();
23. }
24. }*/
25. public void setConfiguration(String dialect, String driverClass,
26. String ipAddress, String port, String dataBaseName,
27. String username, String password) {
28. String connection_url = "";
29. Configuration configuration = new Configuration();
30. if (dialect.indexOf("MySQL") > -1) {
31. System.out.println("%%%% DataBase type is MySql %%%%");
32. connection_url = "jdbc:mysql://" + ipAddress + "/" + dataBaseName;
33. } else if (dialect.indexOf("SQLServer") > -1) {
34. System.out.println("%%%% DataBase type is SQLServer %%%%");
35. connection_url = "jdbc:sqlserver://" + ipAddress + ":" + port
36. + ";DataBaseName=" + dataBaseName;
37. } else if (dialect.indexOf("Oracle") > -1) {
38. System.out.println("%%%% DataBase type is Oracle %%%%");
39. connection_url = "jdbc:oracle:thin:@" + ipAddress + ":" + port
40. + ":" + dataBaseName;
41. // configuration.setProperty("hibernate.connection.oracle.jdbc.V8Compatible","true");
42. }
43. configuration.setProperty("hibernate.dialect", dialect);
44. configuration.setProperty("hibernate.connection.url", connection_url);
45. configuration.setProperty("hibernate.connection.driver_class",
46. driverClass);
47. configuration.setProperty("hibernate.connection.username", username);
48. configuration.setProperty("hibernate.connection.password", password);
49. // configuration.setProperty("hibernate.default_schema", "dbo");
50. // configuration.setProperty("hibernate.default_catalog", dataBaseName);
51. // configuration.setProperty("hibernate.show_sql", "true");
52. this.configuration = configuration;
53. }
54. /**
55. * Returns the ThreadLocal Session instance.  Lazy initialize
56. * the <code>SessionFactory</code> if needed.
57. *
58. *  @return Session
59. *  @throws HibernateException
60. *
61. */
62. public Session getSession() throws HibernateException {
63. Session session = (Session) threadLocal.get();
64. if (session == null || !session.isOpen()) {
65. if (sessionFactory == null) {
66. rebuildSessionFactory();
67. }
68. session = (sessionFactory != null) ? sessionFactory.openSession()
69. : null;
70. threadLocal.set(session);
71. }
72. return session;
73. }
74. /**
75. *  Rebuild hibernate session factory
76. *
77. */
78. public void rebuildSessionFactory() {
79. try {
80. //configuration.configure(configFile);
81. sessionFactory = this.configuration.buildSessionFactory();
82. } catch (Exception e) {
83. System.err
84. .println("%%%% Error Creating SessionFactory %%%%");
85. e.printStackTrace();
86. }
87. }
88. /**
89. *  Close the single hibernate session instance.
90. *
91. *  @throws HibernateException
92. */
93. public void closeSession() throws HibernateException {
94. Session session = (Session) threadLocal.get();
95. threadLocal.set(null);
96. if (session != null) {
97. session.close();
98. }
99. }
100. /**
101. *  return session factory
102. *
103. */
104. public org.hibernate.SessionFactory getSessionFactory() {
105. return sessionFactory;
106. }
107. /**
108. *  return session factory
109. *
110. *  session factory will be rebuilded in the next call
111. */
112. /*  public static void setConfigFile(String configFile) {
113. HibernateSessionFactory.configFile = configFile;
114. sessionFactory = null;
115. }*/
116. /**
117. *  return hibernate configuration
118. *
119. */
120. public Configuration getConfiguration() {
121. return configuration;
122. }
123. }

测试类代码： 
在 databasename1 库中建 testtable表，字段id,name2个 
在 databasename2 库中建 testtable2表，字段id,name2个

1. public class TestCase1 {
2. public static void main(String[] args) {
3. try{
4. Configuration configuration1 = new Configuration();
5. TempSessionFactory tempSessionFactory1 = new TempSessionFactory(configuration1);
6. tempSessionFactory1.setConfiguration("org.hibernate.dialect.SQLServerDialect","com.microsoft.sqlserver.jdbc.SQLServerDriver",
7. "jdbc:sqlserver://*1.*1.*1.*1","1433","databasename1","sa","sa");
8. Session session1=tempSessionFactory1.getSession();
9. Transaction tx1 = session1.beginTransaction();
10. Query query1 = session1.createSQLQuery("select  name as  aaa  from testtable ").setResultTransformer(
11. Transformers.ALIAS_TO_ENTITY_MAP);
12. Map obj1 = (Map)query1.setMaxResults(1).uniqueResult();
13. System.out.println("fd1111===="+obj1.get("aaa"));
14. Configuration configuration2 = new Configuration();
15. TempSessionFactory tempSessionFactory2 = new TempSessionFactory(configuration2);
16. tempSessionFactory2.setConfiguration("org.hibernate.dialect.SQLServerDialect","com.microsoft.sqlserver.jdbc.SQLServerDriver",
17. "jdbc:sqlserver://*2.*2.*2.*2","1433","databasename2","sa","sa");
18. Session session2=tempSessionFactory2.getSession();
19. Transaction tx2 = session2.beginTransaction();
20. Query query2 = session2.createSQLQuery("select  name as  aaa  from testtable2 ").setResultTransformer(
21. Transformers.ALIAS_TO_ENTITY_MAP);
22. Map obj2 = (Map)query2.setMaxResults(1).uniqueResult();
23. System.out.println("fd2222===="+obj2.get("aaa"));
24. }catch (Exception e) {
25. System.err.println(e);
26. // TODO: handle exception
27. }
28. }
29. }