feat(dry): 添加租户设备查询功能并优化代码

bsw215583320

2024-11-22 bdfce3674b99e3bb494f1a8a479834f842968ef2

 jeecg-module-dry/jeecg-module-dry-biz/src/main/java/org/jeecg/modules/dry/controller/DryEquipmentController.java

@@ -36,6 +36,7 @@
import org.jeecgframework.poi.excel.entity.ImportParams;
import org.jeecgframework.poi.excel.view.JeecgEntityExcelView;
import org.jeecg.common.system.base.controller.JeecgController;
import org.jetbrains.annotations.NotNull;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
@@ -106,20 +107,38 @@
      //------------------------------------------------------------------------------------------------
      QueryWrapper<DryEquipment> queryWrapper = QueryGenerator.initQueryWrapper(dryEquipment, req.getParameterMap());

      queryWrapper.orderByAsc("code");
      List<DryEquipment> eqps = dryEquipmentService.list(queryWrapper);
      List<DryEquipmentVo> equipmentVos = new ArrayList<>();
      eqps.stream().forEach(item -> {
         DryEquipmentVo dryEquipmentVo = BeanUtil.toBean(item, DryEquipmentVo.class);
         DryEqpType dryEqpType = dryEqpTypeService.getById(item.getType());
         dryEquipmentVo.setType(dryEqpType.getName());
         dryEquipmentVo.setDryEfficiency(dryEqpType.getDryEfficiency());
         dryEquipmentVo.setSteamConsumption(dryEqpType.getSteamConsumption());
         dryEquipmentVo.setPowerConsumption(dryEqpType.getPowerConsumption());
         equipmentVos.add(dryEquipmentVo);
      });
      return Result.OK(equipmentVos);
       List<DryEquipmentVo> equipmentVos = getDryEquipmentVos(queryWrapper);
       return Result.OK(equipmentVos);
   }

    @NotNull
    private List<DryEquipmentVo> getDryEquipmentVos(QueryWrapper<DryEquipment> queryWrapper) {
       queryWrapper.orderByAsc("code");
       List<DryEquipment> eqps = dryEquipmentService.list(queryWrapper);
       List<DryEquipmentVo> equipmentVos = new ArrayList<>();
       eqps.stream().forEach(item -> {
          DryEquipmentVo dryEquipmentVo = BeanUtil.toBean(item, DryEquipmentVo.class);
          DryEqpType dryEqpType = dryEqpTypeService.getById(item.getType());
          dryEquipmentVo.setType(dryEqpType.getName());
          dryEquipmentVo.setDryEfficiency(dryEqpType.getDryEfficiency());
          dryEquipmentVo.setSteamConsumption(dryEqpType.getSteamConsumption());
          dryEquipmentVo.setPowerConsumption(dryEqpType.getPowerConsumption());
          equipmentVos.add(dryEquipmentVo);
       });
       return equipmentVos;
    }


    @ApiOperation(value="查询租户所有干燥机", notes="查询租户所有干燥机")
    @GetMapping(value = "/listTenantEqp")
    public Result<List<DryEquipmentVo>> queryTenantAllEquipment(DryEquipment dryEquipment, HttpServletRequest req) {
      TenantContext.setTenant(dryEquipment.getTenantId()+"");

       QueryWrapper<DryEquipment> queryWrapper = QueryGenerator.initQueryWrapper(dryEquipment, req.getParameterMap());

       List<DryEquipmentVo> equipmentVos = getDryEquipmentVos(queryWrapper);
       return Result.OK(equipmentVos);
    }



@@ -151,6 +170,7 @@
   @RequiresPermissions("dry:dry_equipment:edit")
   @RequestMapping(value = "/edit", method = {RequestMethod.PUT,RequestMethod.POST})
   public Result<String> edit(@RequestBody DryEquipment dryEquipment) {
      // 删除redis缓存
      redisUtil.del(CacheConstants.RedisKeyEnum.EQP_MAP.getCode());
      dryEquipmentService.updateById(dryEquipment);