From 6ae35089a05b58dc31c243b9bb4f6a80b20c3a7b Mon Sep 17 00:00:00 2001
From: xuhy <3313886187@qq.com>
Date: 星期五, 23 五月 2025 17:44:42 +0800
Subject: [PATCH] 修改接口

---
 ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TSamplingRecordController.java |   64 ++++++++++++++++++++++++++------
 1 files changed, 52 insertions(+), 12 deletions(-)

diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TSamplingRecordController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TSamplingRecordController.java
index 26e8cca..5fb30dd 100644
--- a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TSamplingRecordController.java
+++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TSamplingRecordController.java
@@ -22,6 +22,7 @@
 import io.swagger.annotations.ApiOperation;
 import org.springframework.beans.BeanUtils;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.util.CollectionUtils;
 import org.springframework.web.bind.annotation.*;
 
 import java.text.SimpleDateFormat;
@@ -76,6 +77,37 @@
     @PostMapping(value = "/api/t-sampling-record/pageList")
     public R<PageInfo<TSamplingRecordVO>> pageList(@RequestBody String param) {
         TSamplingRecordQuery query = JSON.parseObject(param, TSamplingRecordQuery.class);
+        // 获取当前用户
+        Long userId = tokenService.getLoginUser().getUserId();
+        Integer roleType = tokenService.getLoginUser().getUser().getRoleType();
+        if (roleType != 1){
+            query.setUserId(userId);
+            if(roleType ==2){
+                // 查询项目组
+                TProjectTeamStaff projectTeamStaff = projectTeamStaffService.getOne(Wrappers.lambdaQuery(TProjectTeamStaff.class)
+                        .eq(TProjectTeamStaff::getUserId, userId));
+                // 查询项目的工艺工程师id
+                TProjectTeamStaff teamStaff = projectTeamStaffService.getOne(Wrappers.lambdaQuery(TProjectTeamStaff.class)
+                        .eq(TProjectTeamStaff::getTeamId, projectTeamStaff.getTeamId())
+                        .eq(TProjectTeamStaff::getRoleType, 3)
+                        .last("LIMIT 1"));
+                // 查询实验参与人员
+                List<TExperimentDispatchParticipants> experimentDispatchParticipants = experimentDispatchParticipantsService.list(Wrappers.lambdaQuery(TExperimentDispatchParticipants.class)
+                        .eq(TExperimentDispatchParticipants::getUserId, teamStaff.getUserId()));
+                if(!CollectionUtils.isEmpty(experimentDispatchParticipants)){
+                    List<String> dispatchIds = experimentDispatchParticipants.stream().map(TExperimentDispatchParticipants::getDispatchId).distinct().collect(Collectors.toList());
+                    query.setDispatchIds(dispatchIds);
+                }
+            }else {
+                // 查询实验参与人员
+                List<TExperimentDispatchParticipants> experimentDispatchParticipants = experimentDispatchParticipantsService.list(Wrappers.lambdaQuery(TExperimentDispatchParticipants.class)
+                        .eq(TExperimentDispatchParticipants::getUserId, userId));
+                if (!CollectionUtils.isEmpty(experimentDispatchParticipants)) {
+                    List<String> dispatchIds = experimentDispatchParticipants.stream().map(TExperimentDispatchParticipants::getDispatchId).distinct().collect(Collectors.toList());
+                    query.setDispatchIds(dispatchIds);
+                }
+            }
+        }
         return R.ok(samplingRecordService.pageList(query));
     }
 
@@ -274,13 +306,17 @@
     public R<Boolean> batchSendSamples(@RequestBody String param) {
         BatchSendSamplesDTO batchSendSamplesDTO = JSON.parseObject(param, BatchSendSamplesDTO.class);
         Long userId = tokenService.getLoginUser().getUserId();
+        List<String> ids = batchSendSamplesDTO.getRecordOperationId();
         // 批量送样
-        samplingRecordOperationService.update(Wrappers.lambdaUpdate(TSamplingRecordOperation.class)
-                .eq(TSamplingRecordOperation::getId, batchSendSamplesDTO.getRecordOperationId())
-                .set(TSamplingRecordOperation::getStatus, 2)
-                .set(TSamplingRecordOperation::getSendSign, batchSendSamplesDTO.getSendSign())
-                .set(TSamplingRecordOperation::getSendTime, LocalDateTime.now())
-                .set(TSamplingRecordOperation::getSendPersonId, userId));
+        List<TSamplingRecordOperation> samplingRecordOperations = samplingRecordOperationService.list(Wrappers.lambdaQuery(TSamplingRecordOperation.class)
+                .in(TSamplingRecordOperation::getId, ids));
+        for (TSamplingRecordOperation samplingRecordOperation : samplingRecordOperations) {
+            samplingRecordOperation.setStatus(2);
+            samplingRecordOperation.setSendPersonId(userId);
+            samplingRecordOperation.setSendSign(samplingRecordOperation.getSendSign());
+            samplingRecordOperation.setSendTime(LocalDateTime.now());
+        }
+        samplingRecordOperationService.updateBatchById(samplingRecordOperations);
         return R.ok();
     }
 
@@ -294,13 +330,17 @@
     public R<Boolean> batchCollectSamples(@RequestBody String param) {
         BatchCollectSamplesDTO batchCollectSamplesDTO = JSON.parseObject(param, BatchCollectSamplesDTO.class);
         Long userId = tokenService.getLoginUser().getUserId();
+        List<String> ids = batchCollectSamplesDTO.getRecordOperationId();
         // 批量送样
-        samplingRecordOperationService.update(Wrappers.lambdaUpdate(TSamplingRecordOperation.class)
-                .eq(TSamplingRecordOperation::getId, batchCollectSamplesDTO.getRecordOperationId())
-                .set(TSamplingRecordOperation::getStatus, 3)
-                .set(TSamplingRecordOperation::getReceiptsSign, batchCollectSamplesDTO.getReceiptsSign())
-                .set(TSamplingRecordOperation::getReceiptsTime, LocalDateTime.now())
-                .set(TSamplingRecordOperation::getReceiptsPersonId, userId));
+        List<TSamplingRecordOperation> samplingRecordOperations = samplingRecordOperationService.list(Wrappers.lambdaQuery(TSamplingRecordOperation.class)
+                .in(TSamplingRecordOperation::getId, ids));
+        for (TSamplingRecordOperation samplingRecordOperation : samplingRecordOperations) {
+            samplingRecordOperation.setStatus(3);
+            samplingRecordOperation.setSendPersonId(userId);
+            samplingRecordOperation.setSendSign(samplingRecordOperation.getReceiptsSign());
+            samplingRecordOperation.setSendTime(LocalDateTime.now());
+        }
+        samplingRecordOperationService.updateBatchById(samplingRecordOperations);
         return R.ok();
     }
 

--
Gitblit v1.7.1