From b5b400a615743a74e9d127261bd3785554aa06aa Mon Sep 17 00:00:00 2001
From: zhou zhou <3272660260@qq.com>
Date: 星期六, 10 一月 2026 13:34:40 +0800
Subject: [PATCH] #
---
rsf-server/src/main/java/com/vincent/rsf/server/system/controller/WarehouseRoleMenuController.java | 113 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 113 insertions(+), 0 deletions(-)
diff --git a/rsf-server/src/main/java/com/vincent/rsf/server/system/controller/WarehouseRoleMenuController.java b/rsf-server/src/main/java/com/vincent/rsf/server/system/controller/WarehouseRoleMenuController.java
new file mode 100644
index 0000000..7c548f0
--- /dev/null
+++ b/rsf-server/src/main/java/com/vincent/rsf/server/system/controller/WarehouseRoleMenuController.java
@@ -0,0 +1,113 @@
+package com.vincent.rsf.server.system.controller;
+
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+import com.vincent.rsf.common.utils.Utils;
+import com.vincent.rsf.framework.common.Cools;
+import com.vincent.rsf.framework.common.R;
+import com.vincent.rsf.framework.exception.CoolException;
+import com.vincent.rsf.server.common.utils.ExcelUtil;
+import com.vincent.rsf.server.common.annotation.OperationLog;
+import com.vincent.rsf.server.common.domain.BaseParam;
+import com.vincent.rsf.server.common.domain.KeyValVo;
+import com.vincent.rsf.server.common.domain.PageParam;
+import com.vincent.rsf.server.manager.entity.MatnrGroup;
+import com.vincent.rsf.server.manager.entity.Warehouse;
+import com.vincent.rsf.server.manager.entity.WarehouseAreas;
+import com.vincent.rsf.server.manager.service.WarehouseAreasService;
+import com.vincent.rsf.server.manager.service.WarehouseService;
+import com.vincent.rsf.server.system.controller.param.RoleScopeParam;
+import com.vincent.rsf.server.system.entity.PdaRoleMenu;
+import com.vincent.rsf.server.system.entity.WarehouseRoleMenu;
+import com.vincent.rsf.server.system.service.WarehouseRoleMenuService;
+import com.vincent.rsf.server.system.controller.BaseController;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.security.access.prepost.PreAuthorize;
+import org.springframework.transaction.annotation.Transactional;
+import org.springframework.web.bind.annotation.*;
+
+import javax.servlet.http.HttpServletResponse;
+import java.util.*;
+
+@RestController
+public class WarehouseRoleMenuController extends BaseController {
+
+ @Autowired
+ private WarehouseRoleMenuService warehouseRoleMenuService;
+
+ @Autowired
+ private WarehouseAreasService warehouseAreasService;
+
+ @Autowired
+ private WarehouseService warehouseService;
+
+ @GetMapping("/roleWarehouse/scope/list")
+ public R scopeList(@RequestParam Long roleId) {
+ return R.ok().add(warehouseRoleMenuService.listStrictlyMenuByRoleId(roleId));
+ }
+
+ @PreAuthorize("hasAuthority('system:role:update')")
+ @OperationLog("Assign Permissions")
+ @PostMapping("/roleWarehouse/scope/update")
+ @Transactional
+ public R scopeUpdate(@RequestBody RoleScopeParam param) {
+ Long roleId = param.getId();
+ List<Long> menuIds = new ArrayList<>(param.getMenuIds().getChecked());
+ menuIds.addAll(param.getMenuIds().getHalfChecked());
+ warehouseRoleMenuService
+ .remove(new LambdaQueryWrapper<WarehouseRoleMenu>().eq(WarehouseRoleMenu::getRoleId, roleId));
+ for (Long menuId : menuIds) {
+ if (!warehouseRoleMenuService.save(new WarehouseRoleMenu(roleId, menuId))) {
+ throw new CoolException("Internal Server Error!");
+ }
+ }
+ return R.ok("Assign Success");
+ }
+
+ @PostMapping("/menuWarehouse/tree")
+ public R tree(@RequestBody Map<String, Object> map) {
+ // 鏌ヨ鎵�鏈変粨搴�
+ List<Warehouse> warehouseList = warehouseService
+ .list(new LambdaQueryWrapper<Warehouse>().orderByAsc(Warehouse::getId));
+ // 鏌ヨ鎵�鏈夊簱鍖�
+ List<WarehouseAreas> areasList = warehouseAreasService
+ .list(new LambdaQueryWrapper<WarehouseAreas>().orderByAsc(WarehouseAreas::getId));
+
+ // 鎸変粨搴揑D鍒嗙粍搴撳尯
+ Map<Long, List<WarehouseAreas>> areasMap = areasList.stream()
+ .collect(java.util.stream.Collectors.groupingBy(WarehouseAreas::getWarehouseId));
+
+ // 鏋勫缓鏍戝舰缁撴瀯锛氬皢搴撳尯璁剧疆涓轰粨搴撶殑children
+ for (Warehouse warehouse : warehouseList) {
+ List<WarehouseAreas> children = areasMap.getOrDefault(warehouse.getId(), new ArrayList<>());
+ // 浠撳簱浣跨敤100000+id锛岄伩鍏嶄笌搴撳尯id閲嶅
+ warehouse.setId(100000L + warehouse.getId());
+ warehouse.setFlagWare(1); // 1琛ㄧず浠撳簱
+ // 搴撳尯淇濇寔鍘焛d涓嶅彉
+ for (WarehouseAreas area : children) {
+ area.setFlagWare(0); // 0琛ㄧず搴撳尯
+ }
+ warehouse.setChildren(children);
+ }
+
+ // 鏉′欢杩囨护
+ if (!Cools.isEmpty(map.get("condition"))) {
+ String condition = String.valueOf(map.get("condition"));
+ // 杩囨护浠撳簱鍚嶇О鍜屽簱鍖哄悕绉�
+ warehouseList.removeIf(warehouse -> {
+ // 鍏堣繃婊ゅ簱鍖�
+ if (warehouse.getChildren() != null) {
+ warehouse.getChildren()
+ .removeIf(area -> area.getName() != null && !area.getName().contains(condition));
+ }
+ // 濡傛灉浠撳簱鍚嶇О涓嶅尮閰嶄笖娌℃湁鍖归厤鐨勫簱鍖猴紝鍒欑Щ闄よ浠撳簱
+ boolean warehouseMatch = warehouse.getName() != null && warehouse.getName().contains(condition);
+ boolean hasMatchingChildren = warehouse.getChildren() != null && !warehouse.getChildren().isEmpty();
+ return !warehouseMatch && !hasMatchingChildren;
+ });
+ }
+
+ return R.ok().add(warehouseList);
+ }
+
+}
--
Gitblit v1.9.1