springboot实现动态分库

springboot实现动态分库

前段时间公司要开发一套企业级的Saas系统，为了方便部署和迁移因此每个企业的数据库是独立的，数据库之间没有关联。库以及表动态创建的，只有用户注册后才会动态创建。

该功能实现起来比较简单，因为数据库是独立的，因此不需要考虑事务问题、跨库查询等分库分表的常见问题。但是因为因为云端部署多个数据库成本很高并且会造资源浪费，因此采用的是逻辑分库的方式及创建多个schema的方式进行分库。

我们目前使用的开发架构是springboot+mybatis，但是mybatis无法满足动态schema的需求，因此改用Springboot+SpringJDBC的方式。

主要的依赖如下：

    <dependency><groupId>com.zaxxer</groupId><artifactId>HikariCP</artifactId><version>3.3.1</version></dependency><dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-jdbc</artifactId></dependency>

JdbcTemplate本身并不支持动态设置schema的功能，因此需要重新改造JdbcTemplate使其支持逻辑schema的动态设置。
在这里只展示部分代码，如果想查看全部代码可以下载查看。。

@Override
@Nullable
public <T> T execute(StatementCallback<T> action, String schema) throws DataAccessException {Null(action, "Callback object must not be null");Connection con = Connection(obtainDataSource());try {con.setCatalog(schema);} catch (SQLException e) {e.printStackTrace();}try {con.setCatalog(schema);} catch (SQLException e) {e.printStackTrace();}Statement stmt = null;try {stmt = ateStatement();applyStatementSettings(stmt);T result = action.doInStatement(stmt);handleWarnings(stmt);return result;}catch (SQLException ex) {// Release Connection early, to avoid potential connection pool deadlock// in the case when the exception translator hasn't been initialized yet.String sql = getSql(action);JdbcUtils.closeStatement(stmt);stmt = leaseConnection(con, getDataSource());con = null;throw translateException("StatementCallback", sql, ex);}finally {JdbcUtils.closeStatement(stmt);leaseConnection(con, getDataSource());}
}

接下来就可以根据自身业务动态设置schema了

List<Map<String,Object>>  list1= db1JdbcTemplate.queryForList("select * from user","demo_ds_0");
List<Map<String,Object>>  list2= db1JdbcTemplate.queryForList("select * from db_info","demo_ds_1");
其中demo_ds_0、demo_ds_0是逻辑分库。

通过Springboot将MyJdbcTemplate注入。

@Bean(name = "db1JdbcTemplate")
public MyJdbcTemplate Db1JdbcTemplate(@Qualifier("db1DataSource")DataSource dataSource) {return new MyJdbcTemplate(dataSource);
}@Primary
@Bean(name = "db1DataSourceProperties")
@Qualifier("db1DataSource")
@ConfigurationProperties(prefix = "spring.datasource.db1")
public DataSourceProperties db1DataSourceProperties() {return new DataSourceProperties();
}@Primary
@Bean(name = "db1DataSource")
@Qualifier("db1DataSource")
@ConfigurationProperties(prefix = "spring.datasource.db1")
public DataSource db1DataSource() {return db1DataSourceProperties().initializeDataSourceBuilder().build();
}

同时SpringJDBC支持多数据源，这样一来业务基本满足。

。