Examples with GroupDataSource com.baomidou.dynamic.datasource.ds.GroupDataSource used on opensource projects

Example 1 with GroupDataSource

use of com.baomidou.dynamic.datasource.ds.GroupDataSource in project dynamic-datasource-spring-boot-starter by baomidou.

the class DynamicRoutingDataSource method determinePrimaryDataSource.

private DataSource determinePrimaryDataSource() {
    log.debug("dynamic-datasource switch to the primary datasource");
    DataSource dataSource = dataSourceMap.get(primary);
    if (dataSource != null) {
        return dataSource;
    }
    GroupDataSource groupDataSource = groupDataSources.get(primary);
    if (groupDataSource != null) {
        return groupDataSource.determineDataSource();
    }
    throw new CannotFindDataSourceException("dynamic-datasource can not find primary datasource");
}

Example 2 with GroupDataSource

use of com.baomidou.dynamic.datasource.ds.GroupDataSource in project dynamic-datasource-spring-boot-starter by baomidou.

the class DynamicRoutingDataSource method addGroupDataSource.

/**
 * 新数据源添加到分组
 *
 * @param ds         新数据源的名字
 * @param dataSource 新数据源
 */
private void addGroupDataSource(String ds, DataSource dataSource) {
    if (ds.contains(UNDERLINE)) {
        String group = ds.split(UNDERLINE)[0];
        GroupDataSource groupDataSource = groupDataSources.get(group);
        if (groupDataSource == null) {
            try {
                groupDataSource = new GroupDataSource(group, strategy.getDeclaredConstructor().newInstance());
                groupDataSources.put(group, groupDataSource);
            } catch (Exception e) {
                throw new RuntimeException("dynamic-datasource - add the datasource named " + ds + " error", e);
            }
        }
        groupDataSource.addDatasource(ds, dataSource);
    }
}