package com.matrix.system.common.service.impl;
|
|
import com.matrix.core.constance.MatrixConstance;
|
import com.matrix.core.constance.SystemErrorCode;
|
import com.matrix.core.exception.GlobleException;
|
import com.matrix.core.pojo.PaginationVO;
|
import com.matrix.core.tools.LogUtil;
|
import com.matrix.core.tools.ModelUtils;
|
import com.matrix.core.tools.WebUtil;
|
import com.matrix.system.common.authority.DefaultAuthorityManager;
|
import com.matrix.system.common.bean.SysFnBtnRel;
|
import com.matrix.system.common.bean.SysFunction;
|
import com.matrix.system.common.bean.SysUsers;
|
import com.matrix.system.common.constance.AppConstance;
|
import com.matrix.system.common.constance.AppMessageCode;
|
import com.matrix.system.common.dao.SysFunctionDao;
|
import com.matrix.system.common.service.SysFunctionService;
|
import com.matrix.system.common.tools.ServiceUtil;
|
import jodd.util.StringUtil;
|
import org.apache.commons.collections.CollectionUtils;
|
import org.apache.commons.lang.ArrayUtils;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
|
import javax.annotation.Resource;
|
import java.util.*;
|
|
/**
|
* 系统功能管理
|
*
|
* @author jiangyouyao
|
* @time 2016-11-18
|
*/
|
@Service
|
public class SysFunctionServiceImpl implements SysFunctionService {
|
|
private static final String FN_ID = "fn_id";
|
|
private static final String FN_CODE = "fn_code";
|
|
@Autowired
|
private SysFunctionDao sysFunctionDao;
|
|
@Autowired
|
private ServiceUtil serviceUtil;
|
|
@Resource
|
private DefaultAuthorityManager matrix;
|
|
private String tableName = "sys_function";
|
|
|
@Override
|
public int add(SysFunction sysFunction) {
|
|
// 自己不能是自己的父节点
|
if (sysFunction.getFnParentId() != null && sysFunction.getFnParentId().equals(sysFunction.getFnId())) {
|
throw new GlobleException(AppMessageCode.Function.PARENT_FUNCTION_CANNOT_BE_SELF);
|
}
|
// 功能code去重
|
if (serviceUtil.addCheckRepeat(tableName, FN_CODE, sysFunction.getFnCode())) {
|
throw new GlobleException(SystemErrorCode.DATA_REPEAT, sysFunction.getFnCode());
|
}
|
//设置默认父节点
|
if(sysFunction.getFnParentId()==null){
|
sysFunction.setFnParentId(0L);
|
}
|
|
sysFunction.setFnId(null);
|
SysUsers user = WebUtil.getSessionAttribute(MatrixConstance.LOGIN_KEY);
|
sysFunction.setCreateBy(user.getSuName());
|
sysFunction.setUpdateBy(user.getSuName());
|
|
// 设置功能等级
|
setFunctionGrade(sysFunction);
|
sysFunction.setFnIsDisable(AppConstance.IS_N);
|
int n = sysFunctionDao.insert(sysFunction);
|
// 添加功能所对应的按钮
|
addFnBtnRel(sysFunction);
|
|
return n;
|
|
}
|
|
/**
|
* 添加功能所对应的按钮
|
*
|
* @author JIANGYOUYAO
|
* @param fnId
|
* @email 935090232@qq.com
|
* @date 2017年12月4日
|
* @param sysFunction
|
*/
|
private void addFnBtnRel(SysFunction function) {
|
List<SysFnBtnRel> rels = function.getSysFnBtnRel();
|
List<SysFnBtnRel> newRels = new ArrayList<>();
|
for (SysFnBtnRel sysFnBtnRel : rels) {
|
if (sysFnBtnRel.getBtnValue() == null) {
|
continue;
|
}
|
sysFnBtnRel.setFnId(function.getFnId());
|
sysFnBtnRel.setCreateBy(function.getCreateBy());
|
sysFnBtnRel.setUpdateBy(function.getUpdateBy());
|
newRels.add(sysFnBtnRel);
|
}
|
if (CollectionUtils.isNotEmpty(newRels)) {
|
sysFunctionDao.bathInsertFnButRel(newRels);
|
}
|
}
|
|
/**
|
* 删除功能所对应的按钮
|
*
|
* @author JIANGYOUYAO
|
* @email 935090232@qq.com
|
* @date 2017年12月4日
|
* @param fnId
|
*/
|
private void removerFnBtnRel(Long fnId) {
|
sysFunctionDao.deleteFnBtnRel(fnId);
|
}
|
|
/**
|
* 批量删除功能所对应的按钮
|
*
|
* @author JIANGYOUYAO
|
* @email 935090232@qq.com
|
* @date 2017年12月4日
|
* @param fnId
|
*/
|
private void batchRemoverFnBtnRel(List<String> ids) {
|
|
sysFunctionDao.deleteFnBtnRels(ids);
|
}
|
|
/**
|
* 设置功能的等级
|
*
|
* @author JIANGYOUYAO
|
* @email 935090232@qq.com
|
* @date 2017年12月4日
|
* @param sysFunction
|
*/
|
private void setFunctionGrade(SysFunction sysFunction) {
|
if (null==sysFunction.getFnParentId()||0==sysFunction.getFnParentId()) {
|
sysFunction.setFnGrade(AppConstance.FUNCTION_FIRST_LEVEL);
|
} else {
|
SysFunction parent = sysFunctionDao.selectById(sysFunction.getFnParentId());
|
sysFunction.setFnGrade(parent.getFnGrade() + 1);
|
}
|
}
|
|
@Override
|
public int batchAdd(List<SysFunction> sysFunctionList) {
|
return sysFunctionDao.batchInsert(sysFunctionList);
|
}
|
|
@Transactional(rollbackFor = Exception.class)
|
@Override
|
public int modifyByMap(SysFunction oldSysFunction, SysFunction newSysFunction) {
|
// 自己不能是自己的父节点
|
if (newSysFunction.getFnParentId() != null && newSysFunction.getFnParentId().equals(newSysFunction.getFnId())) {
|
throw new GlobleException("功能的父级不能是自己");
|
}
|
// 功能code去重
|
if (serviceUtil.updateCheckRepeat(tableName, FN_CODE, newSysFunction.getFnCode(), FN_ID,
|
newSysFunction.getFnId())) {
|
throw new GlobleException("code已经存在！");
|
}
|
// 设置功能等级
|
setFunctionGrade(newSysFunction);
|
SysUsers user = WebUtil.getSessionAttribute(MatrixConstance.LOGIN_KEY);
|
// 创建人的复制是为了按钮关系表能够赋值
|
newSysFunction.setCreateBy(oldSysFunction.getCreateBy());
|
newSysFunction.setUpdateBy(user.getSuName());
|
Map<String, Object> modifyMap = null;
|
int dmlResult = 0;
|
try {
|
if (!ModelUtils.isModified(oldSysFunction, newSysFunction)) {
|
return MatrixConstance.DML_SUCCESSS;
|
}
|
modifyMap = ModelUtils.comparePojo2Map(oldSysFunction, newSysFunction);
|
} catch (Exception e) {
|
throw new GlobleException(SystemErrorCode.DATA_UPDATE_FAIL, e, newSysFunction.getFnName());
|
}
|
if (modifyMap.size() > 0) {
|
modifyMap.put("fnId", oldSysFunction.getFnId());
|
dmlResult = sysFunctionDao.updateByMap(modifyMap);
|
}
|
removerFnBtnRel(oldSysFunction.getFnId());
|
addFnBtnRel(newSysFunction);
|
return dmlResult;
|
}
|
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public int modifyByModel(SysFunction sysFunction) {
|
|
return sysFunctionDao.updateByModel(sysFunction);
|
|
}
|
|
@Override
|
public int remove(List<String> list) {
|
int count = 0;
|
for (String fnId : list) {
|
removeById(fnId);
|
count++;
|
}
|
return count;
|
}
|
|
/**
|
* 删除功能和功能下的子功能
|
*
|
* @param fnId
|
* @return
|
*/
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public int removeById(String fnId) {
|
if (fnId == null) {
|
return 0;
|
}
|
List<String> ids = getFunctionChirds(Long.valueOf(fnId), new ArrayList<String>());
|
if (CollectionUtils.isNotEmpty(ids)) {
|
batchRemoverFnBtnRel(ids);
|
}
|
return sysFunctionDao.deleteByIds(ids);
|
}
|
|
@Override
|
public int removeByModel(SysFunction sysFunction) {
|
|
return sysFunctionDao.deleteByModel(sysFunction);
|
|
}
|
|
@Override
|
public List<SysFunction> findInPage(SysFunction sysFunction, PaginationVO pageVo) {
|
|
return sysFunctionDao.selectInPage(sysFunction, pageVo);
|
|
}
|
|
@Override
|
public List<SysFunction> findByModel(SysFunction sysFunction) {
|
|
return sysFunctionDao.selectByModel(sysFunction);
|
|
}
|
|
@Override
|
public int findTotal(SysFunction sysFunction) {
|
|
return sysFunctionDao.selectTotalRecord(sysFunction);
|
|
}
|
|
@Override
|
public SysFunction findById(String fnId) {
|
|
return sysFunctionDao.selectById(Long.valueOf(fnId));
|
|
}
|
|
/**
|
* 获取功能下的子功能id，包含功能自己的id
|
*
|
* @param id
|
* @param ids
|
* @return
|
*/
|
private List<String> getFunctionChirds(Long id, List<String> ids) {
|
SysFunction fnquery = new SysFunction();
|
fnquery.setFnParentId(id);
|
// 获取子节点
|
List<SysFunction> list = sysFunctionDao.selectByModel(fnquery);
|
if (!list.isEmpty()) {
|
for (SysFunction function : list) {
|
// 删除子节点
|
getFunctionChirds(function.getFnId(), ids);
|
}
|
}
|
ids.add(String.valueOf(id));
|
return ids;
|
}
|
|
@Override
|
public void setIsDisable(Long fnId, String status) {
|
SysFunction dbFunction = sysFunctionDao.selectById(fnId);
|
if (status.equals(AppConstance.IS_Y)) {
|
// 启用功能
|
if (dbFunction.getFnIsDisable().equals(AppConstance.IS_Y)) {
|
throw new GlobleException("功能已经是启用状态");
|
} else {
|
|
List<String> list = getFunctionChirds(fnId, new ArrayList<String>());
|
sysFunctionDao.batchChangeStatu(AppConstance.IS_Y, list);
|
}
|
} else {
|
// 禁用功能
|
if (dbFunction.getFnIsDisable().equals(AppConstance.IS_N)) {
|
throw new GlobleException("功能已是禁用状态");
|
} else {
|
|
// 启用子节点,和父节点，因为目前只考虑2级的情况先写死吧
|
List<String> ids = new ArrayList<String>();
|
if (dbFunction.getFnParentId() != null) {
|
ids.add(String.valueOf(dbFunction.getFnParentId()));
|
}
|
sysFunctionDao.batchChangeStatu(AppConstance.IS_N, getFunctionChirds(fnId, ids));
|
|
}
|
}
|
|
}
|
|
@Override
|
public List<SysFunction> findCompanyFunction(Long companyId) {
|
return sysFunctionDao.selectCompanyFunction(companyId);
|
}
|
|
@Override
|
public List<SysFunction> findFunctionByRoleIds(String roleIds) {
|
return sysFunctionDao.selectFunctionByRoleIds(roleIds);
|
}
|
|
@Override
|
public List<SysFunction> findRoleFuntion(String roleId) {
|
// 获取所有的功能--》获取所有的按钮--》对于功能的按钮|构建功能的父子关系--》
|
Map<String, SysFunction> roleFnMap = new HashMap<>(MatrixConstance.COLLECTION_SIZE);
|
// id为空则不需要查询选中项
|
if (roleId != null) {
|
// 把角色所有的功能变成map
|
List<SysFunction> editorFunction = sysFunctionDao.selectFunctionByRoleIds(roleId);
|
for (SysFunction temp : editorFunction) {
|
roleFnMap.put(String.valueOf(temp.getFnId()), temp);
|
}
|
}
|
|
// 只查询出没有被禁用的
|
SysUsers user = WebUtil.getSessionAttribute(MatrixConstance.LOGIN_KEY);
|
// 只能查本公司的
|
List<SysFunction> allFunction = sysFunctionDao.selectCompanyFunction(user.getCompanyId());
|
Map<String, SysFunction> allFnMap = new TreeMap<>();
|
// 构建树形关系
|
for (SysFunction sysFunction : allFunction) {
|
// 用一个map记录下每一个功能
|
allFnMap.put(String.valueOf(sysFunction.getFnId()), sysFunction);
|
}
|
|
List<SysFunction> treeFunction = new ArrayList<>();
|
Set<String> keys = allFnMap.keySet();
|
for (String currentFnId : keys) {
|
|
SysFunction function = allFnMap.get(currentFnId);
|
|
// 判断功能是否为当前角色拥有
|
initHashPower(roleFnMap, currentFnId, function);
|
|
// 如果是一级节点则直接存入菜单
|
if (null==function.getFnParentId() || 0L==function.getFnParentId()) {
|
treeFunction.add(function);
|
} else {
|
// 非一级节点找到父节点后存入
|
SysFunction parentFn = allFnMap.get(String.valueOf(function.getFnParentId()));
|
if(parentFn!=null){
|
List<SysFunction> childs = parentFn.getChilds();
|
if (childs == null) {
|
parentFn.setChilds(new ArrayList<SysFunction>());
|
}
|
parentFn.getChilds().add(function);
|
}
|
|
}
|
|
}
|
return treeFunction;
|
}
|
|
/**
|
* 给已经拥有的权限打上标记
|
*
|
* @author JIANGYOUYAO
|
* @email 935090232@qq.com
|
* @date 2017年12月6日
|
* @param roleFnMap
|
* @param currentFnId
|
* @param function
|
*/
|
private void initHashPower(Map<String, SysFunction> roleFnMap, String currentFnId, SysFunction function) {
|
if (roleFnMap.containsKey(currentFnId)) {
|
// 功能是当前角色拥有的
|
function.setHasThisFn(true);
|
// 判断按钮是否为当前角色拥有
|
String rpfBns = roleFnMap.get(currentFnId).getRpfBns();
|
String[] rpfBnsArray = rpfBns.split(",");
|
|
List<SysFnBtnRel> fnBtnRelList = function.getSysFnBtnRel();
|
|
if (CollectionUtils.isNotEmpty(fnBtnRelList) && ArrayUtils.isNotEmpty(rpfBnsArray)) {
|
|
for (SysFnBtnRel fnBtnRel : fnBtnRelList) {
|
for (String tempRpfBns : rpfBnsArray) {
|
if (StringUtil.isNotBlank(fnBtnRel.getBtnValue()) &&fnBtnRel.getBtnValue().equals(tempRpfBns)) {
|
fnBtnRel.setHasThisBtn(true);
|
break;
|
}
|
}
|
|
}
|
}
|
}
|
}
|
|
|
}
|
