212 lines
8.2 KiB
Java
212 lines
8.2 KiB
Java
package com.jsh.erp.controller;
|
||
|
||
import com.alibaba.fastjson.JSONArray;
|
||
import com.alibaba.fastjson.JSONObject;
|
||
import com.jsh.erp.constants.ExceptionConstants;
|
||
import com.jsh.erp.datasource.entities.Function;
|
||
import com.jsh.erp.datasource.entities.User;
|
||
import com.jsh.erp.exception.BusinessRunTimeException;
|
||
import com.jsh.erp.service.functions.FunctionService;
|
||
import com.jsh.erp.service.userBusiness.UserBusinessService;
|
||
import com.jsh.erp.utils.BaseResponseInfo;
|
||
import org.slf4j.Logger;
|
||
import org.slf4j.LoggerFactory;
|
||
import org.springframework.dao.DataAccessException;
|
||
import org.springframework.web.bind.annotation.*;
|
||
|
||
import javax.annotation.Resource;
|
||
import javax.servlet.http.HttpServletRequest;
|
||
import java.util.ArrayList;
|
||
import java.util.List;
|
||
|
||
/**
|
||
* @author ji-sheng-hua jshERP
|
||
*/
|
||
@RestController
|
||
@RequestMapping(value = "/function")
|
||
public class FunctionController {
|
||
private Logger logger = LoggerFactory.getLogger(FunctionController.class);
|
||
|
||
@Resource
|
||
private FunctionService functionService;
|
||
|
||
@Resource
|
||
private UserBusinessService userBusinessService;
|
||
|
||
@PostMapping(value = "/findMenu")
|
||
public JSONArray findMenu(@RequestParam(value="pNumber") String pNumber,
|
||
@RequestParam(value="hasFunction") String hasFunction,
|
||
HttpServletRequest request)throws Exception {
|
||
//存放数据json数组
|
||
JSONArray dataArray = new JSONArray();
|
||
try {
|
||
//当前用户所拥有的功能列表，格式如：[1][2][5]
|
||
String fc = hasFunction;
|
||
List<Function> dataList = functionService.getRoleFunction(pNumber);
|
||
if (dataList.size() != 0) {
|
||
dataArray = getMenuByFunction(dataList, fc);
|
||
}
|
||
} catch (DataAccessException e) {
|
||
logger.error(">>>>>>>>>>>>>>>>>>>查找异常", e);
|
||
}
|
||
return dataArray;
|
||
}
|
||
|
||
public JSONArray getMenuByFunction(List<Function> dataList, String fc) throws Exception {
|
||
JSONArray dataArray = new JSONArray();
|
||
for (Function function : dataList) {
|
||
JSONObject item = new JSONObject();
|
||
List<Function> newList = functionService.getRoleFunction(function.getNumber());
|
||
item.put("id", function.getId());
|
||
item.put("text", function.getName());
|
||
item.put("icon", function.getIcon());
|
||
if (newList.size()>0) {
|
||
JSONArray childrenArr = getMenuByFunction(newList, fc);
|
||
if(childrenArr.size()>0) {
|
||
item.put("children", childrenArr);
|
||
dataArray.add(item);
|
||
}
|
||
} else {
|
||
item.put("url", function.getUrl());
|
||
if (fc.indexOf("[" + function.getId().toString() + "]") != -1) {
|
||
dataArray.add(item);
|
||
}
|
||
}
|
||
}
|
||
return dataArray;
|
||
}
|
||
|
||
/**
|
||
* 角色对应功能显示
|
||
* @param request
|
||
* @return
|
||
*/
|
||
@PostMapping(value = "/findRoleFunction")
|
||
public JSONArray findRoleFunction(@RequestParam("UBType") String type, @RequestParam("UBKeyId") String keyId,
|
||
HttpServletRequest request)throws Exception {
|
||
JSONArray arr = new JSONArray();
|
||
try {
|
||
List<Function> dataListFun = functionService.findRoleFunction("0");
|
||
//开始拼接json数据
|
||
JSONObject outer = new JSONObject();
|
||
outer.put("id", 1);
|
||
outer.put("text", "功能列表");
|
||
outer.put("state", "open");
|
||
//存放数据json数组
|
||
JSONArray dataArray = new JSONArray();
|
||
if (null != dataListFun) {
|
||
//根据条件从列表里面移除"系统管理"
|
||
List<Function> dataList = new ArrayList<Function>();
|
||
for (Function fun : dataListFun) {
|
||
//从session中获取租户id
|
||
String loginName = null;
|
||
Object userInfo = request.getSession().getAttribute("user");
|
||
if(userInfo != null) {
|
||
User user = (User) userInfo;
|
||
loginName = user.getLoginName();
|
||
}
|
||
if(("admin").equals(loginName)) {
|
||
dataList.add(fun);
|
||
} else {
|
||
if(!("系统管理").equals(fun.getName())) {
|
||
dataList.add(fun);
|
||
}
|
||
}
|
||
}
|
||
dataArray = getFunctionList(dataList, type, keyId);
|
||
outer.put("children", dataArray);
|
||
}
|
||
arr.add(outer);
|
||
} catch (Exception e) {
|
||
e.printStackTrace();
|
||
}
|
||
return arr;
|
||
}
|
||
|
||
public JSONArray getFunctionList(List<Function> dataList, String type, String keyId) throws Exception {
|
||
JSONArray dataArray = new JSONArray();
|
||
if (null != dataList) {
|
||
for (Function function : dataList) {
|
||
JSONObject item = new JSONObject();
|
||
item.put("id", function.getId());
|
||
item.put("text", function.getName());
|
||
Boolean flag = false;
|
||
try {
|
||
flag = userBusinessService.checkIsUserBusinessExist(type, keyId, "[" + function.getId().toString() + "]");
|
||
} catch (Exception e) {
|
||
logger.error(">>>>>>>>>>>>>>>>>设置角色对应的功能：类型" + type + " KeyId为： " + keyId + " 存在异常！");
|
||
}
|
||
if (flag == true) {
|
||
item.put("checked", true);
|
||
}
|
||
List<Function> funList = functionService.findRoleFunction(function.getNumber());
|
||
if(funList.size()>0) {
|
||
JSONArray funArr = getFunctionList(funList, type, keyId);
|
||
item.put("children", funArr);
|
||
dataArray.add(item);
|
||
} else {
|
||
dataArray.add(item);
|
||
}
|
||
}
|
||
}
|
||
return dataArray;
|
||
}
|
||
|
||
/**
|
||
* 根据id列表查找功能信息
|
||
* @param functionsIds
|
||
* @param request
|
||
* @return
|
||
*/
|
||
@GetMapping(value = "/findByIds")
|
||
public BaseResponseInfo findByIds(@RequestParam("functionsIds") String functionsIds,
|
||
HttpServletRequest request)throws Exception {
|
||
BaseResponseInfo res = new BaseResponseInfo();
|
||
try {
|
||
List<Function> dataList = functionService.findByIds(functionsIds);
|
||
JSONObject outer = new JSONObject();
|
||
outer.put("total", dataList.size());
|
||
//存放数据json数组
|
||
JSONArray dataArray = new JSONArray();
|
||
if (null != dataList) {
|
||
for (Function function : dataList) {
|
||
JSONObject item = new JSONObject();
|
||
item.put("Id", function.getId());
|
||
item.put("Name", function.getName());
|
||
item.put("PushBtn", function.getPushBtn());
|
||
item.put("op", 1);
|
||
dataArray.add(item);
|
||
}
|
||
}
|
||
outer.put("rows", dataArray);
|
||
res.code = 200;
|
||
res.data = outer;
|
||
} catch (Exception e) {
|
||
e.printStackTrace();
|
||
res.code = 500;
|
||
res.data = "获取数据失败";
|
||
}
|
||
return res;
|
||
}
|
||
/**
|
||
* create by: qiankunpingtai
|
||
* description:
|
||
* 批量删除功能模块信息
|
||
* create time: 2019/3/29 11:15
|
||
* @Param: ids
|
||
* @return java.lang.Object
|
||
*/
|
||
@PostMapping(value = "/batchDeleteFunctionByIds")
|
||
public Object batchDeleteFunctionByIds(@RequestParam("ids") String ids) throws Exception {
|
||
JSONObject result = ExceptionConstants.standardSuccess();
|
||
int i= functionService.batchDeleteFunctionByIds(ids);
|
||
if(i<1){
|
||
logger.error("异常码[{}],异常提示[{}],参数,ids[{}]",
|
||
ExceptionConstants.FUNCTIONS_DELETE_FAILED_CODE,ExceptionConstants.FUNCTIONS_DELETE_FAILED_MSG,ids);
|
||
throw new BusinessRunTimeException(ExceptionConstants.FUNCTIONS_DELETE_FAILED_CODE,
|
||
ExceptionConstants.FUNCTIONS_DELETE_FAILED_MSG);
|
||
}
|
||
return result;
|
||
}
|
||
}
|