package com.stylefeng.guns.modular.system.controller.system;
|
|
import com.baomidou.mybatisplus.mapper.EntityWrapper;
|
import com.stylefeng.guns.core.base.controller.BaseController;
|
import com.stylefeng.guns.core.base.tips.Tip;
|
import com.stylefeng.guns.core.common.annotion.BussinessLog;
|
import com.stylefeng.guns.core.common.constant.dictmap.MenuDict;
|
import com.stylefeng.guns.core.common.constant.factory.ConstantFactory;
|
import com.stylefeng.guns.core.common.constant.state.MenuStatus;
|
import com.stylefeng.guns.core.common.exception.BizExceptionEnum;
|
import com.stylefeng.guns.core.exception.GunsException;
|
import com.stylefeng.guns.core.log.LogObjectHolder;
|
import com.stylefeng.guns.core.node.ZTreeNode;
|
import com.stylefeng.guns.core.shiro.ShiroKit;
|
import com.stylefeng.guns.core.shiro.ShiroUser;
|
import com.stylefeng.guns.core.support.BeanKit;
|
import com.stylefeng.guns.core.util.ToolUtil;
|
import com.stylefeng.guns.modular.system.dao.TCompanyMapper;
|
import com.stylefeng.guns.modular.system.model.Menu;
|
import com.stylefeng.guns.modular.system.model.Relation;
|
import com.stylefeng.guns.modular.system.model.TCompany;
|
import com.stylefeng.guns.modular.system.service.IMenuService;
|
import com.stylefeng.guns.modular.system.service.IRelationService;
|
import com.stylefeng.guns.modular.system.warpper.MenuWarpper;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Controller;
|
import org.springframework.ui.Model;
|
import org.springframework.validation.BindingResult;
|
import org.springframework.web.bind.annotation.PathVariable;
|
import org.springframework.web.bind.annotation.RequestMapping;
|
import org.springframework.web.bind.annotation.RequestParam;
|
import org.springframework.web.bind.annotation.ResponseBody;
|
|
import javax.annotation.Resource;
|
import javax.validation.Valid;
|
import java.util.ArrayList;
|
import java.util.Arrays;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.stream.Collectors;
|
|
/**
|
* 菜单控制器
|
*
|
* @author fengshuonan
|
* @Date 2017年2月12日21:59:14
|
*/
|
@Controller
|
@RequestMapping("/menu")
|
public class MenuController extends BaseController {
|
|
private static String PREFIX = "/system/menu/";
|
|
// 快车相关菜单（isSpe=2时过滤）
|
private static final List<String> SPE_MENUS = Arrays.asList(
|
"快车订单", "快车改派管理", "快车车型设置"
|
);
|
|
// 出租车相关菜单（isTaxi=2时过滤）
|
private static final List<String> TAXI_MENUS = Arrays.asList(
|
"出租车订单", "出租车改派管理"
|
);
|
|
// 跨城相关菜单（isCross=2时过滤）
|
private static final List<String> CROSS_MENUS = Arrays.asList(
|
"跨城出行订单", "跨城改派管理", "跨城线路管理", "开通城市管理"
|
);
|
|
// 顺风车相关菜单（isHitchRide=2时过滤）
|
private static final List<String> HITCH_RIDE_MENUS = Arrays.asList(
|
"顺风车订单", "顺风车设置", "顺风车司机审核列表",
|
"顺风车司机", "顺风车投诉", "顺风车评价"
|
);
|
|
@Autowired
|
private IMenuService menuService;
|
|
@Resource
|
private TCompanyMapper companyMapper;
|
|
@Resource
|
private IRelationService rerelationService;
|
|
|
|
/**
|
* 跳转到菜单列表列表页面
|
*/
|
@RequestMapping("")
|
public String index() {
|
return PREFIX + "menu.html";
|
}
|
|
/**
|
* 跳转到菜单列表列表页面
|
*/
|
@RequestMapping(value = "/menu_add")
|
public String menuAdd() {
|
return PREFIX + "menu_add.html";
|
}
|
|
/**
|
* 跳转到菜单详情列表页面
|
*/
|
@RequestMapping(value = "/menu_edit/{menuId}")
|
public String menuEdit(@PathVariable Long menuId, Model model) {
|
if (ToolUtil.isEmpty(menuId)) {
|
throw new GunsException(BizExceptionEnum.REQUEST_NULL);
|
}
|
Menu menu = this.menuService.selectById(menuId);
|
|
//获取父级菜单的id
|
Menu temp = new Menu();
|
temp.setCode(menu.getPcode());
|
Menu pMenu = this.menuService.selectOne(new EntityWrapper<>(temp));
|
|
//如果父级是顶级菜单
|
if (pMenu == null) {
|
menu.setPcode("0");
|
} else {
|
//设置父级菜单的code为父级菜单的id
|
menu.setPcode(String.valueOf(pMenu.getId()));
|
}
|
|
Map<String, Object> menuMap = BeanKit.beanToMap(menu);
|
menuMap.put("pcodeName", ConstantFactory.me().getMenuNameByCode(temp.getCode()));
|
model.addAttribute("menu", menuMap);
|
LogObjectHolder.me().set(menu);
|
return PREFIX + "menu_edit.html";
|
}
|
|
/**
|
* 修该菜单
|
*/
|
@RequestMapping(value = "/edit")
|
@BussinessLog(value = "修改菜单", key = "name", dict = MenuDict.class)
|
@ResponseBody
|
public Tip edit(@Valid Menu menu, BindingResult result) {
|
if (result.hasErrors()) {
|
throw new GunsException(BizExceptionEnum.REQUEST_NULL);
|
}
|
//设置父级菜单编号
|
menuSetPcode(menu);
|
|
this.menuService.updateById(menu);
|
|
|
return SUCCESS_TIP;
|
}
|
|
/**
|
* 获取菜单列表
|
*/
|
@RequestMapping(value = "/list")
|
@ResponseBody
|
public Object list(@RequestParam(required = false) String menuName, @RequestParam(required = false) String level) {
|
List<Map<String, Object>> menus = this.menuService.selectMenus(menuName, level);
|
return super.warpObject(new MenuWarpper(menus));
|
}
|
|
/**
|
* 新增菜单
|
*/
|
@RequestMapping(value = "/add")
|
@BussinessLog(value = "菜单新增", key = "name", dict = MenuDict.class)
|
@ResponseBody
|
public Tip add(@Valid Menu menu, BindingResult result) {
|
if (result.hasErrors()) {
|
throw new GunsException(BizExceptionEnum.REQUEST_NULL);
|
}
|
|
//判断是否存在该编号
|
String existedMenuName = ConstantFactory.me().getMenuNameByCode(menu.getCode());
|
if (ToolUtil.isNotEmpty(existedMenuName)) {
|
throw new GunsException(BizExceptionEnum.EXISTED_THE_MENU);
|
}
|
|
//设置父级菜单编号
|
menuSetPcode(menu);
|
|
menu.setStatus(MenuStatus.ENABLE.getCode());
|
this.menuService.insert(menu);
|
|
|
return SUCCESS_TIP;
|
}
|
|
/**
|
* 删除菜单
|
*/
|
@RequestMapping(value = "/remove")
|
@BussinessLog(value = "删除菜单", key = "menuId", dict = MenuDict.class)
|
@ResponseBody
|
public Tip remove(@RequestParam Long menuId) {
|
if (ToolUtil.isEmpty(menuId)) {
|
throw new GunsException(BizExceptionEnum.REQUEST_NULL);
|
}
|
|
|
//缓存菜单的名称
|
LogObjectHolder.me().set(ConstantFactory.me().getMenuName(menuId));
|
|
this.menuService.delMenuContainSubMenus(menuId);
|
|
return SUCCESS_TIP;
|
}
|
|
/**
|
* 查看菜单
|
*/
|
@RequestMapping(value = "/view/{menuId}")
|
@ResponseBody
|
public Tip view(@PathVariable Long menuId) {
|
if (ToolUtil.isEmpty(menuId)) {
|
throw new GunsException(BizExceptionEnum.REQUEST_NULL);
|
}
|
this.menuService.selectById(menuId);
|
return SUCCESS_TIP;
|
}
|
|
/**
|
* 获取菜单列表(首页用)
|
*/
|
@RequestMapping(value = "/menuTreeList")
|
@ResponseBody
|
public List<ZTreeNode> menuTreeList() {
|
ShiroUser user1 = ShiroKit.getUser();
|
TCompany tCompany = companyMapper.selectById(user1.getObjectId());
|
List<Long> ids = new ArrayList<>();
|
if(1 != tCompany.getType()){
|
List<Relation> relations = rerelationService.selectList(new EntityWrapper<Relation>().eq("roleid", tCompany.getType()));
|
if(relations.size() == 0){
|
ids.add(0L);
|
}else{
|
ids = relations.stream().map(Relation::getMenuid).collect(Collectors.toList());
|
}
|
}
|
List<ZTreeNode> roleTreeList = this.menuService.menuTreeList(ids);
|
// 分公司(type=2)和供应商(type=3)需要根据经营范围过滤菜单
|
if (tCompany.getType() == 2 || tCompany.getType() == 3) {
|
roleTreeList = filterMenusByBusinessScope(roleTreeList, tCompany);
|
}
|
return roleTreeList;
|
}
|
|
/**
|
* 获取菜单列表(选择父级菜单用)
|
*/
|
@RequestMapping(value = "/selectMenuTreeList")
|
@ResponseBody
|
public List<ZTreeNode> selectMenuTreeList() {
|
ShiroUser user1 = ShiroKit.getUser();
|
TCompany tCompany = companyMapper.selectById(user1.getObjectId());
|
List<Long> ids = new ArrayList<>();
|
if(1 != tCompany.getType()){
|
List<Relation> relations = rerelationService.selectList(new EntityWrapper<Relation>().eq("roleid", tCompany.getType()));
|
if(relations.size() == 0){
|
ids.add(0L);
|
}else{
|
ids = relations.stream().map(Relation::getMenuid).collect(Collectors.toList());
|
}
|
}
|
List<ZTreeNode> roleTreeList = this.menuService.menuTreeList(ids);
|
// 分公司(type=2)和供应商(type=3)需要根据经营范围过滤菜单
|
if (tCompany.getType() == 2 || tCompany.getType() == 3) {
|
roleTreeList = filterMenusByBusinessScope(roleTreeList, tCompany);
|
}
|
roleTreeList.add(ZTreeNode.createParent());
|
return roleTreeList;
|
}
|
|
/**
|
* 获取角色列表
|
*/
|
@RequestMapping(value = "/menuTreeListByRoleId/{roleId}")
|
@ResponseBody
|
public List<ZTreeNode> menuTreeListByRoleId(@PathVariable Integer roleId) {
|
List<Long> menuIds = this.menuService.getMenuIdsByRoleId(roleId);
|
ShiroUser user1 = ShiroKit.getUser();
|
TCompany tCompany = companyMapper.selectById(user1.getObjectId());
|
List<Long> ids = new ArrayList<>();
|
if(1 != tCompany.getType()){
|
List<Relation> relations = rerelationService.selectList(new EntityWrapper<Relation>().eq("roleid", tCompany.getType()));
|
if(relations.size() == 0){
|
ids.add(0L);
|
}else{
|
ids = relations.stream().map(Relation::getMenuid).collect(Collectors.toList());
|
}
|
}
|
if (ToolUtil.isEmpty(menuIds)) {
|
List<ZTreeNode> roleTreeList = this.menuService.menuTreeList(ids);
|
// 分公司(type=2)和供应商(type=3)需要根据经营范围过滤菜单
|
if (tCompany.getType() == 2 || tCompany.getType() == 3) {
|
roleTreeList = filterMenusByBusinessScope(roleTreeList, tCompany);
|
}
|
return roleTreeList;
|
} else {
|
List<ZTreeNode> roleTreeListByUserId = this.menuService.menuTreeListByMenuIds(menuIds, ids);
|
// 分公司(type=2)和供应商(type=3)需要根据经营范围过滤菜单
|
if (tCompany.getType() == 2 || tCompany.getType() == 3) {
|
roleTreeListByUserId = filterMenusByBusinessScope(roleTreeListByUserId, tCompany);
|
}
|
return roleTreeListByUserId;
|
}
|
}
|
|
/**
|
* 根据请求的父级菜单编号设置pcode和层级
|
*/
|
private void menuSetPcode(@Valid Menu menu) {
|
if (ToolUtil.isEmpty(menu.getPcode()) || menu.getPcode().equals("0")) {
|
menu.setPcode("0");
|
menu.setPcodes("[0],");
|
menu.setLevels(1);
|
} else {
|
long code = Long.parseLong(menu.getPcode());
|
Menu pMenu = menuService.selectById(code);
|
Integer pLevels = pMenu.getLevels();
|
menu.setPcode(pMenu.getCode());
|
|
//如果编号和父编号一致会导致无限递归
|
if (menu.getCode().equals(menu.getPcode())) {
|
throw new GunsException(BizExceptionEnum.MENU_PCODE_COINCIDENCE);
|
}
|
|
menu.setLevels(pLevels + 1);
|
menu.setPcodes(pMenu.getPcodes() + "[" + pMenu.getCode() + "],");
|
}
|
}
|
|
/**
|
* 根据企业经营范围过滤菜单
|
* @param menuList 原始菜单列表
|
* @param company 企业信息
|
* @return 过滤后的菜单列表
|
*/
|
private List<ZTreeNode> filterMenusByBusinessScope(List<ZTreeNode> menuList, TCompany company) {
|
List<String> excludeMenuNames = new ArrayList<>();
|
|
// isSpe=2 不经营快车
|
if (company.getIsSpe() != null && company.getIsSpe() == 2) {
|
excludeMenuNames.addAll(SPE_MENUS);
|
}
|
// isTaxi=2 不经营出租车
|
if (company.getIsTaxi() != null && company.getIsTaxi() == 2) {
|
excludeMenuNames.addAll(TAXI_MENUS);
|
}
|
// isCross=2 不经营跨城
|
if (company.getIsCross() != null && company.getIsCross() == 2) {
|
excludeMenuNames.addAll(CROSS_MENUS);
|
}
|
// isHitchRide=2 不经营顺风车
|
if (company.getIsHitchRide() != null && company.getIsHitchRide() == 2) {
|
excludeMenuNames.addAll(HITCH_RIDE_MENUS);
|
}
|
|
if (excludeMenuNames.isEmpty()) {
|
return menuList;
|
}
|
|
return menuList.stream()
|
.filter(node -> !excludeMenuNames.contains(node.getName()))
|
.collect(Collectors.toList());
|
}
|
|
}
|
