多租户多数据源切换
在很多系统中，都存在着租户的概念。更具需求的不同，系统可以分为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便是租户的身份特征编号。这些代码就不再描述！
