2112A-ONE
/
muyu-source
6
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

09040919:修改同步资产结构的逻辑

master
冷调 2024-09-04 09:19:50 +08:00
parent 51fdd1f6c1
commit c4a9063516
3 changed files with 26 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
muyu-source-server/src/main/java/com/muyu/source/controller/DataValueController.java
View File

@ -30,8 +30,7 @@ public class DataValueController {
/** /**
* 根据基础表ID和SQL语句查询数据 * 根据基础表ID和SQL语句查询数据
* *
* @param basicId 基础表ID * @param dataValueModel 基础表ID和sql语句
* @param sql SQL语句
* @return DataValue{kltv} * @return DataValue{kltv}
*/ */
@PostMapping("/findTableValue") @PostMapping("/findTableValue")

2
muyu-source-server/src/main/java/com/muyu/source/controller/TableInfoController.java
View File

@ -258,6 +258,4 @@ public class TableInfoController {
} }
return Result.success(tableInfoList); return Result.success(tableInfoList);
} }
} }

25
muyu-source-server/src/main/java/com/muyu/source/service/Impl/DataSourceServiceImpl.java
View File

@ -6,6 +6,7 @@ import com.baomidou.mybatisplus.core.toolkit.Assert;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.muyu.common.core.utils.StringUtils; import com.muyu.common.core.utils.StringUtils;
import com.muyu.common.security.utils.SecurityUtils; import com.muyu.common.security.utils.SecurityUtils;
import com.muyu.source.domain.AssetAuthorization;
import com.muyu.source.domain.DataSource; import com.muyu.source.domain.DataSource;
import com.muyu.source.domain.Structure; import com.muyu.source.domain.Structure;
import com.muyu.source.domain.TableInfo; import com.muyu.source.domain.TableInfo;
@ -46,6 +47,8 @@ public class DataSourceServiceImpl extends ServiceImpl<DataSourceMapper, DataSou
private TableInfoService tableInfoService; private TableInfoService tableInfoService;
@Resource @Resource
private StructureService structureService; private StructureService structureService;
@Resource
private AssetAuthorizationService assetAuthorizationService;
private static final Long PARENTID = 0L; private static final Long PARENTID = 0L;
private static final String DRIVERNAME = "com.mysql.cj.jdbc.Driver"; private static final String DRIVERNAME = "com.mysql.cj.jdbc.Driver";
@ -160,11 +163,33 @@ public class DataSourceServiceImpl extends ServiceImpl<DataSourceMapper, DataSou
*/ */
@Override @Override
public Integer syncAssetStructure(DataSource etlDataScore) { public Integer syncAssetStructure(DataSource etlDataScore) {
try { try {
if (etlDataScore.getDataType().equals(MYSQL)) { if (etlDataScore.getDataType().equals(MYSQL)) {
MysqlPool mysqlPool = new MysqlPool(etlDataScore); MysqlPool mysqlPool = new MysqlPool(etlDataScore);
mysqlPool.init(); mysqlPool.init();
Connection conn = mysqlPool.getConn(); Connection conn = mysqlPool.getConn();
TableInfo tableInfoServiceOne = tableInfoService.getOne(new LambdaQueryWrapper<>() {{
eq(TableInfo::getBasicId, etlDataScore.getId());
}});
//如果存在就删除
if(StringUtils.isNotNull(tableInfoServiceOne)){
List<TableInfo> list = tableInfoService.list(new LambdaQueryWrapper<>() {{
eq(TableInfo::getBasicId, etlDataScore.getId());
}});
list.forEach(tableInfo -> {
structureService.remove(new LambdaQueryWrapper<>(){{
eq(Structure::getTableId, tableInfo.getId());
}});
});
tableInfoService.remove(new LambdaQueryWrapper<>(){{
eq(TableInfo::getBasicId, etlDataScore.getId());
}});
assetAuthorizationService.remove(new LambdaQueryWrapper<>(){{
eq(AssetAuthorization::getTableId, tableInfoServiceOne.getId())
.eq(AssetAuthorization::getBasicId, etlDataScore.getId());
}});
}
TableInfo tableInfoInsert = TableInfo.builder() TableInfo tableInfoInsert = TableInfo.builder()
.basicId(etlDataScore.getId()) .basicId(etlDataScore.getId())