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 | 107 ++++++++++++++++++++++++++++++++++-------------------
1 files changed, 69 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 cb597d4..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
@@ -2,7 +2,10 @@
import com.alibaba.fastjson.JSON;
+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;
@@ -95,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));
}
@@ -113,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());
+ }
+
/**
* 获取菌种库出入库时间轴列表
*/
@@ -183,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){
// 新增入库记录
@@ -236,6 +250,20 @@
noticeService.save(tNotice);
}
return R.ok();
+ }
+
+ public static void main(String[] args) {
+// String jsonStr = "[{\"id\":1,\"num1\":\"00:00\",\"num2\":\"19:00\",\"num3\":1,\"num4\":1,\"num5\":1,\"num6\":1,\"num7\":1}]";
+// JSONArray startingFareJsonArray = JSONArray.parseArray(jsonStr);
+// for (int i = 0; i < startingFareJsonArray.size(); i++) {
+// JSONObject jsonObject = startingFareJsonArray.getJSONObject(i);
+// System.out.println(jsonObject.getString("num1"));
+// System.out.println(jsonObject.getString("num2"));
+// System.out.println(jsonObject.getInteger("num3"));
+// }
+ Long temp =2027571910897541122L;
+ System.out.println(temp%20);
+
}
/**
* 新增菌种库
@@ -327,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()){
// 新增入库记录
@@ -379,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