多租户多数据源切换

在很多系统中，都存在着租户的概念。更具需求的不同，系统可以分为3种类型

• 方式一:每个租户有独立的服务和独立的数据库
• 方式二:每个租户有共享的服务和独立的数据库
• 方式三:每个租户有共享的服务和共享的数据库

方式1和方式3和我们日常的应用并无不同。但方式二的实现就需要做些改动了

这里我参考了一个主从分离的例子，根据租户的身份特征选择相对应的数据源。同时，还应做到动态的添加租户和数据源

参考了读写分离的配置，总共分为4步

1.继承AbstractRoutingDataSource

public class DynamicDataSource extends AbstractRoutingDataSource {
    
}

2.添加数据源
每一个数据源都会有一个标识key，数据源和标识key保存在map，通过标识key找到该数据源

DynamicDataSource dynamicDataSource = new DynamicDataSource();
        DataSource master = masterDataSource();
        DataSource slave = slaveDataSource();
        //设置默认数据源
        dynamicDataSource.setDefaultTargetDataSource(master);//默认从库
        //配置多数据源
        Map<Object, Object> map = new HashMap<>();
        map.put(DataSourceType.Master.getName(), master);    //key需要跟ThreadLocal中的值对应
        map.put(DataSourceType.Slave.getName(), slave);
        dynamicDataSource.setTargetDataSources(map);

3.选择数据源
重写AbstractRoutingDataSource的determineCurrentLookupKey，即每次想切换数据的时候修改CurrentLookupKey，这样就能找到该key对应的数据源。

@Override
    protected Object determineCurrentLookupKey() {
        logger.info("数据源为{}", JdbcContextHolder.getDataSource());
        return JdbcContextHolder.getDataSource();
    }

4.切面判断选择key

---

数据源选择

有了上面的基础，现在我们要实现根据租户选择数据源也是非常简单

我们想让以下方法自动选择数据源

我们制定好了任何需要切换数据源的方法首个参数必须是cusId的规则
(注意:项目可以直接从登录租户或者其他方法拿到cusId)

public List<Custom> getList(String cusId) {
                return  customService.list();
        }

依然还是4步

前三步都一样，最后一步我们需要拿到方法的首位参数，代码如下

1.定义注解

@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface AutoDataSource {
    DataSourceType value() default DataSourceType.Master;  
}

2.切面逻辑

@Before("aspect()")
    private void before(JoinPoint point) {
            Object target = point.getTarget();
            String method = point.getSignature().getName();
            Class<?> classz = target.getClass();
            Class<?>[] parameterTypes = ((MethodSignature) point.getSignature()).getMethod().getParameterTypes();
            try {
                    Method m = classz.getMethod(method, parameterTypes);
                    if (m != null && m.isAnnotationPresent(AutoDataSource.class)) {
//              AutoDataSource data = m.getAnnotation(AutoDataSource.class);
                            Object[] args = point.getArgs();
                            Object sourceKey = args[0];
                            JdbcContextHolder.putDataSource(sourceKey + "");
                            logger.info("{}-当前数据源:{}", method, sourceKey);
                    }
            } catch (Exception e) {
                    e.printStackTrace();
            }
    }

JdbcContextHolder内部是个ThreadLocal

public class JdbcContextHolder {
    private final static ThreadLocal<String> local = new ThreadLocal<>();

    public static void putDataSource(String name) {
        local.set(name);
    }

    public static String getDataSource() {
        return local.get();
    }

3.添加注解即可

@AutoDataSource
        public List<Custom> getList(String cusId) {
                return  customService.list();
        }

---

动态添加数据源

下面要实现的是在不停机的情况下，动态添加数据源。通过上文我们知道，数据源的添加是通过

dynamicDataSource.setTargetDataSources(map)

这行代码实现的。实际上也就是把我们的数据源信息保存在了AbstractRoutingDataSource的一个map集合中。但是这个map是私有类型的，而且也没有提供get方法。我们无法直接获取到map里的数据

@Nullable
private Map<Object, Object> targetDataSources;

有两种办法

• 通过反射拿到AbstractRoutingDataSource的targetDataSources
• 自己维护一个map，相当于加了一层代理

这里选择第二种方法

还是在DynamicDataSource类中创建一个map，并重写setTargetDataSources方法

private Map<Object, Object> dynamicTargetDataSources = new HashMap<>();

@Override
public void setTargetDataSources(Map<Object, Object> targetDataSources) {
        super.setTargetDataSources(targetDataSources);
        this.dynamicTargetDataSources = targetDataSources;
}

在拿到map数据之后我们再添加一个新增方法

/**
 * 新增数据源
 * @param key 数据源标识
 * @param dataSource 数据源
 */
public void addTargetDataSources(Object key, Object dataSource) {
        dynamicTargetDataSources.put(key, dataSource);
        super.setTargetDataSources(dynamicTargetDataSources);
}

这里好像就有点问题了，数据是加载到TargetDataSources了，但是项目只会在启动的时候去解析map中的数据。AbstractRoutingDataSource实现了InitializingBean接口，实现了afterPropertiesSet方法，所以我们需要手动触发下

public void afterPropertiesSet() {
    if (this.targetDataSources == null) {
        throw new IllegalArgumentException("Property 'targetDataSources' is required");
    }
    this.resolvedDataSources = new HashMap<>(this.targetDataSources.size());
    this.targetDataSources.forEach((key, value) -> {
        Object lookupKey = resolveSpecifiedLookupKey(key);
        //解析数据源
        DataSource dataSource = resolveSpecifiedDataSource(value);
        this.resolvedDataSources.put(lookupKey, dataSource);
    });
    if (this.defaultTargetDataSource != null) {
        this.resolvedDefaultDataSource = resolveSpecifiedDataSource(this.defaultTargetDataSource);
    }
}

修改下新增的方法，每次都要调用一次afterPropertiesSet

/**
 * 新增数据源
 * @param key 数据源标识
 * @param dataSource 数据源
 */
public void addTargetDataSources(Object key, Object dataSource) {
        dynamicTargetDataSources.put(key, dataSource);
        super.setTargetDataSources(dynamicTargetDataSources);
        super.afterPropertiesSet();
}

接下来就很简单了，对外暴露一个接口用于新增数据源，数据源的key便是租户的身份特征编号。这些代码就不再描述！

转载于:https://www.cnblogs.com/xmzJava/p/11608145.html