parent
bb018c6053
commit
1cccbfcf08
|
|
@ -6,19 +6,22 @@ import cn.hutool.core.util.StrUtil;
|
|||
import com.czg.account.dto.menu.MenuAddDTO;
|
||||
import com.czg.account.dto.menu.MenuEditDTO;
|
||||
import com.czg.account.entity.SysMenu;
|
||||
import com.czg.account.entity.SysUsersRoles;
|
||||
import com.czg.account.service.SysMenuService;
|
||||
import com.czg.account.vo.MenuVO;
|
||||
import com.czg.exception.ApiNotPrintException;
|
||||
import com.czg.sa.StpKit;
|
||||
import com.czg.service.account.mapper.SysMenuMapper;
|
||||
import com.czg.service.account.mapper.SysUsersRolesMapper;
|
||||
import com.mybatisflex.core.query.QueryWrapper;
|
||||
import com.mybatisflex.spring.service.impl.ServiceImpl;
|
||||
import jakarta.annotation.Resource;
|
||||
import org.springframework.stereotype.Service;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.Arrays;
|
||||
import java.util.List;
|
||||
import java.util.stream.Collectors;
|
||||
import java.util.Arrays;
|
||||
|
||||
/**
|
||||
* 系统菜单 服务层实现。
|
||||
|
|
@ -27,10 +30,25 @@ import java.util.Arrays;
|
|||
@Service
|
||||
public class SysMenuServiceImpl extends ServiceImpl<SysMenuMapper, SysMenu> implements SysMenuService {
|
||||
|
||||
@Resource
|
||||
private SysMenuMapper sysMenuMapper;
|
||||
@Resource
|
||||
private SysUsersRolesMapper sysUsersRolesMapper;
|
||||
|
||||
@Override
|
||||
public List<MenuVO> getMenu() {
|
||||
long sysUserId = StpKit.USER.getLoginIdAsLong();
|
||||
List<SysMenu> allMenus = mapper.selectByUserId(sysUserId, null);
|
||||
List<SysUsersRoles> roleList = sysUsersRolesMapper.selectListByQuery(query().select(SysUsersRoles::getRoleId).eq(SysUsersRoles::getUserId, sysUserId));
|
||||
List<Long> roleIdList = roleList.stream().map(SysUsersRoles::getRoleId).toList();
|
||||
Long shopId = StpKit.USER.getShopId(0L);
|
||||
boolean isEnableSync = StpKit.USER.isEnableSync(shopId);
|
||||
if (isEnableSync && roleIdList.contains(2L)) {
|
||||
List<SysMenu> headShopPromissionList = sysMenuMapper.selectByRoleId(2L);
|
||||
List<SysMenu> branchShopPromissionList = sysMenuMapper.selectByRoleId(3L);
|
||||
allMenus.removeAll(headShopPromissionList);
|
||||
allMenus.addAll(branchShopPromissionList);
|
||||
}
|
||||
List<MenuVO> menuVos = allMenus.stream()
|
||||
.map(menu -> BeanUtil.copyProperties(menu, MenuVO.class))
|
||||
.collect(Collectors.toList());
|
||||
|
|
|
|||
Loading…
Reference in New Issue