nutz中实现多数据库切换

 Java代码nutz很好的实现了表切换

根据它的思路写了个数据库切换，测试可行！

Java代码

1. public class MutiDataSource implements DataSource {
3. private static ConcurrentHashMap<String, DruidDataSource> dataSourceMap = new ConcurrentHashMap<String, DruidDataSource>();
5. protected HashMap<String, String> dsdbMap;
6. protected String driverClassName;
7. protected String url;
8. protected String username;
9. protected String password;
10. protected int maxActive;
12. public MutiDataSource() {
14. }
16. public void init() throws SQLException {
17. Iterator<String> it = dsdbMap.keySet().iterator();
18. String dsname, dbname;
19. while (it.hasNext()) {
20. dsname = it.next();
21. dbname = dsdbMap.get(dsname);
22. DruidDataSource ds = createDataSource(dbname);
23. dataSourceMap.put(dsname, ds);
24. }
25. }
27. public void close() {
28. Iterator<DruidDataSource> it = dataSourceMap.values().iterator();
29. while (it.hasNext()) {
30. it.next().close();
31. }
32. }
34. protected DruidDataSource createDataSource(String dbname) {
35. DruidDataSource ds = new DruidDataSource();
36. ds.setDriverClassName(driverClassName);
37. ds.setUsername(username);
38. ds.setPassword(password);
39. ds.setUrl(url + dbname);
40. return ds;
41. }
43. public DataSource getDs() {
44. return dataSourceMap.get(DataSourceNames.get());
45. }
47. public void setDriverClassName(String driverClassName) {
48. this.driverClassName = driverClassName;
49. }
51. public void setUrl(String url) {
52. this.url = url;
53. }
55. public void setUsername(String username) {
56. this.username = username;
57. }
59. public void setPassword(String password) {
60. this.password = password;
61. }
63. public void setMaxActive(int maxActive) {
64. this.maxActive = maxActive;
65. }
67. @Override
68. public PrintWriter getLogWriter() throws SQLException {
69. return getDs().getLogWriter();
70. }
72. @Override
73. public int getLoginTimeout() throws SQLException {
74. return getDs().getLoginTimeout();
75. }
77. @Override
78. public void setLogWriter(PrintWriter out) throws SQLException {
79. getDs().setLogWriter(out);
80. }
82. @Override
83. public void setLoginTimeout(int seconds) throws SQLException {
84. getDs().setLoginTimeout(seconds);
85. }
87. @Override
88. public boolean isWrapperFor(Class<?> iface) throws SQLException {
89. return getDs().isWrapperFor(iface);
90. }
92. @Override
93. public <T> T unwrap(Class<T> iface) throws SQLException {
94. return getDs().unwrap(iface);
95. }
97. @Override
98. public Connection getConnection() throws SQLException {
99. return getDs().getConnection();
100. }
102. @Override
103. public Connection getConnection(String username, String password)
104. throws SQLException {
105. return getDs().getConnection(username, password);
106. }
108. }

public class MutiDataSource implements DataSource {
	private static ConcurrentHashMap<String, DruidDataSource> dataSourceMap = new ConcurrentHashMap<String, DruidDataSource>();
	protected HashMap<String, String> dsdbMap;
	protected String driverClassName;
	protected String url;
	protected String username;
	protected String password;
	protected int maxActive;
	public MutiDataSource() {
	}
	public void init() throws SQLException {
		Iterator<String> it = dsdbMap.keySet().iterator();
		String dsname, dbname;
		while (it.hasNext()) {
			dsname = it.next();
			dbname = dsdbMap.get(dsname);
			DruidDataSource ds = createDataSource(dbname);
			dataSourceMap.put(dsname, ds);
		}
	}
	public void close() {
		Iterator<DruidDataSource> it = dataSourceMap.values().iterator();
		while (it.hasNext()) {
			it.next().close();
		}
	}
	protected DruidDataSource createDataSource(String dbname) {
		DruidDataSource ds = new DruidDataSource();
		ds.setDriverClassName(driverClassName);
		ds.setUsername(username);
		ds.setPassword(password);
		ds.setUrl(url + dbname);
		return ds;
	}
	public DataSource getDs() {
		return dataSourceMap.get(DataSourceNames.get());
	}
	public void setDriverClassName(String driverClassName) {
		this.driverClassName = driverClassName;
	}
	public void setUrl(String url) {
		this.url = url;
	}
	public void setUsername(String username) {
		this.username = username;
	}
	public void setPassword(String password) {
		this.password = password;
	}
	public void setMaxActive(int maxActive) {
		this.maxActive = maxActive;
	}
	@Override
	public PrintWriter getLogWriter() throws SQLException {
		return getDs().getLogWriter();
	}
	@Override
	public int getLoginTimeout() throws SQLException {
		return getDs().getLoginTimeout();
	}
	@Override
	public void setLogWriter(PrintWriter out) throws SQLException {
		getDs().setLogWriter(out);
	}
	@Override
	public void setLoginTimeout(int seconds) throws SQLException {
		getDs().setLoginTimeout(seconds);
	}
	@Override
	public boolean isWrapperFor(Class<?> iface) throws SQLException {
		return getDs().isWrapperFor(iface);
	}
	@Override
	public <T> T unwrap(Class<T> iface) throws SQLException {
		return getDs().unwrap(iface);
	}
	@Override
	public Connection getConnection() throws SQLException {
		return getDs().getConnection();
	}
	@Override
	public Connection getConnection(String username, String password)
			throws SQLException {
		return getDs().getConnection(username, password);
	}
}

Java代码

1. public class DataSourceNames {
2. private static final ThreadLocal<String> ds = new ThreadLocal<String>();
4. public static String get() {
5. return ds.get();
6. }
8. public static String set(String obj) {
9. String re = get();
10. ds.set(obj);
11. return re;
12. }
14. public static void clear() {
15. set(null);
16. }
18. }  

转载于:https://blog.51cto.com/daheyuan/1139368