pms 瑞恒 瑞鹰 运行记录 添加 井号

yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/controller/admin/iotopeationfill/IotOpeationFillController.java

@@ -1,17 +1,20 @@
 package cn.iocoder.yudao.module.pms.controller.admin.iotopeationfill;
 
 import cn.hutool.core.bean.BeanUtil;
+import cn.hutool.core.collection.CollUtil;
 import cn.iocoder.yudao.framework.apilog.core.annotation.ApiAccessLog;
 import cn.iocoder.yudao.framework.common.pojo.CommonResult;
 import cn.iocoder.yudao.framework.common.pojo.PageParam;
 import cn.iocoder.yudao.framework.common.pojo.PageResult;
 import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
 import cn.iocoder.yudao.framework.excel.core.util.ExcelUtils;
+import cn.iocoder.yudao.framework.tenant.core.aop.TenantIgnore;
 import cn.iocoder.yudao.framework.tenant.core.context.TenantContextHolder;
 import cn.iocoder.yudao.framework.tenant.core.util.TenantUtils;
 import cn.iocoder.yudao.module.pms.annotation.AntiDuplicateSubmit;
 import cn.iocoder.yudao.module.pms.controller.admin.iotmodeltemplateattrs.vo.IotModelTemplateAttrsRespVO;
 import cn.iocoder.yudao.module.pms.controller.admin.iotopeationfill.vo.*;
+import cn.iocoder.yudao.module.pms.controller.admin.iotprojecttask.vo.IotProjectTaskPageReqVO;
 import cn.iocoder.yudao.module.pms.controller.admin.iotrhdailyreport.vo.IotRhDailyReportSaveReqVO;
 import cn.iocoder.yudao.module.pms.controller.admin.iotrydailyreport.vo.IotRyDailyReportSaveReqVO;
 import cn.iocoder.yudao.module.pms.controller.admin.vo.DeviceTdVO;
@@ -25,7 +28,9 @@ import cn.iocoder.yudao.module.pms.dal.dataobject.iotmodeltemplateattrs.IotModel
 import cn.iocoder.yudao.module.pms.dal.dataobject.iotmodeltemplateattrs.IotModelTemplateAttrsDO1;
 import cn.iocoder.yudao.module.pms.dal.dataobject.iotopeationfill.IotOpeationFillDO;
 import cn.iocoder.yudao.module.pms.dal.dataobject.iotopeationfill.IotOpeationFillOrderDO;
+import cn.iocoder.yudao.module.pms.dal.dataobject.iotprojecttask.IotProjectTaskDO;
 import cn.iocoder.yudao.module.pms.dal.dataobject.yanfan.YfDeviceDO;
+import cn.iocoder.yudao.module.pms.dal.mysql.iotprojecttask.IotProjectTaskMapper;
 import cn.iocoder.yudao.module.pms.service.IDeviceService;
 import cn.iocoder.yudao.module.pms.service.iotopeationfill.IotOpeationFillService;
 import cn.iocoder.yudao.module.pms.service.iotrhdailyreport.IotRhDailyReportService;
@@ -86,6 +91,8 @@ public class IotOpeationFillController {
     private IotRhDailyReportService iotRhDailyReportService;
     @Resource
     private IotRyDailyReportService iotRyDailyReportService;
+    @Resource
+    private IotProjectTaskMapper iotProjectTaskMapper;
 
     @PostMapping("/create")
     @Operation(summary = "创建运行记录填报")
@@ -1015,6 +1022,25 @@ public class IotOpeationFillController {
     @PreAuthorize("@ss.hasPermission('rq:iot-opeation-fill:query')")
     public CommonResult<List<IotOpeationFillDO>> getIotOpeationFillPage(@Valid IotOpeationFillRespVO pageReqVO) {
         List<IotOpeationFillDO> fillList = iotOpeationFillService.fillListByUserId(pageReqVO);
+        // 查询 生产日报 当前队伍对应的 任务井 deviceName = '生产日报'
+
+        if (CollUtil.isNotEmpty(fillList)) {
+            TenantUtils.executeIgnore(() -> {
+                fillList.forEach(fill -> {
+                    if ("生产日报".equals(fill.getDeviceName())) {
+                        fill.getDeptId();
+                        IotProjectTaskPageReqVO reqVO = new IotProjectTaskPageReqVO();
+                        reqVO.setDeptId(fill.getDeptId());
+                        reqVO.setJobFlag("Y");
+                        List<IotProjectTaskDO> tasks = iotProjectTaskMapper.selectList(reqVO);
+                        if (CollUtil.isNotEmpty(tasks)) {
+                            // 取最新的1个任务
+                            fill.setWellName(tasks.get(0).getWellName());
+                        }
+                    }
+                });
+            });
+        }
         return success(BeanUtils.toBean(fillList, IotOpeationFillDO.class));
     }
 

yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/dal/dataobject/iotopeationfill/IotOpeationFillDO.java

@@ -114,4 +114,8 @@ public class IotOpeationFillDO extends BaseDO {
     private int fillDev;
     private int unFillDev;
 
+    /**
+     * 扩展字段
+     */
+    private String wellName;
 }