Browse Source

运行记录1219-运行记录日报设备查询优化5

yuanchao 1 week ago
parent
commit
225dcd6cbe

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

@@ -1,6 +1,7 @@
 package cn.iocoder.yudao.module.pms.controller.admin.iotopeationfill;
 
 import cn.hutool.core.bean.BeanUtil;
+import cn.iocoder.yudao.framework.tenant.core.context.TenantContextHolder;
 import cn.iocoder.yudao.framework.tenant.core.util.TenantUtils;
 import cn.iocoder.yudao.module.pms.controller.admin.iotmodeltemplateattrs.vo.IotModelTemplateAttrsRespVO;
 import cn.iocoder.yudao.module.pms.controller.admin.iotopeationfill.vo.*;
@@ -65,6 +66,9 @@ import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success;
 import cn.iocoder.yudao.framework.excel.core.util.ExcelUtils;
 
 import cn.iocoder.yudao.framework.apilog.core.annotation.ApiAccessLog;
+import org.springframework.web.context.request.RequestContextHolder;
+import org.springframework.web.context.request.ServletRequestAttributes;
+
 import static cn.iocoder.yudao.framework.apilog.core.enums.OperateTypeEnum.*;
 
 
@@ -307,7 +311,14 @@ public class IotOpeationFillController {
     @PostMapping("/insertDataList")
     @Operation(summary = "创建运行记录工单设备集填写信息")
     public CommonResult<Integer> insertDataList(@Valid @RequestBody List<IotOperationSaveInfoVO> createReqVO) {
+
+        ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
+        HttpServletRequest request = attributes.getRequest();
+        String tenantId = request.getHeader("Tenant-Id");
+
         try {
+
+            TenantContextHolder.setTenantId(Long.parseLong(tenantId)); // 存入上下文(关键步骤)
             // 1. 数据预处理:提取所有设备数据并扁平化
             List<IotOpeationFillSaveReqVO> allFillData = createReqVO.stream()
                     .flatMap(vo -> vo.getDeviceInfoList().stream())