Search in sources :

Example 1 with SysMenu

use of com.goku.demo.entity.SysMenu in project springBoot-learn-demo by nbfujx.

the class SysUserServiceImpl method listUserPermissionPO.

@Override
public Set<UserPermissionPO> listUserPermissionPO(SysUser sysUser) {
    // SysUser sysUser = sysUserRepository.findById(userId);
    Set<UserPermissionPO> userPermissionPOS = new HashSet<UserPermissionPO>();
    Set<SysMenu> sysMenus = sysUser.getSysMenus();
    Iterator<SysMenu> smit = sysMenus.iterator();
    while (smit.hasNext()) {
        SysMenu sysMenu = smit.next();
        if (sysMenu.getPermission() != null && !sysMenu.getPermission().isEmpty()) {
            UserPermissionPO uppo = new UserPermissionPO();
            uppo.setUserId(sysUser.getId());
            uppo.setPermission(sysMenu.getPermission());
            userPermissionPOS.add(uppo);
        }
    }
    Set<SysRole> sysRoles = sysUser.getSysRoles();
    Iterator<SysRole> srit = sysRoles.iterator();
    while (srit.hasNext()) {
        SysRole sysRole = srit.next();
        Iterator<SysMenu> srmit = sysRole.getSysMenus().iterator();
        while (srmit.hasNext()) {
            SysMenu sysMenu = srmit.next();
            if (sysMenu.getPermission() != null && !sysMenu.getPermission().isEmpty()) {
                UserPermissionPO uppo = new UserPermissionPO();
                uppo.setUserId(sysUser.getId());
                uppo.setPermission(sysMenu.getPermission());
                userPermissionPOS.add(uppo);
            }
        }
    }
    return userPermissionPOS;
}
Also used : SysMenu(com.goku.demo.entity.SysMenu) UserPermissionPO(com.goku.demo.common.po.UserPermissionPO) SysRole(com.goku.demo.entity.SysRole)

Aggregations

UserPermissionPO (com.goku.demo.common.po.UserPermissionPO)1 SysMenu (com.goku.demo.entity.SysMenu)1 SysRole (com.goku.demo.entity.SysRole)1