package cn.stylefeng.roses.kernel.migration.aggregation.scheduling;
|
|
import cn.hutool.core.util.ObjectUtil;
|
import cn.stylefeng.roses.kernel.migration.api.AccessMigrationApi;
|
import cn.stylefeng.roses.kernel.migration.api.constants.MigrationConstants;
|
import cn.stylefeng.roses.kernel.migration.api.pojo.MigrationAggregationPOJO;
|
import cn.stylefeng.roses.kernel.migration.api.pojo.MigrationInfo;
|
|
import java.util.ArrayList;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.concurrent.ConcurrentHashMap;
|
|
/**
|
* 调度中心
|
*
|
* @author majianguo
|
* @date 2021/7/6 16:01
|
*/
|
public class SchedulingCenter {
|
|
/**
|
* 迁移数据接入集合
|
*/
|
private static final ConcurrentHashMap<String, AccessMigrationApi> migrationCollection = new ConcurrentHashMap<>();
|
|
/**
|
* 接入一个迁移类
|
*
|
* @param accessMigrationApi 迁移接口
|
* @author majianguo
|
* @date 2021/7/6 17:21
|
**/
|
public static void addMigration(AccessMigrationApi accessMigrationApi) {
|
|
// 拼接key值
|
String appAndModuleNameItem = accessMigrationApi.getAppName() + MigrationConstants.NAME_SEPARATOR + accessMigrationApi.getModuleName();
|
|
// 存储迁移类
|
migrationCollection.put(appAndModuleNameItem, accessMigrationApi);
|
}
|
|
/**
|
* 导入数据
|
*
|
* @param migrationAggregationPOJO 导入的数据
|
* @author majianguo
|
* @date 2021/7/6 16:10
|
**/
|
public static void importData(MigrationAggregationPOJO migrationAggregationPOJO, String type) {
|
|
// 所有导入数据集合
|
Map<String, MigrationInfo> migrationInfoMap = migrationAggregationPOJO.getData();
|
|
// 找到每个应用的处理程序，调用相关方法
|
for (String appAndModuleName : migrationAggregationPOJO.getAppAndModuleNameList()) {
|
AccessMigrationApi accessMigrationApi = migrationCollection.get(appAndModuleName);
|
if (ObjectUtil.isNotEmpty(accessMigrationApi)) {
|
|
// 找到数据
|
MigrationInfo migrationInfo = migrationInfoMap.get(appAndModuleName);
|
|
// 发送给实现类处理
|
try {
|
accessMigrationApi.importData(type, migrationInfo);
|
} catch (Exception e) {
|
e.printStackTrace();
|
}
|
}
|
}
|
}
|
|
/**
|
* 导出数据
|
*
|
* @param migrationAggregationPOJO 导出要求
|
* @return {@link MigrationAggregationPOJO}
|
* @author majianguo
|
* @date 2021/7/6 16:11
|
**/
|
public static MigrationAggregationPOJO exportData(MigrationAggregationPOJO migrationAggregationPOJO) {
|
Map<String, MigrationInfo> data = migrationAggregationPOJO.getData();
|
if (ObjectUtil.isEmpty(data)) {
|
data = new HashMap<>();
|
migrationAggregationPOJO.setData(data);
|
}
|
|
// 导出所有指定项
|
for (String appAndModuleName : migrationAggregationPOJO.getAppAndModuleNameList()) {
|
// 查找对应的实现类
|
AccessMigrationApi accessMigrationApi = migrationCollection.get(appAndModuleName);
|
if (ObjectUtil.isNotEmpty(accessMigrationApi)) {
|
// 执行导出
|
try {
|
// 获取数据
|
MigrationInfo migrationInfo = accessMigrationApi.exportData();
|
|
// 放入返回结果中
|
data.put(appAndModuleName, migrationInfo);
|
} catch (Exception e) {
|
e.printStackTrace();
|
}
|
} else {
|
data.put(appAndModuleName, new MigrationInfo());
|
}
|
}
|
|
return migrationAggregationPOJO;
|
}
|
|
/**
|
* 获取所有迁移数据列表
|
*
|
* @return {@link java.util.List<MigrationInfo>}
|
* @author majianguo
|
* @date 2021/7/6 16:13
|
**/
|
public static List<String> getAllMigrationInfo() {
|
return new ArrayList<>(migrationCollection.keySet());
|
}
|
}
|
