修改查询模版方法

This commit is contained in:
lxd 2025-03-28 20:55:57 +08:00
parent bd1d340de0
commit f415fbc559

View File

@ -67,15 +67,31 @@ public class InspectTemplateServiceImpl implements InspectTemplateService {
@Override @Override
public List<String> getcategory(String type){ public List<String> getcategory(String type){
List<String> list1 = new ArrayList<>(); List<String> list1 = new ArrayList<>();
QueryWrapper<InspectTemplateDO> queryWrapper = new QueryWrapper<>(); if(type==null&&type.equals(""))
queryWrapper.eq("type", type); {
List<InspectTemplateDO> doList=templateMapper.selectList(queryWrapper); QueryWrapper<InspectTemplateDO> queryWrapper = new QueryWrapper<>();
for (InspectTemplateDO inspectTemplateDO : doList) { queryWrapper.select("DISTINCT category");
String category = inspectTemplateDO.getCategory(); List<InspectTemplateDO> doList = templateMapper.selectList(queryWrapper);
if (!list1.contains(category)) { // 判断是否已经存在 for (InspectTemplateDO inspectTemplateDO : doList) {
list1.add(category); if(inspectTemplateDO!=null)
{
list1.add(inspectTemplateDO.getCategory());
}
} }
} }
else
{
QueryWrapper<InspectTemplateDO> queryWrapper = new QueryWrapper<>();
queryWrapper.eq("type", type);
List<InspectTemplateDO> doList=templateMapper.selectList(queryWrapper);
for (InspectTemplateDO inspectTemplateDO : doList) {
String category = inspectTemplateDO.getCategory();
if (!list1.contains(category)) { // 判断是否已经存在
list1.add(category);
}
}
}
return list1; return list1;
} }
@Override @Override