iidi
/
bocloud.ams
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

查询实例总数

develop
guyuliang 2024-08-24 19:29:02 +08:00
parent 1f83038120
commit a816dffe8c
1 changed files with 33 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
bocloud.ams.service/src/main/java/com/bocloud/ams/service/module/impl/ModuleServiceImpl.java
View File

@ -70,6 +70,7 @@ import com.megatron.framework.lock.AutoCloseLock;
import com.megatron.framework.lock.LockFactory;
import lombok.extern.slf4j.Slf4j;
import org.yaml.snakeyaml.Yaml;
/**
* @author 胡立伟
@ -441,11 +442,10 @@ public class ModuleServiceImpl implements ModuleService {
for (ModuleGroup rootGroup : rootGroups) {
//获取每个模型的所有实例总数
getInstaceCount(rootGroup,resultMap,countDownLatch);
getInstaceCount(rootGroup, resultMap, countDownLatch, context.getTarget());
}
countDownLatch.await();
//查询云主机总数
Pager pager = new Pager();
pager.setPage(1);
@ -471,10 +471,11 @@ public class ModuleServiceImpl implements ModuleService {
}
}
private void getInstaceCount(ModuleGroup rootGroup, Map<String, Object> resultMap, CountDownLatch countDownLatch) {
private void getInstaceCount(ModuleGroup rootGroup, Map<String, Object> resultMap, CountDownLatch countDownLatch, Long userId) {
new Thread(new Runnable() {
@Override
public void run() {
try {
//获取当前ModuleGroup下所有的modle
Pager pager = new Pager();
GeneralResult<GridBean<CiModule>> modelList = getModelList(pager);
@ -483,12 +484,22 @@ public class ModuleServiceImpl implements ModuleService {
countDownLatch.countDown();
return;
}
Integer total = 0;
for (CiModule model : modelList.getData().getRows()) {
//获取每个模型下所有的实例总数
String table = CmdbUtils.CMDB_INSTANCE + model.getCode().toLowerCase();
String valueTable = CmdbUtils.CMDB_INSTANCE_VALUE + model.getCode().toLowerCase();
List<Param> paramList = new ArrayList<>();
Param param = new Param();
param.setParam(MapTools.simpleMap("moduleId", model.getId()));
boolean adminRole = instanceService.checkAdminRole(userId);
int count = instanceRepository.count((List<Param>) param, table, valueTable, userId, adminRole);
total+=count;
}
resultMap.put(rootGroup.getCode(), total);
} catch (Exception e) {
log.info("查询失败", e);
}
resultMap.put("","");
countDownLatch.countDown();
}
}).start();