修改查询模版方法

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
public List<String> getcategory(String type){
List<String> list1 = new ArrayList<>();
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);
if(type==null&&type.equals(""))
{
QueryWrapper<InspectTemplateDO> queryWrapper = new QueryWrapper<>();
queryWrapper.select("DISTINCT category");
List<InspectTemplateDO> doList = templateMapper.selectList(queryWrapper);
for (InspectTemplateDO inspectTemplateDO : doList) {
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;
}
@Override