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));
|
|
// 按仓库ID分组库区
|
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表示仓库
|
// 库区保持原id不变
|
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);
|
}
|
|
}
|
