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. }