Spring利用继承DelegatingDataSource实现多数据源切换

最新推荐文章于 2024-09-23 21:54:20 发布

原创最新推荐文章于 2024-09-23 21:54:20 发布 · 755 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Spring #thread #JDBC

j2ee 专栏收录该内容

4 篇文章

订阅专栏

本文介绍了一种通过自定义DataSource实现按用户所在区域切换数据源的方法。该方法通过继承DelegatingDataSource并重写getConnection方法来实现，能够根据用户所在的区域选择对应的数据源。

根据用户所在的区域切换到自己区域的数据源，

自定义DataSource继承org.springframework.jdbc.datasource.DelegatingDataSource并且重写

public Connection getConnection() throws SQLException方法

 public Connection getConnection() throws SQLException {
    	AgilityUserBean userBean = ThreadContext.getUser();
    	if(userBean != null){
			System.out.println("the system will user " + dataSourceName + userBean.getLocale() + " dataSource");
			ds = ApplicationContext.getInstance().getBean(dataSourceName + userBean.getLocale());
				
    		if(ds == null){
    			throw new SQLException("can't find the dataSource for this region.");
    		}
			
    		setTargetDataSource((DataSource)ds);
    	}else{
    		System.out.println("System use default dataSource " + dataSourceName);
    	}

        Assert.state(getTargetDataSource() != null, "targetDataSource is required");
        Connection con = null;
        try {
            con = getTargetDataSource().getConnection();
        } catch (SQLException e) {
            logger.error("Error occured during connection request. Thread name = " +
                    Thread.currentThread().getName());
            logger.error("Listing binded to thread resources...");
            throw e;
        }
        return con;
    }

经过简单的验证可行，不知道这样是否完全正确，希望大家拍砖！