liyudong/simulation-backend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

医院,仓库查询代码

Browse Source
This commit is contained in:
wangwenhua 2025-09-15 00:35:19 +08:00
parent 36b0878659
commit afd861e39d
3 changed files with 35 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/main/java/com/hivekion/scenario/service/IResourceService.java
View File

@ -5,6 +5,7 @@ import com.baomidou.mybatisplus.extension.service.IService;
import com.hivekion.scenario.entity.Vehicle;
import java.util.List;
import java.util.Map;
/**
* <p>
@ -18,6 +19,8 @@ public interface IResourceService extends IService<Resource> {
public List<Resource> listAllBuildResourceByType(Integer type);
public Map<Integer,Resource> listBuildResourceByType(Integer type);
public List<Vehicle> listAllVehiclesByType(Integer type,Integer resourceId);
}

10
src/main/java/com/hivekion/scenario/service/impl/ResourceServiceImpl.java
View File

@ -15,6 +15,7 @@ import org.springframework.util.CollectionUtils;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
/**
@ -43,6 +44,15 @@ public class ResourceServiceImpl extends ServiceImpl<ResourceMapper, Resource> i
return this.resourceMapper.selectList(queryWrapper);
}
@Override
public Map<Integer, Resource> listBuildResourceByType(Integer type) {
QueryWrapper<Resource> queryWrapper = new QueryWrapper<>();
queryWrapper.eq(type != null,"resource_type",type);
List<Resource> resources = this.resourceMapper.selectList(queryWrapper);
Map<Integer,Resource> resourceMap = resources.stream().collect(Collectors.toMap(Resource::getId,resource -> resource));
return resourceMap;
}
@Override
public List<Vehicle> listAllVehiclesByType(Integer type, Integer resourceId) {
List<ResourceVehicleRela> rVRela = this.resourceVehicleRelaService.list(new QueryWrapper<ResourceVehicleRela>().eq("resource_id",resourceId));

22
src/main/java/com/hivekion/scenario/service/impl/ScenarioResourceServiceImpl.java
View File

@ -7,6 +7,7 @@ import com.hivekion.baseData.service.ITblEntityService;
import com.hivekion.icon.service.IconService;
import com.hivekion.scenario.entity.ScenarioResource;
import com.hivekion.scenario.mapper.ScenarioResourceMapper;
import com.hivekion.scenario.service.IResourceService;
import com.hivekion.scenario.service.ScenarioResourceService;
import com.hivekion.team.entity.Teaminfo;
import com.hivekion.team.service.ITeaminfoService;
@ -35,6 +36,9 @@ public class ScenarioResourceServiceImpl extends
@Resource
private ITeaminfoService teamInfoService;
@Resource
private IResourceService resourceService;
@Override
public List<ScenarioResource> getResourceList(ScenarioResource resource) {
//图标Map
@ -43,6 +47,8 @@ public class ScenarioResourceServiceImpl extends
Map<Integer, TblEntity> entityMap = iTblEntityService.entityMap();
//分队Map
Map<Integer, Teaminfo> teamInfoMap = teamInfoService.teamInfoMap();
Map<Integer,com.hivekion.scenario.entity.Resource> hResourceMap = resourceService.listBuildResourceByType(7);
Map<Integer,com.hivekion.scenario.entity.Resource> wResourceMap = resourceService.listBuildResourceByType(8);
QueryWrapper<ScenarioResource> queryWrapper = new QueryWrapper<>();
queryWrapper.eq("scenario_id", resource.getScenarioId());
if (resource.getType() != null) {
@ -72,6 +78,22 @@ public class ScenarioResourceServiceImpl extends
iconMap.get(teaminfo.getIconId()) == null ? "" : iconMap.get(teaminfo.getIconId()));
}
break;
case 7:
if(hResourceMap.get(scenarioResource.getResourceId()) != null){
com.hivekion.scenario.entity.Resource resource1 = hResourceMap.get(scenarioResource.getResourceId());
scenarioResource.setTitle(resource1.getResourceName());
scenarioResource.setImgBase64(
iconMap.get(resource1.getIcon()) == null ? "" : iconMap.get(resource1.getIcon()));
}
break;
case 8:
if(wResourceMap.get(scenarioResource.getResourceId()) != null){
com.hivekion.scenario.entity.Resource resource1 = wResourceMap.get(scenarioResource.getResourceId());
scenarioResource.setTitle(resource1.getResourceName());
scenarioResource.setImgBase64(
iconMap.get(resource1.getIcon()) == null ? "" : iconMap.get(resource1.getIcon()));
}
break;
}
}
return scenarioResourceList;