From 082bddee7df8a05ca95f473ec2643701d41a5213 Mon Sep 17 00:00:00 2001
From: puzhibing <393733352@qq.com>
Date: 星期四, 08 十月 2026 11:09:01 +0800
Subject: [PATCH] 更新代码
---
ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TTrainLibraryController.java | 91 ++++++++++++++++++++++++++-------------------
1 files changed, 53 insertions(+), 38 deletions(-)
diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TTrainLibraryController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TTrainLibraryController.java
index 4234335..310bd35 100644
--- a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TTrainLibraryController.java
+++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TTrainLibraryController.java
@@ -5,6 +5,7 @@
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.ruoyi.common.annotation.Log;
import com.ruoyi.common.basic.PageInfo;
@@ -97,12 +98,14 @@
public R<PageInfo<TProjectTeamVO>> choiceProjectList(@RequestBody String param) {
LoginUser loginUser = tokenService.getLoginUser();
SysUser sysUser = sysUserService.selectUserById(loginUser.getUserId());
- List<String> projectIds = projectTeamStaffService.lambdaQuery().eq(TProjectTeamStaff::getUserId, sysUser.getUserId()).list().stream().map(TProjectTeamStaff::getTeamId).collect(Collectors.toList());
- if (projectIds.isEmpty()){
- projectIds.add("0");
- }
ChoiceProjectDTO query = JSON.parseObject(param, ChoiceProjectDTO.class);
- query.setProjectIds(projectIds);
+ if (!"1".equals(sysUser.getRoleType())) {
+ List<String> projectIds = getAccessibleProjectIds(sysUser.getUserId());
+ if (projectIds.isEmpty()) {
+ projectIds.add("0");
+ }
+ query.setProjectIds(projectIds);
+ }
return R.ok(projectTeamService.choiceProjectList(query));
}
@@ -115,36 +118,43 @@
Long userId = tokenService.getLoginUser().getUserId();
SysUser sysUser = sysUserService.selectUserById(userId);
TTrainLibraryQuery query = JSON.parseObject(param, TTrainLibraryQuery.class);
- if (sysUser.getRoleType().equals("2")){
- List<String> allProjectIds = projectTeamStaffService.lambdaQuery().eq(TProjectTeamStaff::getUserId, tokenService.getLoginUser().getUserId())
- .list().stream().map(TProjectTeamStaff::getTeamId).collect(Collectors.toList());
- if (!allProjectIds.isEmpty()){ List<String> projectIds = projectTeamService.lambdaQuery().in(TProjectTeam::getId, allProjectIds)
- .list().stream().map(TProjectTeam::getId).collect(Collectors.toList());
-
- if (projectIds.isEmpty()){
- return R.ok(new PageInfo<>());
- }
- query.setProjectIds(projectIds);
- }else {
+ if (!sysUser.getRoleType().equals("1")){
+ List<String> accessibleProjectIds = getAccessibleProjectIds(userId);
+ if (accessibleProjectIds.isEmpty()) {
return R.ok(new PageInfo<>());
}
- }else if (!sysUser.getRoleType().equals("1")){
- List<String> allProjectIds = projectTeamStaffService.lambdaQuery().eq(TProjectTeamStaff::getUserId, tokenService.getLoginUser().getUserId())
- .list().stream().map(TProjectTeamStaff::getTeamId).collect(Collectors.toList());
- if (!allProjectIds.isEmpty()){
- List<String> projectIds = projectTeamService.lambdaQuery().in(TProjectTeam::getId, allProjectIds)
- .eq(TProjectTeam::getStatus, 1).list().stream().map(TProjectTeam::getId).collect(Collectors.toList());
-
- if (projectIds.isEmpty()){
+ if (CollectionUtils.isNotEmpty(query.getProjectIds())) {
+ List<String> requestedProjectIds = new ArrayList<>(query.getProjectIds());
+ requestedProjectIds.retainAll(accessibleProjectIds);
+ if (requestedProjectIds.isEmpty()) {
return R.ok(new PageInfo<>());
}
- query.setProjectIds(projectIds);
- }else {
- return R.ok(new PageInfo<>());
+ query.setProjectIds(requestedProjectIds);
+ } else {
+ query.setProjectIds(accessibleProjectIds);
}
}
- return R.ok(tTrainLibraryService.pageList(query));
+ return R.ok(tTrainLibraryService.pageList(query, sysUser.getRoleType()));
}
+
+ private List<String> getAccessibleProjectIds(Long userId) {
+ List<String> teamIds = projectTeamStaffService.lambdaQuery()
+ .eq(TProjectTeamStaff::getUserId, userId)
+ .list().stream()
+ .map(TProjectTeamStaff::getTeamId)
+ .distinct()
+ .collect(Collectors.toList());
+ if (teamIds.isEmpty()) {
+ return new ArrayList<>();
+ }
+ return projectTeamService.lambdaQuery()
+ .in(TProjectTeam::getId, teamIds)
+ .eq(TProjectTeam::getStatus, 1)
+ .list().stream()
+ .map(TProjectTeam::getId)
+ .collect(Collectors.toList());
+ }
+
/**
* 获取菌种库出入库时间轴列表
*/
@@ -185,15 +195,17 @@
return R.fail(401,"登陆失效");
}
TTrainLibraryDTO dto = JSON.parseObject(param,TTrainLibraryDTO.class);
- List<String> projectIds = projectTeamStaffService.lambdaQuery().eq(TProjectTeamStaff::getUserId, tokenService.getLoginUser().getUserId())
- .list().stream().map(TProjectTeamStaff::getTeamId).collect(Collectors.toList());
- if (!projectIds.isEmpty()){
- TProjectTeam projectTeam = projectTeamService.lambdaQuery().in(TProjectTeam::getId, projectIds)
- .eq(TProjectTeam::getStatus, 1)
- .orderByDesc(BaseModel::getCreateTime)
- .last("limit 1").one();
- dto.setProjectId(projectTeam.getId());
- }
+
+ //改为前端传值
+// List<String> projectIds = projectTeamStaffService.lambdaQuery().eq(TProjectTeamStaff::getUserId, tokenService.getLoginUser().getUserId())
+// .list().stream().map(TProjectTeamStaff::getTeamId).collect(Collectors.toList());
+// if (!projectIds.isEmpty()){
+// TProjectTeam projectTeam = projectTeamService.lambdaQuery().in(TProjectTeam::getId, projectIds)
+// .eq(TProjectTeam::getStatus, 1)
+// .orderByDesc(BaseModel::getCreateTime)
+// .last("limit 1").one();
+// dto.setProjectId(projectTeam.getId());
+// }
tTrainLibraryService.save(dto);
if (dto.getIsDraft()==0){
// 新增入库记录
@@ -343,7 +355,7 @@
}
TTrainLibraryDTO dto = JSON.parseObject(param,TTrainLibraryDTO.class);
tTrainLibraryService.updateById(dto);
- if (dto.getIsDraft()==0){
+ if (dto.getIsDraft()==0) {
List<TTrainLibraryWarehousing> list = tTrainLibraryWarehousingService.lambdaQuery().eq(TTrainLibraryWarehousing::getTrainLibraryId, dto.getId()).list();
if (list.isEmpty()){
// 新增入库记录
@@ -395,6 +407,9 @@
@GetMapping(value = "/open/t-train-library/getDetailEditById")
public R<TTrainLibraryDetailVO> getDetailEditById(@RequestParam String id) {
TTrainLibrary trainLibrary = tTrainLibraryService.getById(id);
+ if (trainLibrary == null) {
+ return R.fail("菌种库记录不存在或已删除");
+ }
TTrainLibraryDetailVO tTrainLibraryDetailVO = new TTrainLibraryDetailVO();
BeanUtils.copyProperties(trainLibrary,tTrainLibraryDetailVO);
return R.ok(tTrainLibraryDetailVO);
--
Gitblit v1.7.1