forked from github/dataease
113 lines
3.6 KiB
Java
113 lines
3.6 KiB
Java
package io.dataease.service.sys;
|
||
|
||
import io.dataease.base.domain.SysMenu;
|
||
import io.dataease.base.domain.SysMenuExample;
|
||
import io.dataease.base.mapper.SysMenuMapper;
|
||
import io.dataease.base.mapper.ext.ExtMenuMapper;
|
||
import io.dataease.commons.utils.BeanUtils;
|
||
import io.dataease.controller.sys.request.MenuCreateRequest;
|
||
import io.dataease.controller.sys.request.MenuDeleteRequest;
|
||
import org.apache.commons.lang3.ObjectUtils;
|
||
import org.springframework.stereotype.Service;
|
||
import org.springframework.transaction.annotation.Transactional;
|
||
import javax.annotation.Resource;
|
||
import java.util.List;
|
||
|
||
@Service
|
||
public class MenuService {
|
||
|
||
|
||
|
||
private final static Integer DEFAULT_SUBCOUNT = 0;
|
||
public final static Long MENU_ROOT_PID = 0L;
|
||
|
||
@Resource
|
||
private SysMenuMapper sysMenuMapper;
|
||
|
||
|
||
@Resource
|
||
private ExtMenuMapper extMenuMapper;
|
||
|
||
public List<SysMenu> nodesByPid(Long pid){
|
||
SysMenuExample example = new SysMenuExample();
|
||
SysMenuExample.Criteria criteria = example.createCriteria();
|
||
if (ObjectUtils.isEmpty(pid)){
|
||
criteria.andPidEqualTo(MENU_ROOT_PID);
|
||
}else {
|
||
criteria.andPidEqualTo(pid);
|
||
}
|
||
example.setOrderByClause("menu_sort");
|
||
List<SysMenu> sysMenus = sysMenuMapper.selectByExample(example);
|
||
return sysMenus;
|
||
}
|
||
|
||
@Transactional
|
||
public boolean add(MenuCreateRequest menuCreateRequest){
|
||
SysMenu sysMenu = BeanUtils.copyBean(new SysMenu(), menuCreateRequest);
|
||
if (menuCreateRequest.isTop()){
|
||
sysMenu.setPid(MENU_ROOT_PID);
|
||
}
|
||
long now = System.currentTimeMillis();
|
||
sysMenu.setCreateTime(now);
|
||
sysMenu.setUpdateTime(now);
|
||
sysMenu.setCreateBy(null);
|
||
sysMenu.setUpdateBy(null);
|
||
sysMenu.setSubCount(DEFAULT_SUBCOUNT);
|
||
try {
|
||
int insert = sysMenuMapper.insert(sysMenu);
|
||
Long pid = null;
|
||
if ((pid = sysMenu.getPid()) != MENU_ROOT_PID ){
|
||
//这里需要更新上级节点SubCount
|
||
extMenuMapper.incrementalSubcount(pid);
|
||
}
|
||
if (insert == 1){
|
||
return true;
|
||
}
|
||
}catch (Exception e){
|
||
e.printStackTrace();
|
||
}
|
||
return false;
|
||
}
|
||
|
||
@Transactional
|
||
public int delete(MenuDeleteRequest request){
|
||
Long pid = request.getPid();
|
||
if (pid != MENU_ROOT_PID){
|
||
extMenuMapper.decreasingSubcount(pid);
|
||
}
|
||
Long menuId = request.getMenuId();
|
||
|
||
return sysMenuMapper.deleteByPrimaryKey(menuId);
|
||
}
|
||
|
||
|
||
|
||
@Transactional
|
||
public int update(MenuCreateRequest menuCreateRequest){
|
||
SysMenu sysMenu = BeanUtils.copyBean(new SysMenu(), menuCreateRequest);
|
||
if (menuCreateRequest.isTop()){
|
||
sysMenu.setPid(MENU_ROOT_PID);
|
||
}
|
||
|
||
sysMenu.setUpdateBy(null);
|
||
sysMenu.setUpdateTime(System.currentTimeMillis());
|
||
Long menuId = sysMenu.getMenuId();
|
||
SysMenu menu_old = sysMenuMapper.selectByPrimaryKey(menuId);
|
||
//如果PID发生了改变
|
||
//判断oldPid是否是跟节点PID ？ nothing : parent.subcount-1
|
||
//判断newPid是否是跟节点PID ？ nothing : parent.subcount+1
|
||
if (menu_old.getPid() != sysMenu.getPid()){
|
||
Long oldPid = menu_old.getPid();
|
||
if (oldPid != MENU_ROOT_PID){
|
||
extMenuMapper.decreasingSubcount(oldPid);
|
||
}
|
||
if (sysMenu.getPid() != MENU_ROOT_PID){
|
||
extMenuMapper.incrementalSubcount(sysMenu.getPid());
|
||
}
|
||
}
|
||
return sysMenuMapper.updateByPrimaryKeySelective(sysMenu);
|
||
}
|
||
|
||
|
||
}
|