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